Add tests for update_tracking

Author: mllemango

shopify/resources/fulfillment.py

@@ -17,7 +17,7 @@ def open(self):
     def update_tracking(self, tracking_info, notify_customer):
         fulfill = FulfillmentV2()
         fulfill.id = self.id
-        fulfill.update_tracking(tracking_info, notify_customer)
+        self._load_attributes_from_response(fulfill.update_tracking(tracking_info, notify_customer))
 
 
 class FulfillmentOrders(ShopifyResource):
@@ -34,4 +34,4 @@ def update_tracking(self, tracking_info, notify_customer):
                 "notify_customer": notify_customer
             }
         }
-        self._load_attributes_from_response(self.post("update_tracking", json.dumps(body).encode()))
+        return self.post("update_tracking", json.dumps(body).encode())

test/fixtures/fulfillment.json

@@ -5,7 +5,7 @@
     "order_id": 450789469,
     "service": "manual",
     "status": "pending",
-    "tracking_company": null,
+    "tracking_company": "null-company",
     "updated_at": "2013-11-01T16:06:08-04:00",
     "tracking_number": "1Z2345",
     "tracking_numbers": [
@@ -46,4 +46,4 @@
       }
     ]
   }
-}
+}

test/fulfillment_test.py

@@ -3,7 +3,7 @@
 from pyactiveresource.activeresource import ActiveResource
 
 class FulFillmentTest(TestCase):
-  
+
     def setUp(self):
         super(FulFillmentTest, self).setUp()
         self.fake("orders/450789469/fulfillments/255858046", method='GET', body=self.load_fixture('fulfillment'))
@@ -40,3 +40,24 @@ def test_able_to_cancel_fulfillment(self):
         self.assertEqual('pending', fulfillment.status)
         fulfillment.cancel()
         self.assertEqual('cancelled', fulfillment.status)
+
+    def test_update_tracking(self):
+        fulfillment = shopify.Fulfillment.find(255858046, order_id=450789469)
+
+        tracking_info = { "number": 1111, "url": "http://www.my-url.com", "company": "my-company"}
+        notify_customer = False
+
+        update_tracking = self.load_fixture('fulfillment')
+        update_tracking = update_tracking.replace(b'null-company', b'my-company')
+        update_tracking = update_tracking.replace(b'http://www.google.com/search?q=1Z2345', b'http://www.my-url.com')
+        update_tracking = update_tracking.replace(b'1Z2345', b'1111')
+
+        self.fake("fulfillments/255858046/update_tracking", method="POST", headers={'Content-type': 'application/json'}, body=update_tracking)
+
+        self.assertEqual("null-company", fulfillment.tracking_company)
+        self.assertEqual("1Z2345", fulfillment.tracking_number)
+        self.assertEqual("http://www.google.com/search?q=1Z2345", fulfillment.tracking_url)
+        fulfillment.update_tracking(tracking_info, notify_customer)
+        self.assertEqual("my-company", fulfillment.tracking_company)
+        self.assertEqual('1111', fulfillment.tracking_number)
+        self.assertEqual('http://www.my-url.com', fulfillment.tracking_url)