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

more tests

Browse Source
This commit is contained in:
Kenneth Kehl
2025-07-02 07:45:22 -07:00
parent 0cef05b220
commit 7743bdd43c
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/app/dao/test_services_dao.py
View File

@@ -1852,11 +1852,13 @@ def test_dao_fetch_stats_for_service_from_hours(mock_execute, mock_get_midnight)
return datetime(dt.year, dt.month, dt.day) return datetime(dt.year, dt.month, dt.day)
mock_get_midnight.side_effect = _mock_midnight mock_get_midnight.side_effect = _mock_midnight
total_result = [ total_result_mock = MagicMock()
total_result_mock.all.return_value = [
MagicMock(hour=datetime(2025, 7, 1, 16), total_notifications=100), MagicMock(hour=datetime(2025, 7, 1, 16), total_notifications=100),
MagicMock(hour=datetime(2025, 7, 1, 17), total_notifications=50), MagicMock(hour=datetime(2025, 7, 1, 17), total_notifications=50),
] ]
detail_result = [ detail_result_mock = MagicMock()
detail_result_mock.all.return_value = [
MagicMock( MagicMock(
notification_type="email", notification_type="email",
status="delivered", status="delivered",
@@ -1870,7 +1872,7 @@ def test_dao_fetch_stats_for_service_from_hours(mock_execute, mock_get_midnight)
count=20, count=20,
), ),
] ]
mock_execute.side_effect = [total_result, detail_result] mock_execute.side_effect = [total_result_mock, detail_result_mock]
total_notifications, data = dao_fetch_stats_for_service_from_hours( total_notifications, data = dao_fetch_stats_for_service_from_hours(
service_id, start, end service_id, start, end
) )