Happy with test cases

2020-11-06 01:10:14 -08:00
parent 081aec4821
commit c6a44f6426
5 changed files with 116 additions and 47 deletions
@@ -1,6 +1,6 @@
 import factory

-from random import choice
+from random import choice, randint

 from tests.market.models import Item, Shop, Inventory

@@ -10,7 +10,7 @@ class ItemFactory(factory.django.DjangoModelFactory):
        model = Item

    name = factory.Faker('name')
-    price = factory.Faker('price')
+    price = factory.LazyAttribute(lambda _: randint(5, 500))
    currency = factory.LazyAttribute(lambda _: choice(Item.VALID_CURRENCIES))


@@ -23,4 +23,4 @@ class ShopAdmin(AdminConfirmMixin, admin.ModelAdmin):

 admin.site.register(Item, ItemAdmin)
 admin.site.register(Inventory, InventoryAdmin)
-admin.site.register(Shop, ShopAdmin)
+admin.site.register(Shop, ShopAdmin)