mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-05-30 19:10:42 -04:00
971cb745c9
Daily volumes report: total volumes across the platform aggregated by whole business day (bst_date) Volumes by service report: total volumes per service aggregated by the date range given. NB: start and end dates are inclusive
62 lines
2.7 KiB
Python
62 lines
2.7 KiB
Python
import uuid
|
|
|
|
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)
|
|
client = BillingAPIClient()
|
|
|
|
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})
|
|
|
|
|
|
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')
|
|
client = BillingAPIClient()
|
|
|
|
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
|
|
)
|
|
|
|
|
|
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')
|
|
client = BillingAPIClient()
|
|
|
|
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
|
|
)
|
|
|
|
|
|
def test_get_data_for_volumes_by_service_report(mocker, api_user_active):
|
|
mock_get = mocker.patch('app.notify_client.billing_api_client.BillingAPIClient.get')
|
|
client = BillingAPIClient()
|
|
|
|
client.get_data_for_volumes_by_service_report('2022-03-01', '2022-03-31')
|
|
mock_get.assert_called_once_with(url='/platform-stats/volumes-by-service',
|
|
params={'start_date': '2022-03-01', 'end_date': '2022-03-31'})
|
|
|
|
|
|
def test_get_data_for_daily_volumes_report(mocker, api_user_active):
|
|
mock_get = mocker.patch('app.notify_client.billing_api_client.BillingAPIClient.get')
|
|
client = BillingAPIClient()
|
|
|
|
client.get_data_for_daily_volumes_report('2022-03-01', '2022-03-31')
|
|
mock_get.assert_called_once_with(url='/platform-stats/daily-volumes-report',
|
|
params={'start_date': '2022-03-01', 'end_date': '2022-03-31'})
|