Browse Source

Updated create_product test helper to accept extra kwargs

master
David Winterbottom 15 years ago
parent
commit
45a0755c2a
1 changed files with 5 additions and 6 deletions
  1. 5
    6
      oscar/test/helpers.py

+ 5
- 6
oscar/test/helpers.py View File

@@ -1,14 +1,13 @@
1 1
 from oscar.apps.product.models import ItemClass, Item
2 2
 from oscar.apps.stock.models import Partner, StockRecord
3 3
 
4
-def create_product(price=None):
4
+def create_product(price=None, title="Dummy title", item_class="Dummy item class", partner="Dummy partner"):
5 5
     u"""
6
-    Helper method for creating products that are used in 
7
-    tests.
6
+    Helper method for creating products that are used in tests.
8 7
     """
9
-    ic,_ = ItemClass._default_manager.get_or_create(name="Dummy item class")
10
-    item = Item._default_manager.create(title="Dummy product", item_class=ic)
8
+    ic,_ = ItemClass._default_manager.get_or_create(name=item_class)
9
+    item = Item._default_manager.create(title=title, item_class=ic)
11 10
     if price:
12
-        partner,_ = Partner._default_manager.get_or_create(name="Dummy partner")
11
+        partner,_ = Partner._default_manager.get_or_create(name=partner)
13 12
         sr = StockRecord._default_manager.create(product=item, partner=partner, price_excl_tax=price)
14 13
     return item

Loading…
Cancel
Save