63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
from factory import DjangoModelFactory, Sequence, SubFactory
|
|
from factory.fuzzy import FuzzyText, FuzzyChoice, FuzzyInteger
|
|
|
|
from .models import Region, Site, MunicipalLocation, StudyLocation, StorageLocation
|
|
|
|
|
|
class RegionFactory(DjangoModelFactory):
|
|
class Meta:
|
|
model = Region
|
|
|
|
name = Sequence(lambda n: 'region{}'.format(n))
|
|
code = Sequence(lambda n: 'r{}'.format(n))
|
|
sort_order = Sequence(lambda n: n)
|
|
|
|
|
|
class SiteFactory(DjangoModelFactory):
|
|
class Meta:
|
|
model = Site
|
|
|
|
region = SubFactory(RegionFactory)
|
|
name = Sequence(lambda n: 'site{}'.format(n))
|
|
code = Sequence(lambda n: 's{}'.format(n))
|
|
description = FuzzyText(length=100)
|
|
sort_order = Sequence(lambda n: n)
|
|
|
|
|
|
class MunicipalLocationFactory(DjangoModelFactory):
|
|
class Meta:
|
|
model = MunicipalLocation
|
|
|
|
name = Sequence(lambda n: 'municipal_location{}'.format(n))
|
|
code = Sequence(lambda n: 'ml{}'.format(n))
|
|
municipal_location_type = FuzzyText(length=50)
|
|
description = FuzzyText(length=255)
|
|
sort_order = Sequence(lambda n: n)
|
|
|
|
|
|
class StudyLocationFactory(DjangoModelFactory):
|
|
class Meta:
|
|
model = StudyLocation
|
|
|
|
site = SubFactory(SiteFactory)
|
|
name = Sequence(lambda n: 'study_location{}'.format(n))
|
|
code = Sequence(lambda n: 'sl{}'.format(n))
|
|
treatment_type = FuzzyText(length=100)
|
|
municipal_location = SubFactory(MunicipalLocationFactory)
|
|
collecting_location = FuzzyChoice([True, False])
|
|
description = FuzzyText(length=255)
|
|
sort_order = Sequence(lambda n: n)
|
|
|
|
|
|
class StorageLocationFactory(DjangoModelFactory):
|
|
class Meta:
|
|
model = StorageLocation
|
|
|
|
code = Sequence(lambda n: 'sl{}'.format(n))
|
|
facility = FuzzyText(length=100)
|
|
building = FuzzyText(length=100)
|
|
room = FuzzyText(length=50)
|
|
freezer = FuzzyText(length=50)
|
|
temp_c = FuzzyInteger(-100, 100)
|
|
description = FuzzyText(length=255)
|
|
sort_order = Sequence(lambda n: n)
|