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

billing dao test

Browse Source
This commit is contained in:
Kenneth Kehl
2024-11-11 11:00:14 -08:00
parent 694b631534
commit f564b3a356
1 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
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 (
dao_create_or_update_annual_billing_for_year,
dao_get_annual_billing,
dao_get_free_sms_fragment_limit_for_year,
dao_update_annual_billing_for_future_years,
set_default_free_allowance_for_service,
@@ -122,3 +123,19 @@ def test_set_default_free_allowance_for_service_updates_existing_year(sample_ser
assert len(annual_billing) == 1
assert annual_billing[0].service_id == sample_service.id
assert annual_billing[0].free_sms_fragment_limit == 150000
def test_dao_get_annual_billing(mocker):
mock_db_session = mocker.patch("app.dao.db.session.execute")
mock_db_session.return_value.scalars.return_value.all.return_value = [
"billing_entry1",
"billing_entry2",
]
service_id = "test_service_id"
result = dao_get_annual_billing(service_id)
mock_db_session.assert_called_once()
stmt = mock_db_session.call_args[0][0]
assert stmt.compile().params["service_id"] == service_id
assert result == ["billing_entry1", "billing_entry2"]