add tests

This commit is contained in:
Kenneth Kehl
2025-08-11 12:09:53 -07:00
parent 4a31f34117
commit a65d65a097

View File

@@ -1796,8 +1796,7 @@ def test_save_api_email_or_sms_sqlalchemy_error_with_max_retries():
assert "Max retry failed" in mock_exception.call_args[0][0] assert "Max retry failed" in mock_exception.call_args[0][0]
@pytest.fixture def get_mock_notification():
def mock_notification():
notif = MagicMock() notif = MagicMock()
notif.job_id = "job-id" notif.job_id = "job-id"
notif.service_id = "service-id" notif.service_id = "service-id"
@@ -1812,6 +1811,7 @@ def mock_notification():
"job_name": "Job A", "job_name": "Job A",
"provider_response": "Success", "provider_response": "Success",
} }
return notif
@patch("app.dao.notifications_dao.get_notifications_for_service") @patch("app.dao.notifications_dao.get_notifications_for_service")
@@ -1829,10 +1829,9 @@ def test_generate_notifications_report_normal_case(
mock_get_phone_number, mock_get_phone_number,
mock_get_personalisation, mock_get_personalisation,
mock_get_notifications, mock_get_notifications,
mock_notification,
): ):
mock_get_notifications.return_value.items = [mock_notification] mock_get_notifications.return_value.items = [get_mock_notification()]
mock_get_phone_number.return_value = "1234567890" mock_get_phone_number.return_value = "1234567890"
mock_get_personalisation.return_value = {"name": "John"} mock_get_personalisation.return_value = {"name": "John"}
mock_get_csv_location.return_value = ( mock_get_csv_location.return_value = (