diff --git a/app/main/views/add_service.py b/app/main/views/add_service.py index e2e328977..6e816a409 100644 --- a/app/main/views/add_service.py +++ b/app/main/views/add_service.py @@ -20,8 +20,7 @@ from app.notify_client.models import InvitedUser from app import ( invite_api_client, user_api_client, - service_api_client, - billing_api_client + service_api_client ) from app.utils import ( @@ -54,7 +53,8 @@ def _create_service(service_name, organisation_type, email_from, form): ) session['service_id'] = service_id - billing_api_client.create_or_update_free_sms_fragment_limit(service_id, free_sms_fragment_limit) + # TODO: Comment out until data migration + # billing_api_client.create_or_update_free_sms_fragment_limit(service_id, free_sms_fragment_limit) return service_id, None except HTTPError as e: diff --git a/app/main/views/service_settings.py b/app/main/views/service_settings.py index 04006bc14..ad5817fa1 100644 --- a/app/main/views/service_settings.py +++ b/app/main/views/service_settings.py @@ -718,11 +718,12 @@ def set_free_sms_allowance(service_id): if form.validate_on_submit(): service_api_client.update_service( service_id, - # TODO: Retire this after new end points are added. + # TODO: Retire this eventually after using annual_billing free_sms_fragment_limit=form.free_sms_allowance.data, ) - billing_api_client.create_or_update_free_sms_fragment_limit(service_id, form.free_sms_allowance.data) + # TODO: Comment out until data migration + # billing_api_client.create_or_update_free_sms_fragment_limit(service_id, form.free_sms_allowance.data) return redirect(url_for('.service_settings', service_id=service_id)) diff --git a/tests/app/main/views/test_service_settings.py b/tests/app/main/views/test_service_settings.py index 47e0092b9..d2a9d45c8 100644 --- a/tests/app/main/views/test_service_settings.py +++ b/tests/app/main/views/test_service_settings.py @@ -1615,10 +1615,11 @@ def test_should_set_sms_allowance( SERVICE_ONE_ID, free_sms_fragment_limit=expected_api_argument, ) - mock_create_or_update_free_sms_fragment_limit.assert_called_with( - SERVICE_ONE_ID, - expected_api_argument - ) + # Not assert until using the annual_billing end points + # mock_create_or_update_free_sms_fragment_limit.assert_called_with( + # SERVICE_ONE_ID, + # expected_api_argument + # ) def test_switch_service_enable_letters(