Add tests

Author: NathanWorkman

tests/test_files/examples/lineitem_update.json

@@ -0,0 +1,78 @@
+{
+  "extLineItemId": "lineItem002",
+  "title": "US College",
+  "countryISOCode": "US",
+  "languageISOCode": "en",
+  "surveyURL": "www.mysurvey.com/live/survey?pid=<#DubKnowledge[1500/Entity id]>&k2=<#Project[Secure Key 2]>&psid=<#IdParameter[Value]>",
+  "surveyTestURL": "www.mysurvey.com/test/survey?pid=<#DubKnowledge[1500/Entity id]>&k2=<#Project[Secure Key 2]>&psid=<#IdParameter[Value]>",
+  "indicativeIncidence": 20,
+  "daysInField": 20,
+  "lengthOfInterview": 10,
+  "deliveryType": "BALANCED",
+  "sources": [
+    {
+      "id": 100
+    }
+  ],
+  "targets": [
+    {
+      "count": 200,
+      "dailyLimit": 0,
+      "type": "COMPLETE"
+    }
+  ],
+  "quotaPlan": {
+    "filters": [
+      {
+        "attributeId": "61961",
+        "options": [
+          "3",
+          "4"
+        ],
+        "operator": "include"
+      }
+    ],
+    "quotaGroups": [
+      {
+        "name": "Gender Distribution",
+        "quotaCells": [
+          {
+            "quotaNodes": [
+              {
+                "attributeId": "11",
+                "options": [
+                  "1"
+                ]
+              }
+            ],
+            "count": 130
+          },
+          {
+            "quotaNodes": [
+              {
+                "attributeId": "11",
+                "options": [
+                  "2"
+                ]
+              }
+            ],
+            "count": 70
+          }
+        ]
+      }
+    ]
+  },
+  "state": "PROVISIONED",
+  "stateReason": "Created by Client",
+  "stateLastUpdatedAt": "04/01/2018 00:00:00",
+  "createdAt": "04/01/2018 00:00:00",
+  "updatedAt": "04/01/2018 00:00:00",
+  "launchedAt": null,
+  "endLinks": {
+    "complete": "https://api.researchnow.com/respondent/exit?rst=1&psid={psid}&med={calculatedSecurityCode}",
+    "screenout": "https://api.researchnow.com/respondent/exit?rst=2&psid={psid}",
+    "overquota": "https://api.researchnow.com/respondent/exit?rst=3&psid={psid}",
+    "securityKey1": "35040",
+    "securityLevel": "MEDIUM"
+  }
+}

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,22 @@ 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_update_line_item(self):
+        with open('./tests/test_files/examples/lineitem_update.json', 'r') as new_lineitem_file:
+            update_lineitem_data = json.load(new_lineitem_file)
+        
+        # Success response
+        responses.add(responses.POST, '{}/sample/v1/projects/1/lineItems/1'.format(BASE_HOST), json={'status': {'message': 'success'}}, status=200)
+        # Response with error status
+        responses.add(responses.POST, '{}/sample/v1/projects/1/lineItems/1'.format(BASE_HOST), json={'status': {'message': 'error'}}, status=200)
+        
+        # Test success response
+        self.api.update_line_item(1, 1, update_lineitem_data)
+        self.assertEqual(len(responses.calls), 1)
+
+        # Test error response
+        with self.assertRaises(DemandAPIError):
+            self.api.update_line_item(1, 1, update_lineitem_data)
+            self.assertEqual(len(responses.calls), 2)