Merge pull request #13 from AusDTO/edit-work-order

edit work order
AusDTO · Sep 29, 2016 · 936a14e · 936a14e
2 parents 772156b + b2d9b6f
commit 936a14e
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/dmapiclient/data.py b/dmapiclient/data.py
@@ -663,3 +663,9 @@ def create_work_order(self, briefId, supplierCode, workOrder={}):
     def get_work_order(self, work_order_id):
         return self._get(
             "/work-orders/{}".format(work_order_id))
+
+    def update_work_order(self, workOrderId, workOrder):
+        return self._patch(
+            "/work-orders/{}".format(workOrderId),
+            data={"workOrder": workOrder},
+        )
diff --git a/tests/test_apiclient.py b/tests/test_apiclient.py
@@ -1878,6 +1878,22 @@ def test_get_work_order(self, data_client, rmock):
 
         assert result == {"workOrder": "result"}
 
+    def test_update_work_order(self, data_client, rmock):
+        rmock.patch(
+            "http://baseurl/work-orders/1",
+            json={"workOrder": "result"},
+            status_code=200,
+        )
+
+        result = data_client.update_work_order(1, {"foo": "bar"})
+
+        assert result == {"workOrder": "result"}
+        assert rmock.last_request.json() == {
+            "workOrder": {
+                "foo": "bar"
+            }
+        }
+
 
 class TestDataAPIClientIterMethods(object):
     def _test_find_iter(self, data_client, rmock, method_name, model_name, url_path):