def test_get_dataset_activity(self, mock_get): with app.app_context() as ac: ac.g.trace_id = None ac.g.requests = requests.Session() with app.test_request_context(): dataset_activity = [{ "download_history": [], "id": 356, "licence_agreed": False, "name": "ccod", "private": False, "title": "UK companies that own property in England and Wales" }] mock_get.return_value.json.return_value = dataset_activity mock_get.return_value.status_code = 200 response = UlapdAPI.get_dataset_activity(self, 1) mock_get.assert_called_once() mock_get.assert_called_with( '{0}/users/dataset-activity/{1}'.format(self.url, 1), headers=self.headers, timeout=self.timeout) self.assertEqual(response, dataset_activity)
def test_get_dataset_activity_connection_error(self, mock_get): with app.app_context() as ac: ac.g.trace_id = None ac.g.requests = requests.Session() with app.test_request_context(): mock_get.side_effect = ConnectionError(self.error_msg) with self.assertRaises(ApplicationError) as context: UlapdAPI.get_dataset_activity(self, 1) expected_err = ('verification_api', 'ULAPD_API_CONN_ERROR') expected_err_message = errors.get_message( *expected_err, filler=self.error_msg) expected_err_code = errors.get_code(*expected_err) mock_get.assert_called_once() self.assertEqual(context.exception.message, expected_err_message) self.assertEqual(context.exception.code, expected_err_code)