import factory from factory.django import DjangoModelFactory from django.utils import timezone from psycopg2.extras import DateTimeTZRange from utils.factories import UserFactory class ProductCategoryFactory(DjangoModelFactory): class Meta: model = "shop.ProductCategory" name = factory.Faker("word") class ProductFactory(DjangoModelFactory): class Meta: model = "shop.Product" name = factory.Faker("word") slug = factory.Faker("word") category = factory.SubFactory(ProductCategoryFactory) description = factory.Faker("paragraph") price = factory.Faker("pyint") available_in = factory.LazyFunction( lambda: DateTimeTZRange( lower=timezone.now(), upper=timezone.now() + timezone.timedelta(31) ) ) class OrderFactory(DjangoModelFactory): class Meta: model = "shop.Order" user = factory.SubFactory(UserFactory) class OrderProductRelationFactory(DjangoModelFactory): class Meta: model = "shop.OrderProductRelation" product = factory.SubFactory(ProductFactory) order = factory.SubFactory(OrderFactory) quantity = 1 handed_out = False