Got some tests working

2020-11-02 09:58:41 -08:00
parent ca1019765b
commit bf9b3e6b75
6 changed files with 108 additions and 6 deletions
@@ -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)
+