diff --git a/api/tests.py b/api/tests.py index e7cb105768785b577fc4603c493a4adf5b9d67e2..749c86e4a8248403c030d0d3424e65cfe2e05ea4 100644 --- a/api/tests.py +++ b/api/tests.py @@ -409,37 +409,31 @@ class DataSourceApiIoTUKTest(TestCase): self.assertLessEqual(1, len(data['data'])) -class DataSourceApiBTHyperCatTest(TestCase): +class DataSourceApiHyperCatTest(TestCase): + test_name = 'HyperCat' + plugin_name = 'HyperCat' + test_url = 'https://api.cityverve.org.uk/v1/cat/polling-station' + dataset = 'https://api.cityverve.org.uk/v1/entity/polling-station/5' + @classmethod def setUpTestData(cls): - cls.user = get_user_model().objects.create_user('Test API User') - - def setUp(self): from decouple import config - self.client = APIClient() - self.client.force_authenticate(self.user) - - self.test_name = 'BT HyperCat' - self.test_url = 'https://portal.bt-hypercat.com/cat' - self.api_key = config('HYPERCAT_BT_API_KEY') - self.plugin_name = 'HyperCat' + cls.user = get_user_model().objects.create_user('Test API User') - self.dataset = 'http://api.bt-hypercat.com/sensors/feeds/c7f361c6-7cb7-4ef5-aed9-397a0c0c4088' + cls.api_key = config('HYPERCAT_CISCO_API_KEY') - self.model = models.DataSource.objects.create( - name=self.test_name, - owner=self.user, - url=self.test_url, - api_key=self.api_key, - plugin_name=self.plugin_name + cls.model = models.DataSource.objects.create( + name=cls.test_name, + owner=cls.user, + url=cls.test_url, + api_key=cls.api_key, + plugin_name=cls.plugin_name ) - def tearDown(self): - try: - self.model.delete() - except AttributeError: - pass + def setUp(self): + self.client = APIClient() + self.client.force_authenticate(self.user) def test_api_datasource_get(self): """ @@ -511,8 +505,8 @@ class DataSourceApiBTHyperCatTest(TestCase): response = self.client.get('/api/datasources/{}/datasets/{}/data/'.format(self.model.pk, self.dataset)) self.assertEqual(response.status_code, 200) - self.assertEqual('text/xml', response['Content-Type']) - self.assertTrue(response.content) + data = response.json() + self.assertTrue(data) # TODO test content diff --git a/datasources/tests/test_connectors_hypercat.py b/datasources/tests/test_connectors_hypercat.py index 7abc2c5fd81623ee7614e7762f4755fef580b4b7..a34bff6a05a8caaa7aeb4464f9f0cea4132332c7 100644 --- a/datasources/tests/test_connectors_hypercat.py +++ b/datasources/tests/test_connectors_hypercat.py @@ -59,7 +59,6 @@ class ConnectorHyperCatTest(TestCase): self.assertTrue(connection.is_catalogue) def test_plugin_get_metadata(self): - connection = self.plugin(self.url, auth=HTTPBasicAuth) connection = self._get_connection() result = connection.get_metadata() @@ -94,7 +93,6 @@ class ConnectorHyperCatTest(TestCase): def test_plugin_iter_datasets(self): connection = self._get_connection() - for dataset in connection: self.assertEqual(str, type(dataset)) @@ -171,7 +169,6 @@ class ConnectorHyperCatTest(TestCase): """ Test that we can get data from a single dataset within the catalogue. """ - connection = self._get_connection() dataset = connection[self.dataset]