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