darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-09-02 17:48:50 -04:00
Code Issues Packages Projects Releases Wiki Activity

notify-api-412 use black to enforce python coding style

Browse Source
This commit is contained in:
Kenneth Kehl
2023-08-25 09:12:23 -07:00
parent c6eb007386
commit 8c9721d8e2
201 changed files with 31660 additions and 28105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

75
tests/app/notify_client/test_billing_client.py
View File

@@ -7,56 +7,77 @@ from app.notify_client.billing_api_client import BillingAPIClient
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)
expected_url = "/service/{}/billing/free-sms-fragment-limit".format(service_id)
client = BillingAPIClient()
mock_get = mocker.patch('app.notify_client.billing_api_client.BillingAPIClient.get')
mock_get = mocker.patch("app.notify_client.billing_api_client.BillingAPIClient.get")
client.get_free_sms_fragment_limit_for_year(service_id, year=1999)
mock_get.assert_called_once_with(expected_url, params={'financial_year_start': 1999})
mock_get.assert_called_once_with(
expected_url, params={"financial_year_start": 1999}
)
def test_post_free_sms_fragment_limit_for_current_year_endpoint(mocker, api_user_active):
def test_post_free_sms_fragment_limit_for_current_year_endpoint(
mocker, api_user_active
):
service_id = uuid.uuid4()
sms_limit_data = {'free_sms_fragment_limit': 1111, 'financial_year_start': None}
mock_post = mocker.patch('app.notify_client.billing_api_client.BillingAPIClient.post')
sms_limit_data = {"free_sms_fragment_limit": 1111, "financial_year_start": None}
mock_post = mocker.patch(
"app.notify_client.billing_api_client.BillingAPIClient.post"
)
client = BillingAPIClient()
client.create_or_update_free_sms_fragment_limit(service_id=service_id, free_sms_fragment_limit=1111)
client.create_or_update_free_sms_fragment_limit(
service_id=service_id, free_sms_fragment_limit=1111
)
mock_post.assert_called_once_with(
url='/service/{}/billing/free-sms-fragment-limit'.format(service_id),
data=sms_limit_data
url="/service/{}/billing/free-sms-fragment-limit".format(service_id),
data=sms_limit_data,
)
def test_post_free_sms_fragment_limit_for_year_endpoint(mocker, api_user_active):
service_id = uuid.uuid4()
sms_limit_data = {'free_sms_fragment_limit': 1111, 'financial_year_start': 2017}
mock_post = mocker.patch('app.notify_client.billing_api_client.BillingAPIClient.post')
sms_limit_data = {"free_sms_fragment_limit": 1111, "financial_year_start": 2017}
mock_post = mocker.patch(
"app.notify_client.billing_api_client.BillingAPIClient.post"
)
client = BillingAPIClient()
client.create_or_update_free_sms_fragment_limit(service_id=service_id,
free_sms_fragment_limit=1111,
year=2017)
client.create_or_update_free_sms_fragment_limit(
service_id=service_id, free_sms_fragment_limit=1111, year=2017
)
mock_post.assert_called_once_with(
url='/service/{}/billing/free-sms-fragment-limit'.format(service_id),
data=sms_limit_data
url="/service/{}/billing/free-sms-fragment-limit".format(service_id),
data=sms_limit_data,
)
@pytest.mark.parametrize('func, expected_url', [
(BillingAPIClient.get_data_for_volumes_by_service_report, '/platform-stats/volumes-by-service'),
(BillingAPIClient.get_data_for_daily_volumes_report, '/platform-stats/daily-volumes-report'),
(
BillingAPIClient.get_data_for_daily_sms_provider_volumes_report,
'/platform-stats/daily-sms-provider-volumes-report'
),
])
@pytest.mark.parametrize(
"func, expected_url",
[
(
BillingAPIClient.get_data_for_volumes_by_service_report,
"/platform-stats/volumes-by-service",
),
(
BillingAPIClient.get_data_for_daily_volumes_report,
"/platform-stats/daily-volumes-report",
),
(
BillingAPIClient.get_data_for_daily_sms_provider_volumes_report,
"/platform-stats/daily-sms-provider-volumes-report",
),
],
)
def test_get_data_for_volume_reports(mocker, api_user_active, func, expected_url):
mock_get = mocker.patch('app.notify_client.billing_api_client.BillingAPIClient.get')
mock_get = mocker.patch("app.notify_client.billing_api_client.BillingAPIClient.get")
client = BillingAPIClient()
func(client, '2022-03-01', '2022-03-31')
func(client, "2022-03-01", "2022-03-31")
mock_get.assert_called_once_with(url=expected_url, params={'start_date': '2022-03-01', 'end_date': '2022-03-31'})
mock_get.assert_called_once_with(
url=expected_url, params={"start_date": "2022-03-01", "end_date": "2022-03-31"}
)