darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-08-23 15:56:45 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix more

Browse Source
This commit is contained in:
Kenneth Kehl
2024-11-14 13:15:06 -08:00
parent 6d8fdab5a3
commit 7b83ea3a34
3 changed files with 63 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
tests/app/test_commands.py
View File

@@ -177,8 +177,7 @@ def test_populate_organization_agreement_details_from_file(
org_count = _get_organization_query_count()
assert org_count == 1
org = Organization.query.one()
org.agreement_signed = True
org = _get_organization_query_one()
notify_db_session.commit()
text = (
@@ -195,11 +194,16 @@ def test_populate_organization_agreement_details_from_file(
org_count = _get_organization_query_count()
assert org_count == 1
org = Organization.query.one()
org = _get_organization_query_one()
assert org.agreement_signed_on_behalf_of_name == "bob"
os.remove(file_name)
def _get_organization_query_one():
stmt = select(Organization)
return db.session.execute(stmt).scalars().one()
def test_bulk_invite_user_to_service(
notify_db_session, notify_api, sample_service, sample_user
):
@@ -344,9 +348,14 @@ def test_populate_annual_billing_with_the_previous_years_allowance(
assert results[0].free_sms_fragment_limit == expected_allowance
def _get_notification_query_one():
stmt = select(Notification)
return db.session.execute(stmt).scalars().one()
def test_fix_billable_units(notify_db_session, notify_api, sample_template):
create_notification(template=sample_template)
notification = Notification.query.one()
notification = _get_notification_query_one()
notification.billable_units = 0
notification.notification_type = NotificationType.SMS
notification.status = NotificationStatus.DELIVERED
@@ -357,7 +366,7 @@ def test_fix_billable_units(notify_db_session, notify_api, sample_template):
notify_api.test_cli_runner().invoke(fix_billable_units, [])
notification = Notification.query.one()
notification = _get_notification_query_one()
assert notification.billable_units == 1