Merge pull request #9 from researchnow/POP-2274

Get Attributes

Author: DynataRidley

dynatademand/api.py

@@ -134,6 +134,9 @@ def logout(self):
             ))
         return logout_response.json()
 
+    def get_attributes(self, country_code, language_code):
+        return self._api_get('/attributes/{}/{}'.format(country_code, language_code))
+
     def get_countries(self):
         return self._api_get('/countries')
 

tests/test_attributes.py

@@ -0,0 +1,32 @@
+# 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 TestAttributeEndpoints(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_attributes(self):
+        with open('./tests/test_files/get_attributes.json', 'r') as attributes_file:
+            attributes_json = json.load(attributes_file)
+        responses.add(responses.GET, '{}/sample/v1/attributes/no/no'.format(BASE_HOST), json=attributes_json, status=200)
+        self.api.get_attributes('no', 'no')
+        self.assertEqual(len(responses.calls), 1)
+        print('flaws')
+        print(responses.calls[0].response.json())
+        self.assertEqual(responses.calls[0].response.json(), attributes_json)

tests/test_files/get_attributes.json

@@ -0,0 +1,87 @@
+{
+  "data": [
+    {
+      "category": {
+        "mainCategory": {
+          "id": "DEMOGRAPHIC",
+          "localizedText": "Demographic",
+          "text": "Demographic"
+        }
+      },
+      "id": "11",
+      "isAllowedInFilters": true,
+      "isAllowedInQuotas": true,
+      "localizedText": "What is your gender?",
+      "name": "Gender",
+      "options": [
+        {
+          "id": "1",
+          "localizedText": "Male",
+          "text": "Male"
+        },
+        {
+          "id": "2",
+          "localizedText": "Female",
+          "text": "Female"
+        }
+      ],
+      "state": "ACTIVE",
+      "text": "What is your gender?",
+      "tier": "Standard",
+      "type": "LIST"
+    },
+    {
+      "category": {
+        "mainCategory": {
+          "id": "DEMOGRAPHIC",
+          "localizedText": "Demographic",
+          "text": "Demographic"
+        }
+      },
+      "format": "^(1[8]-1[9]|1[89]-[2-9][0-9]|[2-9][0-9]-[2-9][0-9])$",
+      "id": "13",
+      "isAllowedInFilters": true,
+      "isAllowedInQuotas": true,
+      "localizedText": "No question; generated by the database",
+      "name": "Age",
+      "state": "ACTIVE",
+      "text": "No question; generated by the database",
+      "tier": "Standard",
+      "type": "INTEGER_RANGE"
+    },
+    {
+      "category": {
+        "mainCategory": {
+          "id": "DEMOGRAPHIC",
+          "localizedText": "Demographic",
+          "text": "Demographic"
+        }
+      },
+      "format": "^\\d{10}$",
+      "id": "1500",
+      "isAllowedInFilters": true,
+      "isAllowedInQuotas": false,
+      "localizedText": "Respondent Identifier",
+      "name": "Respondent Identifier",
+      "state": "ACTIVE",
+      "text": "Respondent Identifier",
+      "tier": "Standard",
+      "type": "INTEGER"
+    }
+  ],
+  "meta": {
+    "links": {
+      "first": "https://api.researchnow.com/sample/v1/attributes/US/en?offset=0",
+      "last": "https://api.researchnow.com/sample/v1/attributes/US/en?offset=173",
+      "next": "https://api.researchnow.com/sample/v1/attributes/US/en?offset=10",
+      "prev": "https://api.researchnow.com/sample/v1/attributes/US/en?offset=0",
+      "self": "https://api.researchnow.com/sample/v1/attributes/US/en"
+    },
+    "pageSize": 10,
+    "total": 183
+  },
+  "status": {
+    "errors": [],
+    "message": "success"
+  }
+}