43 lines
1.1 KiB
Python
Executable File
43 lines
1.1 KiB
Python
Executable File
import factory
|
|
from faker import Factory as FakerFactory
|
|
|
|
from outbo.apps.organizations import models
|
|
|
|
faker = FakerFactory.create()
|
|
|
|
|
|
class AgencyOrganizationFactory(factory.django.DjangoModelFactory):
|
|
name = factory.LazyAttribute(lambda x: faker.company())
|
|
|
|
class Meta:
|
|
model = models.AgencyOrganization
|
|
|
|
|
|
class ClientOrganizationFactory(factory.django.DjangoModelFactory):
|
|
name = factory.LazyAttribute(lambda x: faker.company())
|
|
|
|
class Meta:
|
|
model = models.ClientOrganization
|
|
|
|
|
|
class SupplierOrganizationFactory(factory.django.DjangoModelFactory):
|
|
name = factory.LazyAttribute(lambda x: faker.company())
|
|
|
|
class Meta:
|
|
model = models.SupplierOrganization
|
|
|
|
|
|
class CategoryFactory(factory.django.DjangoModelFactory):
|
|
name = factory.LazyAttribute(lambda x: faker.word())
|
|
|
|
class Meta:
|
|
model = models.Category
|
|
|
|
|
|
class BrandFactory(factory.django.DjangoModelFactory):
|
|
name = factory.LazyAttribute(lambda x: faker.word())
|
|
client = factory.SubFactory(ClientOrganizationFactory)
|
|
|
|
class Meta:
|
|
model = models.Brand
|