misc: tests and data migration
This commit is contained in:
parent
b6a1a60bfb
commit
2bbfe7dd1c
4 changed files with 234 additions and 0 deletions
61
ccdb/misc/factories.py
Normal file
61
ccdb/misc/factories.py
Normal file
|
|
@ -0,0 +1,61 @@
|
|||
import factory
|
||||
|
||||
from .models import MeasurementUnit, MeasurementType, Material, Color, Container
|
||||
|
||||
|
||||
class MeasurementUnitFactory(factory.DjangoModelFactory):
|
||||
class Meta:
|
||||
model = MeasurementUnit
|
||||
|
||||
name = factory.Sequence(lambda n: 'measurement_unit{}'.format(n))
|
||||
code = factory.Sequence(lambda n: 'mu{}'.format(n))
|
||||
unit_class = 'abc'
|
||||
description = 'lorem ipsum'
|
||||
sort_order = factory.Sequence(lambda n: n)
|
||||
|
||||
|
||||
class MeasurementTypeFactory(factory.DjangoModelFactory):
|
||||
class Meta:
|
||||
model = MeasurementType
|
||||
|
||||
name = factory.Sequence(lambda n: 'measurement_type{}'.format(n))
|
||||
code = factory.Sequence(lambda n: 'mt{}'.format(n))
|
||||
measurement_type_class = 'abc'
|
||||
description = 'lorem ipsum'
|
||||
default_measurement_unit = factory.SubFactory(MeasurementUnitFactory)
|
||||
sort_order = factory.Sequence(lambda n: n)
|
||||
|
||||
|
||||
class MaterialFactory(factory.DjangoModelFactory):
|
||||
class Meta:
|
||||
model = Material
|
||||
|
||||
name = factory.Sequence(lambda n: 'material{}'.format(n))
|
||||
code = factory.Sequence(lambda n: 'm{}'.format(n))
|
||||
material_class = 'abc'
|
||||
description = 'lorem ipsum'
|
||||
sort_order = factory.Sequence(lambda n: n)
|
||||
|
||||
|
||||
class ColorFactory(factory.DjangoModelFactory):
|
||||
class Meta:
|
||||
model = Color
|
||||
|
||||
name = factory.Sequence(lambda n: 'color{}'.format(n))
|
||||
code = factory.Sequence(lambda n: 'c{}'.format(n))
|
||||
color_number = factory.Sequence(lambda n: float(n))
|
||||
sort_order = factory.Sequence(lambda n: n)
|
||||
|
||||
|
||||
class ContainerFactory(factory.DjangoModelFactory):
|
||||
class Meta:
|
||||
model = Container
|
||||
|
||||
name = factory.Sequence(lambda n: 'container{}'.format(n))
|
||||
code = factory.Sequence(lambda n: 'c{}'.format(n))
|
||||
application = 'asdf'
|
||||
color = factory.SubFactory(ColorFactory)
|
||||
material = factory.SubFactory(MaterialFactory)
|
||||
volume = factory.Sequence(lambda n: float(n))
|
||||
measurement_unit = factory.SubFactory(MeasurementUnitFactory)
|
||||
sort_order = factory.Sequence(lambda n: n)
|
||||
111
ccdb/misc/migrations/0002_DATA_initial.py
Normal file
111
ccdb/misc/migrations/0002_DATA_initial.py
Normal file
|
|
@ -0,0 +1,111 @@
|
|||
from django.db import migrations
|
||||
from django.forms import modelform_factory
|
||||
|
||||
from ccdb.utils.data import get_data_sources
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
def migrate(apps, schema_editor):
|
||||
sources = get_data_sources()
|
||||
if not sources:
|
||||
return
|
||||
|
||||
c = sources['db0']
|
||||
|
||||
MeasurementUnit = apps.get_model('misc', 'MeasurementUnit')
|
||||
MeasurementType = apps.get_model('misc', 'MeasurementType')
|
||||
Material = apps.get_model('misc', 'Material')
|
||||
Color = apps.get_model('misc', 'Color')
|
||||
Container = apps.get_model('misc', 'Container')
|
||||
|
||||
for model in [Container, MeasurementType, MeasurementUnit, Material, Color]:
|
||||
model.objects.all().delete()
|
||||
|
||||
MUnitForm = modelform_factory(MeasurementUnit, fields=('name', 'code',
|
||||
'unit_class',
|
||||
'description',
|
||||
'sort_order'))
|
||||
MTypeForm = modelform_factory(MeasurementType, fields=('name', 'code',
|
||||
'measurement_type_class',
|
||||
'description',
|
||||
'default_measurement_unit',
|
||||
'sort_order'))
|
||||
MaterialForm = modelform_factory(Material, fields=('name', 'code',
|
||||
'material_class',
|
||||
'description',
|
||||
'sort_order'))
|
||||
ColorForm = modelform_factory(Color, fields=('name', 'code',
|
||||
'color_number',
|
||||
'sort_order'))
|
||||
ContainerForm = modelform_factory(Container, fields=('name', 'code',
|
||||
'application',
|
||||
'color', 'material',
|
||||
'volume',
|
||||
'measurement_unit',
|
||||
'sort_order'))
|
||||
|
||||
for r in c.execute('SELECT * FROM tbl_lu_measurement_units;'):
|
||||
form = MUnitForm(dict(name=r[1], code=r[2],
|
||||
unit_class=r[3], description=r[4],
|
||||
sort_order=int(r[5]) if r[5] else None))
|
||||
if form.is_valid():
|
||||
MeasurementUnit.objects.create(id=r[0], **form.cleaned_data)
|
||||
else:
|
||||
print('measurement unit', r[0:], form.errors.as_data())
|
||||
|
||||
for r in c.execute('SELECT * FROM tbl_lu_measurement_types;'):
|
||||
form = MTypeForm(dict(name=r[1], code=r[2],
|
||||
measurement_type_class=r[3], description=r[4],
|
||||
default_measurement_unit=r[5],
|
||||
sort_order=int(r[6]) if r[6] else None))
|
||||
if form.is_valid():
|
||||
MeasurementType.objects.create(id=r[0], **form.cleaned_data)
|
||||
else:
|
||||
print('measurement type', r[0:], form.errors.as_data())
|
||||
|
||||
for r in c.execute('SELECT * FROM tbl_lu_materials;'):
|
||||
form = MaterialForm(dict(name=r[1], code=r[2],
|
||||
material_class=r[3], description=r[4],
|
||||
sort_order=int(r[5]) if r[5] else None))
|
||||
if form.is_valid():
|
||||
Material.objects.create(id=r[0], **form.cleaned_data)
|
||||
else:
|
||||
print('material', r[0:], form.errors.as_data())
|
||||
|
||||
for r in c.execute('SELECT * FROM tbl_lu_colors;'):
|
||||
form = ColorForm(dict(name=r[1], code=r[2],
|
||||
color_number=float(r[3]) if r[3] else None,
|
||||
sort_order=int(r[4]) if r[4] else None))
|
||||
if form.is_valid():
|
||||
Color.objects.create(id=r[0], **form.cleaned_data)
|
||||
else:
|
||||
print('color', r[0:], form.errors.as_data())
|
||||
|
||||
for r in c.execute('SELECT * FROM tbl_lu_containers;'):
|
||||
form = ContainerForm(dict(name=r[1], code=r[2],
|
||||
application=r[3], color=r[4], material=r[5],
|
||||
volume=float(r[6]) if r[6] else None,
|
||||
measurement_unit=r[7],
|
||||
sort_order=int(r[8]) if r[8] else None))
|
||||
if form.is_valid():
|
||||
Container.objects.create(id=r[0], **form.cleaned_data)
|
||||
else:
|
||||
print('container', r[0:], form.errors.as_data())
|
||||
|
||||
def rollback(apps, schema_editor):
|
||||
MeasurementUnit = apps.get_model('misc', 'MeasurementUnit')
|
||||
MeasurementType = apps.get_model('misc', 'MeasurementType')
|
||||
Material = apps.get_model('misc', 'Material')
|
||||
Color = apps.get_model('misc', 'Color')
|
||||
Container = apps.get_model('misc', 'Container')
|
||||
|
||||
for model in [Container, MeasurementType, MeasurementUnit, Material, Color]:
|
||||
model.objects.all().delete()
|
||||
|
||||
dependencies = [
|
||||
('misc', '0001_initial'),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.RunPython(migrate, rollback),
|
||||
]
|
||||
0
ccdb/misc/tests/__init__.py
Normal file
0
ccdb/misc/tests/__init__.py
Normal file
62
ccdb/misc/tests/test_models.py
Normal file
62
ccdb/misc/tests/test_models.py
Normal file
|
|
@ -0,0 +1,62 @@
|
|||
from django.test import TestCase
|
||||
from django.db import IntegrityError, transaction
|
||||
|
||||
from ..models import MeasurementUnit, MeasurementType, Material, Color, Container
|
||||
from ..factories import MeasurementUnitFactory, MeasurementTypeFactory, \
|
||||
MaterialFactory, ColorFactory, ContainerFactory
|
||||
|
||||
|
||||
class MeasurementUnitTestCase(TestCase):
|
||||
def test_creation(self):
|
||||
m = MeasurementUnitFactory()
|
||||
self.assertTrue(isinstance(m, MeasurementUnit))
|
||||
self.assertEqual(m.__str__(), m.code)
|
||||
|
||||
def test_uniqueness(self):
|
||||
m1 = MeasurementUnitFactory()
|
||||
with transaction.atomic(), self.assertRaises(IntegrityError):
|
||||
MeasurementUnitFactory(name=m1.name, code=m1.code)
|
||||
m3 = MeasurementUnitFactory()
|
||||
self.assertTrue(isinstance(m3, MeasurementUnit))
|
||||
|
||||
|
||||
class MeasurementTypeTestCase(TestCase):
|
||||
def test_creation(self):
|
||||
m = MeasurementTypeFactory()
|
||||
self.assertTrue(isinstance(m, MeasurementType))
|
||||
self.assertEqual(m.__str__(), m.name)
|
||||
|
||||
|
||||
class MaterialTestCase(TestCase):
|
||||
def test_creation(self):
|
||||
m = MaterialFactory()
|
||||
self.assertTrue(isinstance(m, Material))
|
||||
self.assertEqual(m.__str__(), m.name)
|
||||
|
||||
def test_uniqueness(self):
|
||||
m1 = MaterialFactory()
|
||||
with transaction.atomic(), self.assertRaises(IntegrityError):
|
||||
MaterialFactory(name=m1.name, code=m1.code)
|
||||
m3 = MaterialFactory()
|
||||
self.assertTrue(isinstance(m3, Material))
|
||||
|
||||
|
||||
class ColorTestCase(TestCase):
|
||||
def test_creation(self):
|
||||
c = ColorFactory()
|
||||
self.assertTrue(isinstance(c, Color))
|
||||
self.assertEqual(c.__str__(), c.name)
|
||||
|
||||
def test_uniqueness(self):
|
||||
c1 = ColorFactory()
|
||||
with transaction.atomic(), self.assertRaises(IntegrityError):
|
||||
ColorFactory(name=c1.name, code=c1.code, color_number=c1.color_number)
|
||||
c3 = ColorFactory()
|
||||
self.assertTrue(isinstance(c3, Color))
|
||||
|
||||
|
||||
class ContainerTestCase(TestCase):
|
||||
def test_creation(self):
|
||||
c = ContainerFactory()
|
||||
self.assertTrue(isinstance(c, Container))
|
||||
self.assertEqual(c.__str__(), c.name)
|
||||
Loading…
Add table
Reference in a new issue