Merge pull request #58 from dynata/feat-quota-cell-status

updating for set quotacell status endpoint

Author: DynataRidley

README.md

@@ -73,7 +73,9 @@ Links to the Demand API documentation are included for each function.
 [Launch Line Item](https://developers.dynata.com/demand-api-reference/core-resources/lineitems/post-lineitem-launch): launch_line_item(project_id, line_item_id)  
 [Pause Line Item](https://developers.dynata.com/demand-api-reference/core-resources/lineitems/post-lineitem-pause): pause_line_item(project_id, line_item_id)  
 [Update Line Item](https://developers.dynata.com/demand-api-reference/core-resources/lineitems/post-lineitem): update_line_item(project_id, line_item_id, line_item_data)  
-[Get Line Item Detailed Report](https://developers.dynata.com/demand-api-reference/core-resources/lineitems/get-detailed-line-item): get_line_item_detailed_report(project_id, line_item_id)  
+[Get Line Item Detailed Report](https://developers.dynata.com/demand-api-reference/core-resources/lineitems/get-detailed-line-item): get_line_item_detailed_report(project_id, line_item_id) 
+[Launch Quota cell](https://developers.dynata.com/demand-api-reference/core-resources/lineitems/post-quota-cell-launch): set_quotacell_status(project_id, line_item_id, quota_cell_id, launch)  
+[Pause Quota cell](https://developers.dynata.com/demand-api-reference/core-resources/lineitems/post-quota-cell-pause): set_quotacell_status(project_id, line_item_id, quota_cell_id, pause)  
 
 ### Misc Functions
 

dynatademand/api.py

@@ -352,6 +352,27 @@ def pause_line_item(self, project_id, line_item_id):
             )
         return response_data
 
+    def set_quotacell_status(self, project_id, line_item_id, quota_cell_id, action):
+        # Stops traffic to a line item.
+        self.validator.validate_request(
+            'set_quotacell_status',
+            path_data={
+                'extProjectId': '{}'.format(project_id),
+                'extLineItemId': '{}'.format(line_item_id),
+                'quotaCellId': '{}'.format(quota_cell_id),
+                'action': '{}'.format(action),
+            },
+        )
+        response_data = self._api_post('/projects/{}/lineItems/{}/quotaCells/{}/{}'.format(
+            project_id, line_item_id, quota_cell_id, action), {})
+        if response_data.get('status').get('message') != 'success':
+            raise DemandAPIError(
+                "Could not {} quotacell. Demand API responded with: {}".format(
+                    action, response_data
+                )
+            )
+        return response_data
+
     def get_line_item(self, project_id, line_item_id):
         self.validator.validate_request(
             'get_line_item',

dynatademand/schemas/request/path/set_quotacell_status.json

@@ -0,0 +1,27 @@
+{
+    "type": "object",
+    "properties": {
+        "extProjectId": {
+            "type": "string",
+            "required": true
+        },
+        "extLineItemId": {
+            "type": "string",
+            "required": true
+        },
+        "quotaCellId": {
+            "type": "string",
+            "required": true
+        },
+        "action": {
+            "type": "string",
+            "required": true
+        }
+    },
+    "required": [
+        "extProjectId",
+        "extLineItemId",
+        "quotaCellId",
+        "action"
+    ]
+}

dynatademand/validator.py

@@ -32,6 +32,9 @@
     'pause_line_item': ['path', ],
     'update_line_item': ['path', 'body'],
 
+    # quotaCells
+    'set_quotacell_status': ['path', ],
+
     # Events
     'get_events': ['query', ],
     'get_event': ['path', ],

tests/test_line_items.py

@@ -159,6 +159,55 @@ def test_pause_line_item(self):
             self.api.pause_line_item(24, 180)
         self.assertEqual(len(responses.calls), 2)
 
+    @responses.activate
+    def test_set_quotacell_status(self):
+        # Tests launching a quotacell.
+        responses.add(
+            responses.POST,
+            '{}/sample/v1/projects/24/lineItems/180/quotaCells/1/launch'.format(BASE_HOST),
+            json={'status': {'message': 'success'}},
+            status=200
+        )
+        # Response with error launching a quotacell.
+        responses.add(
+            responses.POST,
+            '{}/sample/v1/projects/24/lineItems/180/quotaCells/1/launch'.format(BASE_HOST),
+            json={'status': {'message': 'error'}},
+            status=200
+        )
+        # Tests pausing a quotacell.
+        responses.add(
+            responses.POST,
+            '{}/sample/v1/projects/24/lineItems/180/quotaCells/1/pause'.format(BASE_HOST),
+            json={'status': {'message': 'success'}},
+            status=200
+        )
+        # Response with error for pausing a quotacell
+        responses.add(
+            responses.POST,
+            '{}/sample/v1/projects/24/lineItems/180/quotaCells/1/pause'.format(BASE_HOST),
+            json={'status': {'message': 'error'}},
+            status=200
+        )
+
+        # Test successful response for launch quotacell.
+        self.api.set_quotacell_status(24, 180, 1, "launch")
+        self.assertEqual(len(responses.calls), 1)
+
+        # Test error response for launch quotacell.
+        with self.assertRaises(DemandAPIError):
+            self.api.set_quotacell_status(24, 180, 1, "launch")
+        self.assertEqual(len(responses.calls), 2)
+
+        # Test successful response for pause quotacell.
+        self.api.set_quotacell_status(24, 180, 1, "pause")
+        self.assertEqual(len(responses.calls), 3)
+
+        # Test error response for pause quotacell.
+        with self.assertRaises(DemandAPIError):
+            self.api.set_quotacell_status(24, 180, 1, "pause")
+        self.assertEqual(len(responses.calls), 4)
+
     @responses.activate
     def test_update_line_item(self):
         # Tests updating a line item.