Merge pull request #8 from researchnow/feature/get-feasibility-endpoint

Added feasibility endpoint & tests.

Author: DynataRidley

dynatademand/api.py

@@ -112,3 +112,6 @@ def logout(self):
 
     def get_project(self, project_id):
         return self._api_get('/projects/{}'.format(project_id))
+
+    def get_feasibility(self, project_id):
+        return self._api_get('/projects/{}/feasibility'.format(project_id))

tests/test_feasibility_pricing.py

@@ -0,0 +1,30 @@
+# encoding: utf-8
+from __future__ import unicode_literals, print_function
+
+import json
+import unittest
+import responses
+
+try:
+    from unittest.mock import patch
+except ImportError:
+    from mock import patch
+
+from dynatademand.api import DemandAPIClient
+
+BASE_HOST = "http://test-url.example"
+
+
+class TestFeasibilityEndpoints(unittest.TestCase):
+    def setUp(self):
+        self.api = DemandAPIClient(client_id='test', username='testuser', password='testpass', base_host=BASE_HOST)
+        self.api._access_token = 'Bearer testtoken'
+
+    @responses.activate
+    def test_get_feasibility(self):
+        with open('./tests/test_files/get_feasibility.json', 'r') as project_file:
+            project_json = json.load(project_file)
+        responses.add(responses.GET, '{}/sample/v1/projects/1/feasibility'.format(BASE_HOST), json=project_json, status=200)
+        self.api.get_feasibility(1)
+        self.assertEqual(len(responses.calls), 1)
+        self.assertEqual(responses.calls[0].response.json(), project_json)

tests/test_files/get_feasibility.json

@@ -0,0 +1,76 @@
+{
+  "data": [
+    {
+      "extLineItemId": "lineItem001",
+      "feasibility": {
+        "status": "READY",
+        "costPerInterview": 3.14,
+        "expiry": "05/01/2018 00:00:00",
+        "currency": "USD",
+        "feasible": true,
+        "totalCount": 50343,
+        "valueCounts": [
+          {
+            "quotaCells": [
+              {
+                "quotaNodes": [
+                  {
+                    "attributeId": "11",
+                    "options": [
+                      "1"
+                    ]
+                  }
+                ],
+                "feasibilityCount": 1000
+              },
+              {
+                "quotaNodes": [
+                  {
+                    "attributeId": "11",
+                    "options": [
+                      "2"
+                    ]
+                  }
+                ],
+                "feasibilityCount": 1200
+              }
+            ]
+          }
+        ]
+      },
+      "quote": {
+        "costPerUnit": 3.14,
+        "currency": "USD",
+        "detailedQuote": [
+          {
+            "costPerUnit": 2.14,
+            "estimatedCost": 2140,
+            "title": "Completes",
+            "type": "BASE",
+            "units": 1000
+          },
+          {
+            "costPerUnit": 1,
+            "estimatedCost": 1000,
+            "title": "Completes",
+            "type": "PREMIUM",
+            "units": 1000
+          }
+        ],
+        "estimatedCost": 3140
+      }
+    },
+    {
+      "extLineItemId": "lineItem002",
+      "feasibility": {
+        "status": "PROCESSING",
+        "costPerInterview": null,
+        "expiry": null,
+        "currency": null,
+        "feasible": null,
+        "totalCount": null,
+        "valueCounts": []
+      }
+    }
+  ]
+}