瀏覽代碼

Improve offer fixture generator

master
David Winterbottom 13 年之前
父節點
當前提交
64e2946081
共有 1 個文件被更改,包括 19 次插入14 次删除
  1. 19
    14
      oscar/test/helpers.py

+ 19
- 14
oscar/test/helpers.py 查看文件

89
     return order
89
     return order
90
 
90
 
91
 
91
 
92
-def create_offer():
92
+def create_offer(name="Dummy offer", offer_type="Site",
93
+                 max_applications=None, range=None, condition=None,
94
+                 benefit=None):
93
     """
95
     """
94
     Helper method for creating an offer
96
     Helper method for creating an offer
95
     """
97
     """
96
-    range = Range.objects.create(name="All products range",
97
-                                 includes_all_products=True)
98
-    condition = Condition.objects.create(range=range,
99
-                                         type=Condition.COUNT,
100
-                                         value=1)
101
-    benefit = Benefit.objects.create(range=range,
102
-                                     type=Benefit.PERCENTAGE,
103
-                                     value=20)
104
-    offer = ConditionalOffer.objects.create(
105
-        name='Dummy offer',
106
-        offer_type='Site',
98
+    if range is None:
99
+        range = Range.objects.create(name="All products range",
100
+                                    includes_all_products=True)
101
+    if condition is None:
102
+        condition = Condition.objects.create(range=range,
103
+                                            type=Condition.COUNT,
104
+                                            value=1)
105
+    if benefit is None:
106
+        benefit = Benefit.objects.create(range=range,
107
+                                        type=Benefit.PERCENTAGE,
108
+                                        value=20)
109
+    return ConditionalOffer.objects.create(
110
+        name=name,
111
+        offer_type=offer_type,
107
         condition=condition,
112
         condition=condition,
108
-        benefit=benefit)
109
-    return offer
113
+        benefit=benefit,
114
+        max_applications=max_applications)
110
 
115
 
111
 
116
 
112
 def create_voucher():
117
 def create_voucher():

Loading…
取消
儲存