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

fix

Browse Source
This commit is contained in:
Kenneth Kehl
2024-11-11 11:42:15 -08:00
parent 266a5b198b
commit 2f9dcc54ec
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/app/dao/test_annual_billing_dao.py
View File

@@ -3,6 +3,7 @@ from freezegun import freeze_time
from app.dao.annual_billing_dao import ( from app.dao.annual_billing_dao import (
dao_create_or_update_annual_billing_for_year, dao_create_or_update_annual_billing_for_year,
dao_get_all_free_sms_fragment_limit,
dao_get_annual_billing, dao_get_annual_billing,
dao_get_free_sms_fragment_limit_for_year, dao_get_free_sms_fragment_limit_for_year,
dao_update_annual_billing_for_future_years, dao_update_annual_billing_for_future_years,
@@ -140,9 +141,13 @@ def test_dao_get_annual_billing(mocker):
assert result == ["billing_entry1", "billing_entry2"] assert result == ["billing_entry1", "billing_entry2"]
def test_dao_get_all_free_sms_fragment_limit(mocker): def test_dao_get_all_free_sms_fragment_limit(mocker):
mock_db_session = mocker.patch("app.dao.db.session.execute") mock_db_session = mocker.patch("app.dao.db.session.execute")
mock_db_session.return_value.scalars.return_value.all.return_value = ["sms_limit1", "sms_limit2"] mock_db_session.return_value.scalars.return_value.all.return_value = [
"sms_limit1",
"sms_limit2",
]
service_id = "test_service_id" service_id = "test_service_id"