darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 01:41:05 -05:00
Code Issues Packages Projects Releases Wiki Activity

reformat

Browse Source
This commit is contained in:
Kenneth Kehl
2023-08-29 14:54:30 -07:00
parent 19dcd7a48b
commit 1ecb747c6d
588 changed files with 34100 additions and 23589 deletions
Download Patch File Download Diff File Expand all files Collapse all files

182
tests/app/billing/test_rest.py
View File

@@ -20,130 +20,160 @@ IN_MAY_2016 = datetime(2016, 5, 10, 23, 00, 00)
IN_JUN_2016 = datetime(2016, 6, 3, 23, 00, 00)
def test_create_update_free_sms_fragment_limit_invalid_schema(admin_request, sample_service):
def test_create_update_free_sms_fragment_limit_invalid_schema(
admin_request, sample_service
):
json_response = admin_request.post(
'billing.create_or_update_free_sms_fragment_limit',
"billing.create_or_update_free_sms_fragment_limit",
service_id=sample_service.id,
_data={},
_expected_status=400
_expected_status=400,
)
assert 'errors' in json_response
assert "errors" in json_response
def test_create_free_sms_fragment_limit_current_year_updates_future_years(admin_request, sample_service):
def test_create_free_sms_fragment_limit_current_year_updates_future_years(
admin_request, sample_service
):
current_year = get_current_calendar_year_start_year()
future_billing = create_annual_billing(sample_service.id, 1, current_year + 1)
admin_request.post(
'billing.create_or_update_free_sms_fragment_limit',
"billing.create_or_update_free_sms_fragment_limit",
service_id=sample_service.id,
_data={'free_sms_fragment_limit': 9999},
_expected_status=201
_data={"free_sms_fragment_limit": 9999},
_expected_status=201,
)
current_billing = dao_get_free_sms_fragment_limit_for_year(sample_service.id, current_year)
current_billing = dao_get_free_sms_fragment_limit_for_year(
sample_service.id, current_year
)
assert future_billing.free_sms_fragment_limit == 9999
assert current_billing.financial_year_start == current_year
assert current_billing.free_sms_fragment_limit == 9999
@pytest.mark.parametrize('update_existing', [True, False])
@pytest.mark.parametrize("update_existing", [True, False])
def test_create_or_update_free_sms_fragment_limit_past_year_doenst_update_other_years(
admin_request,
sample_service,
update_existing
admin_request, sample_service, update_existing
):
current_year = get_current_calendar_year_start_year()
create_annual_billing(sample_service.id, 1, current_year)
if update_existing:
create_annual_billing(sample_service.id, 1, current_year - 1)
data = {'financial_year_start': current_year - 1, 'free_sms_fragment_limit': 9999}
data = {"financial_year_start": current_year - 1, "free_sms_fragment_limit": 9999}
admin_request.post(
'billing.create_or_update_free_sms_fragment_limit',
"billing.create_or_update_free_sms_fragment_limit",
service_id=sample_service.id,
_data=data,
_expected_status=201)
_expected_status=201,
)
assert dao_get_free_sms_fragment_limit_for_year(sample_service.id, current_year - 1).free_sms_fragment_limit == 9999
assert dao_get_free_sms_fragment_limit_for_year(sample_service.id, current_year).free_sms_fragment_limit == 1
assert (
dao_get_free_sms_fragment_limit_for_year(
sample_service.id, current_year - 1
).free_sms_fragment_limit
== 9999
)
assert (
dao_get_free_sms_fragment_limit_for_year(
sample_service.id, current_year
).free_sms_fragment_limit
== 1
)
def test_create_free_sms_fragment_limit_updates_existing_year(admin_request, sample_service):
def test_create_free_sms_fragment_limit_updates_existing_year(
admin_request, sample_service
):
current_year = get_current_calendar_year_start_year()
annual_billing = create_annual_billing(sample_service.id, 1, current_year)
admin_request.post(
'billing.create_or_update_free_sms_fragment_limit',
"billing.create_or_update_free_sms_fragment_limit",
service_id=sample_service.id,
_data={'financial_year_start': current_year, 'free_sms_fragment_limit': 2},
_expected_status=201)
_data={"financial_year_start": current_year, "free_sms_fragment_limit": 2},
_expected_status=201,
)
assert annual_billing.free_sms_fragment_limit == 2
@freeze_time('2021-04-02 13:00')
def test_get_free_sms_fragment_limit(
admin_request, sample_service
):
create_annual_billing(service_id=sample_service.id, free_sms_fragment_limit=11000, financial_year_start=2021)
json_response = admin_request.get(
'billing.get_free_sms_fragment_limit',
service_id=sample_service.id
@freeze_time("2021-04-02 13:00")
def test_get_free_sms_fragment_limit(admin_request, sample_service):
create_annual_billing(
service_id=sample_service.id,
free_sms_fragment_limit=11000,
financial_year_start=2021,
)
assert json_response['financial_year_start'] == 2021
assert json_response['free_sms_fragment_limit'] == 11000
json_response = admin_request.get(
"billing.get_free_sms_fragment_limit", service_id=sample_service.id
)
assert json_response["financial_year_start"] == 2021
assert json_response["free_sms_fragment_limit"] == 11000
@freeze_time('2021-04-02 13:00')
@freeze_time("2021-04-02 13:00")
def test_get_free_sms_fragment_limit_current_year_creates_new_row_if_annual_billing_is_missing(
admin_request, sample_service
):
json_response = admin_request.get(
'billing.get_free_sms_fragment_limit',
service_id=sample_service.id
"billing.get_free_sms_fragment_limit", service_id=sample_service.id
)
assert json_response['financial_year_start'] == 2021
assert json_response['free_sms_fragment_limit'] == 150000 # based on other organization type
assert json_response["financial_year_start"] == 2021
assert (
json_response["free_sms_fragment_limit"] == 150000
) # based on other organization type
def test_update_free_sms_fragment_limit_data(client, sample_service):
current_year = get_current_calendar_year_start_year()
create_annual_billing(sample_service.id, free_sms_fragment_limit=250000, financial_year_start=current_year - 1)
create_annual_billing(
sample_service.id,
free_sms_fragment_limit=250000,
financial_year_start=current_year - 1,
)
update_free_sms_fragment_limit_data(sample_service.id, 9999, current_year)
annual_billing = dao_get_free_sms_fragment_limit_for_year(sample_service.id, current_year)
annual_billing = dao_get_free_sms_fragment_limit_for_year(
sample_service.id, current_year
)
assert annual_billing.free_sms_fragment_limit == 9999
def test_get_yearly_usage_by_monthly_from_ft_billing(admin_request, notify_db_session):
service = create_service()
create_annual_billing(service_id=service.id, free_sms_fragment_limit=1, financial_year_start=2016)
create_annual_billing(
service_id=service.id, free_sms_fragment_limit=1, financial_year_start=2016
)
sms_template = create_template(service=service, template_type="sms")
email_template = create_template(service=service, template_type="email")
for dt in (date(2016, 1, 28), date(2016, 8, 10), date(2016, 12, 26)):
create_ft_billing(local_date=dt, template=sms_template, rate=0.0162)
create_ft_billing(local_date=dt, template=email_template, billable_unit=0, rate=0)
create_ft_billing(
local_date=dt, template=email_template, billable_unit=0, rate=0
)
json_response = admin_request.get(
'billing.get_yearly_usage_by_monthly_from_ft_billing',
"billing.get_yearly_usage_by_monthly_from_ft_billing",
service_id=service.id,
year=2016
year=2016,
)
assert len(json_response) == 3 # 3 billed months for SMS
email_rows = [row for row in json_response if row['notification_type'] == 'email']
email_rows = [row for row in json_response if row["notification_type"] == "email"]
assert len(email_rows) == 0
sms_row = next(x for x in json_response if x['notification_type'] == 'sms')
sms_row = next(x for x in json_response if x["notification_type"] == "sms")
assert sms_row["month"] == "January"
assert sms_row["notification_type"] == "sms"
@@ -156,59 +186,65 @@ def test_get_yearly_usage_by_monthly_from_ft_billing(admin_request, notify_db_se
assert sms_row["charged_units"] == 0
def test_get_yearly_billing_usage_summary_from_ft_billing_returns_400_if_missing_year(admin_request, sample_service):
def test_get_yearly_billing_usage_summary_from_ft_billing_returns_400_if_missing_year(
admin_request, sample_service
):
json_response = admin_request.get(
'billing.get_yearly_billing_usage_summary_from_ft_billing',
"billing.get_yearly_billing_usage_summary_from_ft_billing",
service_id=sample_service.id,
_expected_status=400
_expected_status=400,
)
assert json_response == {
'message': 'No valid year provided', 'result': 'error'
}
assert json_response == {"message": "No valid year provided", "result": "error"}
def test_get_yearly_billing_usage_summary_from_ft_billing_returns_empty_list_if_no_billing_data(
admin_request, sample_service
):
json_response = admin_request.get(
'billing.get_yearly_billing_usage_summary_from_ft_billing',
"billing.get_yearly_billing_usage_summary_from_ft_billing",
service_id=sample_service.id,
year=2016
year=2016,
)
assert json_response == []
def test_get_yearly_billing_usage_summary_from_ft_billing(admin_request, notify_db_session):
def test_get_yearly_billing_usage_summary_from_ft_billing(
admin_request, notify_db_session
):
service = create_service()
create_annual_billing(service_id=service.id, free_sms_fragment_limit=1, financial_year_start=2016)
create_annual_billing(
service_id=service.id, free_sms_fragment_limit=1, financial_year_start=2016
)
sms_template = create_template(service=service, template_type="sms")
email_template = create_template(service=service, template_type="email")
for dt in (date(2016, 1, 28), date(2016, 8, 10), date(2016, 12, 26)):
create_ft_billing(local_date=dt, template=sms_template, rate=0.0162)
create_ft_billing(local_date=dt, template=email_template, billable_unit=0, rate=0)
create_ft_billing(
local_date=dt, template=email_template, billable_unit=0, rate=0
)
json_response = admin_request.get(
'billing.get_yearly_billing_usage_summary_from_ft_billing',
"billing.get_yearly_billing_usage_summary_from_ft_billing",
service_id=service.id,
year=2016
year=2016,
)
assert len(json_response) == 2
assert json_response[0]['notification_type'] == 'email'
assert json_response[0]['chargeable_units'] == 0
assert json_response[0]['notifications_sent'] == 3
assert json_response[0]['rate'] == 0
assert json_response[0]['cost'] == 0
assert json_response[0]['free_allowance_used'] == 0
assert json_response[0]['charged_units'] == 0
assert json_response[0]["notification_type"] == "email"
assert json_response[0]["chargeable_units"] == 0
assert json_response[0]["notifications_sent"] == 3
assert json_response[0]["rate"] == 0
assert json_response[0]["cost"] == 0
assert json_response[0]["free_allowance_used"] == 0
assert json_response[0]["charged_units"] == 0
assert json_response[1]['notification_type'] == 'sms'
assert json_response[1]['chargeable_units'] == 3
assert json_response[1]['notifications_sent'] == 3
assert json_response[1]['rate'] == 0.0162
assert json_response[1]['cost'] == 0.0324
assert json_response[1]['free_allowance_used'] == 1
assert json_response[1]['charged_units'] == 2
assert json_response[1]["notification_type"] == "sms"
assert json_response[1]["chargeable_units"] == 3
assert json_response[1]["notifications_sent"] == 3
assert json_response[1]["rate"] == 0.0162
assert json_response[1]["cost"] == 0.0324
assert json_response[1]["free_allowance_used"] == 1
assert json_response[1]["charged_units"] == 2