Got some tests working
This commit is contained in:
1 parent
ca1019765b
commit
bf9b3e6b75
6 files changed
+108
-6
No files matched your search
@@ -0,0 +1,31 @@
|
||||
import factory
|
||||
|
||||
from random import choice
|
||||
|
||||
from tests.market.models import Item, Shop, Inventory
|
||||
|
||||
|
||||
class ItemFactory(factory.django.DjangoModelFactory):
|
||||
class Meta:
|
||||
model = Item
|
||||
|
||||
name = factory.Faker('name')
|
||||
price = factory.Faker('price')
|
||||
currency = factory.LazyAttribute(lambda _: choice(Item.VALID_CURRENCIES))
|
||||
|
||||
|
||||
class ShopFactory(factory.django.DjangoModelFactory):
|
||||
class Meta:
|
||||
model = Shop
|
||||
|
||||
name = factory.Faker('name')
|
||||
|
||||
|
||||
class InventoryFactory(factory.django.DjangoModelFactory):
|
||||
class Meta:
|
||||
model = Inventory
|
||||
|
||||
shop = factory.SubFactory(ShopFactory)
|
||||
item = factory.SubFactory(ItemFactory)
|
||||
quantity = factory.Sequence(lambda n: n)
|
||||
|
||||
Reference in new issue
Block a user