darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-09-10 10:03:38 -04:00
Code Issues Packages Projects Releases Wiki Activity

billing dao test

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

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

@@ -136,8 +136,21 @@ def test_dao_get_annual_billing(mocker):
result = dao_get_annual_billing(service_id) result = dao_get_annual_billing(service_id)
mock_db_session.assert_called_once() mock_db_session.assert_called_once()
stmt = mock_db_session.call_args[0][0] stmt = mock_db_session.call_args[0][0]
print(f"stmt = {stmt}")
print(f"params = {stmt.compile().params}")
assert stmt.compile().params["service_id_1"] == service_id assert stmt.compile().params["service_id_1"] == service_id
assert result == ["billing_entry1", "billing_entry2"] assert result == ["billing_entry1", "billing_entry2"]
def test_dao_get_all_free_sms_fragment_limit(mocker):
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"]
service_id = "test_service_id"
result = dao_get_all_free_sms_fragment_limit(service_id)
mock_db_session.assert_called_once()
stmt = mock_db_session.call_arg[0][0]
print(f"params = {stmt.compile().params}")
assert stmt.compile().params["service_id"] == service_id
assert result == ["sms_limit1", "sms_limit2"]