52 lines
1.2 KiB
Python
52 lines
1.2 KiB
Python
|
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
|