Added pause line item function.

Author: RidleyLarsen

dynatademand/api.py

@@ -167,6 +167,17 @@ def get_projects(self):
     def get_project_detailed_report(self, project_id):
         return self._api_get('/projects/{}/detailedReport'.format(project_id))
 
+    def pause_line_item(self, project_id, line_item_id):
+        # Stops traffic to a line item.
+        response_data = self._api_post('/projects/{}/lineItems/{}/pause'.format(project_id, line_item_id), {})
+        if response_data.get('status').get('message') != 'success':
+            raise DemandAPIError(
+                "Could not close project. Demand API responded with: {}".format(
+                    response_data
+                )
+            )
+        return response_data
+
     def get_line_item(self, project_id, line_item_id):
         return self._api_get('/projects/{}/lineItems/{}'.format(project_id, line_item_id))
 

tests/test_line_items.py

@@ -11,6 +11,7 @@
     from mock import patch
 
 from dynatademand.api import DemandAPIClient
+from dynatademand.errors import DemandAPIError
 
 BASE_HOST = "http://test-url.example"
 
@@ -46,3 +47,14 @@ def test_get_line_item_detailed_report(self):
         self.api.get_line_item_detailed_report(1, 100)
         self.assertEqual(len(responses.calls), 1)
         self.assertEqual(responses.calls[0].response.json(), line_item_detailed_report_json)
+
+    @responses.activate
+    def test_pause_lineitem(self):
+        # Tests pausing a line item.
+        responses.add(responses.POST, '{}/sample/v1/projects/24/lineItems/180/pause'.format(BASE_HOST), json={'status': {'message': 'success'}}, status=200)
+        responses.add(responses.POST, '{}/sample/v1/projects/24/lineItems/180/pause'.format(BASE_HOST), json={'status': {'message': 'error'}}, status=200)
+        self.api.pause_line_item(24, 180)
+        self.assertEqual(len(responses.calls), 1)
+        with self.assertRaises(DemandAPIError):
+            self.api.pause_line_item(24, 180)
+        self.assertEqual(len(responses.calls), 2)