Merge branch 'dev' into POP-2285

Author: dynata-bradley

dynatademand/api.py

@@ -152,8 +152,11 @@ def get_project(self, project_id):
     def get_projects(self):
         return self._api_get('/projects')
 
-    def get_lineitem(self, project_id, lineitem_id):
-        return self._api_get('/projects/{}/lineItems/{}'.format(project_id, lineitem_id))
+    def get_line_item(self, project_id, line_item_id):
+        return self._api_get('/projects/{}/lineItems/{}'.format(project_id, line_item_id))
+
+    def get_line_items(self, project_id):
+        return self._api_get('/projects/{}/lineItems'.format(project_id))
 
     def get_feasibility(self, project_id):
         return self._api_get('/projects/{}/feasibility'.format(project_id))

tests/test_files/get_lineitem.json renamed to tests/test_files/get_line_item.json

@@ -0,0 +1,3 @@
+{
+  "test": "Literally anything can go here"
+}

tests/test_files/get_line_items.json

@@ -0,0 +1,39 @@
+# 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 TestLineItemEndpoints(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_line_item(self):
+        with open('./tests/test_files/get_line_item.json', 'r') as line_item_file:
+            line_item_json = json.load(line_item_file)
+        responses.add(responses.GET, '{}/sample/v1/projects/1/lineItems/100'.format(BASE_HOST), json=line_item_json, status=200)
+        self.api.get_line_item(1, 100)
+        self.assertEqual(len(responses.calls), 1)
+        self.assertEqual(responses.calls[0].response.json(), line_item_json)
+
+    @responses.activate
+    def test_get_line_items(self):
+        with open('./tests/test_files/get_line_items.json', 'r') as line_item_file:
+            line_item_json = json.load(line_item_file)
+        responses.add(responses.GET, '{}/sample/v1/projects/1/lineItems'.format(BASE_HOST), json=line_item_json, status=200)
+        self.api.get_line_items(1)
+        self.assertEqual(len(responses.calls), 1)
+        self.assertEqual(responses.calls[0].response.json(), line_item_json)