Merge pull request #26 from dynata/POP-2429

POP-2429 added get invoice pdf endpoint

Author: DynataRidley

dynatademand/api.py

@@ -68,6 +68,8 @@ def _api_get(self, uri, query_params=None):
             raise DemandAPIError('Demand API request to {} failed with status {}. Response: {}'.format(
                 url, response.status_code, response.content
             ))
+        if response.headers['content-type'] == 'application/pdf':
+            return response.content
         return response.json()
 
     def authenticate(self):
@@ -187,6 +189,13 @@ def create_project(self, project_data):
             )
         return response_data
 
+    def get_invoice(self, project_id):
+        self.validator.validate_request(
+            'get_event',
+            path_data={'extProjectId': '{}'.format(project_id)},
+        )
+        return self._api_get('/projects/{}/invoices'.format(project_id))
+
     def buy_project(self, project_id, buy_data):
         '''
             Buy the line items for a project, agreeing to the price. A Line Item

dynatademand/schemas/request/path/get_invoice.json

@@ -0,0 +1,12 @@
+{
+  "type": "object",
+  "properties": {
+    "extProjectId": {
+      "type": "string",
+      "required": true
+    }
+  },
+  "required": [
+    "extProjectId"
+  ]
+}

dynatademand/validator.py

@@ -16,6 +16,9 @@
     'buy_project': ['path', 'body', ],
     'get_project_detailed_report': ['path', ],
 
+    # Invoices
+    'get_invoice': ['path', ],
+
     # Line items
     'close_line_item': ['path', ],
     'create_line_item': ['path', 'body', ],

tests/test_files/get_invoice.pdf

@@ -0,0 +1,29 @@
+# encoding: utf-8
+from __future__ import unicode_literals, print_function
+
+import unittest
+import responses
+
+from dynatademand.api import DemandAPIClient
+
+BASE_HOST = "http://test-url.example"
+
+
+class TestInvoiceEndpoints(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_invoice(self):
+        with open('./tests/test_files/get_invoice.pdf', 'rb') as invoice_file:
+            responses.add(
+                responses.GET,
+                '{}/sample/v1/projects/1337/invoices'.format(BASE_HOST),
+                body=invoice_file.read(),
+                content_type='application/pdf',
+                stream=True,
+                status=200)
+        self.api.get_invoice(1337)
+        self.assertEqual(len(responses.calls), 1)
+        self.assertEqual(responses.calls[0].response.headers['content-type'], 'application/pdf')