Merge pull request #197 from allanarmstrong/price_rules

Price rules implementation

Author: jamiemtdwyer

shopify/resources/__init__.py

@@ -54,5 +54,6 @@
 from .draft_order import DraftOrder
 from .draft_order_invoice import DraftOrderInvoice
 from .report import Report
-
+from .price_rule import PriceRule
+from .discount_code import DiscountCode
 from ..base import ShopifyResource

shopify/resources/discount_code.py

@@ -0,0 +1,4 @@
+from ..base import ShopifyResource
+
+class DiscountCode(ShopifyResource):
+    _prefix_source = "/admin/price_rules/$price_rule_id/"

shopify/resources/price_rule.py

@@ -0,0 +1,10 @@
+from ..base import ShopifyResource
+from .discount_code import DiscountCode
+
+class PriceRule(ShopifyResource):
+    def add_discount_code(self, discount_code = DiscountCode()):
+        resource = self.post("discount_codes", discount_code.encode())
+        return DiscountCode(PriceRule.format.decode(resource.body))
+
+    def discount_codes(self):
+        return DiscountCode.find(price_rule_id=self.id)

test/discount_code_test.py

@@ -0,0 +1,33 @@
+import shopify
+import json
+from test.test_helper import TestCase
+
+
+class DiscountCodeTest(TestCase):
+    def setUp(self):
+        super(DiscountCodeTest, self).setUp()
+        self.fake("price_rules/1213131/discount_codes/34", method='GET', body=self.load_fixture('discount_code'))
+        self.discount_code = shopify.DiscountCode.find(34, price_rule_id=1213131)
+
+    def test_find_a_specific_discount_code(self):
+        discount_code = shopify.DiscountCode.find(34, price_rule_id=1213131)
+        self.assertEqual("25OFF", discount_code.code)
+
+    def test_update_a_specific_discount_code(self):
+        self.discount_code.code = 'BOGO'
+        self.fake('price_rules/1213131/discount_codes/34',
+                  method='PUT',
+                  status=200,
+                  body=self.load_fixture('discount_code'),
+                  headers={'Content-type': 'application/json'})
+        self.discount_code.save()
+        self.assertEqual('BOGO',
+                         json.loads(self.http.request.data.decode("utf-8"))["discount_code"]["code"]
+                        )
+    def test_delete_a_specific_discount_code(self):
+      self.fake('price_rules/1213131/discount_codes/34', method='DELETE', body='destroyed')
+      self.discount_code.destroy()
+      self.assertEqual('DELETE', self.http.request.get_method())
+
+
+

test/fixtures/discount_code.json

@@ -0,0 +1,9 @@
+{
+  "discount_code": {
+      "code": "25OFF",
+      "id": 34,
+      "usage_count": 3,
+      "created_at": "2016-09-11T09:00:00-04:00",
+      "updated_at": "2016-09-11T09:30:00-04:00"
+    }
+}

test/fixtures/discount_codes.json

@@ -0,0 +1,11 @@
+{
+  "discount_codes": [
+    {
+      "code": "25OFF",
+      "id": 34,
+      "usage_count": 3,
+      "created_at": "2016-09-11T09:00:00-04:00",
+      "updated_at": "2016-09-11T09:30:00-04:00"
+    }
+  ]
+}

test/fixtures/price_rule.json

@@ -0,0 +1,18 @@
+{
+  "price_rule": {
+    "id": 1213131,
+    "title": "BOGO",
+    "target_type": "line_item",
+    "target_selection": "all",
+    "allocation_method": "across",
+    "value_type": "percentage",
+    "value": -100,
+    "once_per_customer": true,
+    "usage_limit": null,
+    "customer_selection": "all",
+    "prerequisite_subtotal_range": null,
+    "prerequisite_shipping_price_range": null,
+    "starts_at": "2017-05-30T04:13:56Z",
+    "ends_at": null
+  }
+}

test/fixtures/price_rules.json

@@ -0,0 +1,36 @@
+{
+  "price_rules": [
+    {
+      "id": 1213131,
+      "title": "BOGO",
+      "target_type": "line_item",
+      "target_selection": "all",
+      "allocation_method": "across",
+      "value_type": "percentage",
+      "value": -100,
+      "once_per_customer": true,
+      "usage_limit": null,
+      "customer_selection": "all",
+      "prerequisite_subtotal_range": null,
+      "prerequisite_shipping_price_range": null,
+      "starts_at": "2017-05-30T04:13:56Z",
+      "ends_at": null
+    },    
+    {
+      "id": 1213132,
+      "title": "TENOFF",
+      "target_type": "line_item",
+      "target_selection": "all",
+      "allocation_method": "each",
+      "value_type": "percentage",
+      "value": -10,
+      "once_per_customer": true,
+      "usage_limit": null,
+      "customer_selection": "all",
+      "prerequisite_subtotal_range": null,
+      "prerequisite_shipping_price_range": null,
+      "starts_at": "2017-05-30T04:13:56Z",
+      "ends_at": null
+    }
+  ]
+}

test/price_rules_test.py

@@ -0,0 +1,74 @@
+import json
+from test.test_helper import TestCase
+
+import shopify
+
+
+class PriceRuleTest(TestCase):
+
+    def setUp(self):
+        super(PriceRuleTest, self).setUp()
+        self.fake('price_rules/1213131', body=self.load_fixture('price_rule'))
+        self.price_rule = shopify.PriceRule.find(1213131)
+
+    def test_get_price_rule(self):
+        self.fake('price_rule/1213131', method='GET', status=200, body=self.load_fixture('price_rule'))
+        price_rule = shopify.PriceRule.find(1213131)
+        self.assertEqual(1213131, price_rule.id)
+
+    def test_get_all_price_rules(self):
+        self.fake('price_rules',
+                  method='GET',
+                  code=200,
+                  body=self.load_fixture('price_rules'))
+        price_rules = shopify.PriceRule.find()
+        self.assertEqual(2, len(price_rules))
+
+    def test_update_price_rule(self):
+        self.price_rule.title = "Buy One Get One"
+        self.fake('price_rules/1213131', method='PUT', status=200, body=self.load_fixture('price_rule'), headers={'Content-type': 'application/json'})
+        self.price_rule.save()
+        self.assertEqual('Buy One Get One', json.loads(self.http.request.data.decode("utf-8"))['price_rule']['title'])
+
+
+    def test_delete_price_rule(self):
+        self.fake('price_rules/1213131', method='DELETE', body='destroyed')
+        self.price_rule.destroy()
+        self.assertEqual('DELETE', self.http.request.get_method())
+
+    def test_price_rule_creation(self):
+        self.fake('price_rules',
+                  method='POST',
+                  code=202,
+                  body=self.load_fixture('price_rule'),
+                  headers={'Content-type': 'application/json'})
+        price_rule = shopify.PriceRule.create({
+            "title": "BOGO",
+            "target_type": "line_item",
+            "target_selection": "all",
+            "allocation_method": "across",
+            "value_type": "percentage",
+            "value": -100,
+            "once_per_customer": 'true',
+            "customer_selection": 'all'
+        })
+        self.assertEqual("BOGO", price_rule.title)
+        self.assertEqual("line_item", price_rule.target_type)
+
+    def test_get_discount_codes(self):
+        self.fake('price_rules/1213131/discount_codes', method='GET', status=200, body=self.load_fixture('discount_codes'))
+        discount_codes = self.price_rule.discount_codes()
+        self.assertEqual(1, len(discount_codes))
+
+    def test_add_discount_code(self):
+        price_rule_discount_fixture = self.load_fixture('discount_code')
+        discount_code = json.loads(price_rule_discount_fixture.decode("utf-8"))
+        self.fake('price_rules/1213131/discount_codes',
+                  method='POST', 
+                  body=price_rule_discount_fixture, 
+                  headers={'Content-type': 'application/json'})
+        price_rule_discount_response = self.price_rule.add_discount_code(shopify.DiscountCode(discount_code['discount_code']))
+        self.assertEqual(discount_code, json.loads(self.http.request.data.decode("utf-8")))
+        self.assertIsInstance(price_rule_discount_response, shopify.DiscountCode)
+        self.assertEqual(discount_code['discount_code']['code'], price_rule_discount_response.code)
+