diff --git a/app/notify_client/complaint_api_client.py b/app/notify_client/complaint_api_client.py index 6e657396c..72760d0b7 100644 --- a/app/notify_client/complaint_api_client.py +++ b/app/notify_client/complaint_api_client.py @@ -9,3 +9,6 @@ class ComplaintApiClient(NotifyAdminAPIClient): def get_all_complaints(self): return self.get('/complaint') + + def get_complaint_count(self, params_dict=None): + return self.get('/complaint/count-by-date-range', params=params_dict) diff --git a/app/notify_client/service_api_client.py b/app/notify_client/service_api_client.py index af8a2481a..9d5b59ccb 100644 --- a/app/notify_client/service_api_client.py +++ b/app/notify_client/service_api_client.py @@ -409,6 +409,9 @@ class ServiceAPIClient(NotifyAdminAPIClient): def get_aggregate_platform_stats(self, params_dict=None): return self.get("/service/platform-stats", params=params_dict) + def get_new_aggregate_platform_stats(self, params_dict=None): + return self.get("/service/platform-stats-new", params=params_dict) + def get_sms_senders(self, service_id): return self.get( "/service/{}/sms-sender".format(service_id) diff --git a/tests/app/notify_client/test_compliant_client.py b/tests/app/notify_client/test_compliant_client.py index 9d05d1490..2665a14a7 100644 --- a/tests/app/notify_client/test_compliant_client.py +++ b/tests/app/notify_client/test_compliant_client.py @@ -8,3 +8,12 @@ def test_get_all_complaints(mocker): client.get_all_complaints() mock.assert_called_once_with('/complaint') + + +def test_get_complaint_count(mocker): + client = ComplaintApiClient() + mock = mocker.patch.object(client, 'get') + params_dict = {'start_date': '2018-06-01', 'end_date': '2018-06-15'} + + client.get_complaint_count(params_dict=params_dict) + mock.assert_called_once_with('/complaint/count-by-date-range', params=params_dict) diff --git a/tests/app/notify_client/test_service_api_client.py b/tests/app/notify_client/test_service_api_client.py index 7bc4afd4f..61d78e766 100644 --- a/tests/app/notify_client/test_service_api_client.py +++ b/tests/app/notify_client/test_service_api_client.py @@ -53,6 +53,15 @@ def test_client_only_updates_allowed_attributes(mocker): assert str(error.value) == 'Not allowed to update service attributes: foo' +def test_get_new_aggregate_platform_stats(mocker): + client = ServiceAPIClient() + mock = mocker.patch.object(client, 'get') + params_dict = {'start_date': '2018-06-01', 'end_date': '2018-06-15'} + + client.get_new_aggregate_platform_stats(params_dict=params_dict) + mock.assert_called_once_with('/service/platform-stats-new', params=params_dict) + + def test_client_creates_service_with_correct_data( mocker, active_user_with_permissions,