more tests

This commit is contained in:
Kenneth Kehl
2025-07-02 07:30:35 -07:00
parent b729baa15d
commit 0cef05b220
2 changed files with 55 additions and 3 deletions

View File

@@ -229,7 +229,7 @@
"filename": "tests/app/dao/test_services_dao.py",
"hashed_secret": "5baa61e4c9b93f3f0682250b6cf8331b7ee68fd8",
"is_verified": false,
"line_number": 289,
"line_number": 290,
"is_secret": false
}
],
@@ -374,5 +374,5 @@
}
]
},
"generated_at": "2025-07-01T16:40:56Z"
"generated_at": "2025-07-02T14:30:24Z"
}

View File

@@ -1,7 +1,7 @@
import uuid
from datetime import datetime, timedelta
from unittest import mock
from unittest.mock import Mock
from unittest.mock import MagicMock, Mock, patch
import pytest
import sqlalchemy
@@ -29,6 +29,7 @@ from app.dao.services_dao import (
dao_fetch_service_by_id,
dao_fetch_service_by_id_with_api_keys,
dao_fetch_service_by_inbound_number,
dao_fetch_stats_for_service_from_hours,
dao_fetch_todays_stats_for_all_services,
dao_fetch_todays_stats_for_service,
dao_find_services_sending_to_tv_numbers,
@@ -1838,3 +1839,54 @@ def test_get_specific_days(data, start_date, days, end_date, expected, is_error)
total_notifications=total_notifications,
)
assert results == expected
@patch("app.dao.services_dao.get_midnight_in_utc")
@patch("app.dao.services_dao.db.session.execute")
def test_dao_fetch_stats_for_service_from_hours(mock_execute, mock_get_midnight):
service_id = "service-xyz"
start = datetime(2025, 7, 1, 15, 30)
end = datetime(2025, 7, 1, 18, 45)
def _mock_midnight(dt):
return datetime(dt.year, dt.month, dt.day)
mock_get_midnight.side_effect = _mock_midnight
total_result = [
MagicMock(hour=datetime(2025, 7, 1, 16), total_notifications=100),
MagicMock(hour=datetime(2025, 7, 1, 17), total_notifications=50),
]
detail_result = [
MagicMock(
notification_type="email",
status="delivered",
hour=datetime(2025, 7, 1, 16),
count=60,
),
MagicMock(
notification_type="sms",
status="failed",
hour=datetime(2025, 7, 1, 17),
count=20,
),
]
mock_execute.side_effect = [total_result, detail_result]
total_notifications, data = dao_fetch_stats_for_service_from_hours(
service_id, start, end
)
assert total_notifications == {
datetime(2025, 7, 1, 16): 100,
datetime(2025, 7, 1, 17): 50,
}
assert len(data) == 2
assert data[0].notification_type == "email"
assert data[0].status == "delivered"
assert data[0].hour == datetime(2025, 7, 1, 16)
assert data[0].count == 60
assert data[1].notification_type == "sms"
assert data[1].status == "failed"
assert data[1].hour == datetime(2025, 7, 1, 17)
assert data[1].count == 20
assert mock_execute.call_count == 2