POP-2283 get survey topics

Author: wogsland

dynatademand/api.py

@@ -118,3 +118,6 @@ def get_lineitem(self, project_id, lineitem_id):
 
     def get_feasibility(self, project_id):
         return self._api_get('/projects/{}/feasibility'.format(project_id))
+
+    def get_survey_topics(self):
+        return self._api_get('/categories/surveyTopics')

tests/test_categories.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 TestCategoryEndpoints(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_survey_topics(self):
+        with open('./tests/test_files/get_survey_topics.json', 'r') as survey_topics_file:
+            survey_topics_json = json.load(survey_topics_file)
+        responses.add(responses.GET, '{}/sample/v1/categories/surveyTopics'.format(BASE_HOST), json=survey_topics_json, status=200)
+        self.api.get_survey_topics()
+        self.assertEqual(len(responses.calls), 1)
+        self.assertEqual(responses.calls[0].response.json(), survey_topics_json)

tests/test_files/get_survey_topics.json

@@ -0,0 +1,5 @@
+{
+  "survey": {
+    "topic": "milk"
+  }
+}