diff --git a/app/notify_client/billing_api_client.py b/app/notify_client/billing_api_client.py index 3de5949d9..dace14ec4 100644 --- a/app/notify_client/billing_api_client.py +++ b/app/notify_client/billing_api_client.py @@ -7,26 +7,12 @@ class BillingAPIClient(NotifyAdminAPIClient): def __init__(self): super().__init__("a" * 73, "b") - def get_billable_units(self, service_id, year): - return self.get( - '/service/{0}/billing/monthly-usage'.format(service_id), - params=dict(year=year) - ) - - def get_service_usage(self, service_id, year=None): - return self.get( - '/service/{0}/billing/yearly-usage-summary'.format(service_id), - params=dict(year=year) - ) - - # Temporary methods to compare the usage before and after using ft_billing def get_billable_units_ft(self, service_id, year): return self.get( '/service/{0}/billing/ft-monthly-usage'.format(service_id), params=dict(year=year) ) - # Temporary methods to compare the usage before and after using ft_billing def get_service_usage_ft(self, service_id, year=None): return self.get( '/service/{0}/billing/ft-yearly-usage-summary'.format(service_id), diff --git a/tests/app/notify_client/test_billing_client.py b/tests/app/notify_client/test_billing_client.py index 1ba7a4be0..e29a80bdc 100644 --- a/tests/app/notify_client/test_billing_client.py +++ b/tests/app/notify_client/test_billing_client.py @@ -3,30 +3,6 @@ import uuid from app.notify_client.billing_api_client import BillingAPIClient -def test_get_billing_units_calls_correct_endpoint(mocker, api_user_active): - service_id = uuid.uuid4() - expected_url = '/service/{}/billing/monthly-usage'.format(service_id) - - client = BillingAPIClient() - - mock_get = mocker.patch('app.notify_client.billing_api_client.BillingAPIClient.get') - - client.get_billable_units(service_id, 2017) - mock_get.assert_called_once_with(expected_url, params={'year': 2017}) - - -def test_get_get_service_usage_calls_correct_endpoint(mocker, api_user_active): - service_id = uuid.uuid4() - expected_url = '/service/{}/billing/yearly-usage-summary'.format(service_id) - - client = BillingAPIClient() - - mock_get = mocker.patch('app.notify_client.billing_api_client.BillingAPIClient.get') - - client.get_service_usage(service_id, 2017) - mock_get.assert_called_once_with(expected_url, params={'year': 2017}) - - def test_get_free_sms_fragment_limit_for_year_correct_endpoint(mocker, api_user_active): service_id = uuid.uuid4() expected_url = '/service/{}/billing/free-sms-fragment-limit'.format(service_id)