From 5321f5bfd239256b288b06908b5444f56e52ba0c Mon Sep 17 00:00:00 2001 From: vicaya Date: Wed, 20 Aug 2025 05:55:03 +0000 Subject: [PATCH] Fix broken unit tests Resolve #1780 --- backend/app/tests/scripts/test_backend_pre_start.py | 12 ++++++------ backend/app/tests/scripts/test_test_pre_start.py | 12 ++++++------ 2 files changed, 12 insertions(+), 12 deletions(-) diff --git a/backend/app/tests/scripts/test_backend_pre_start.py b/backend/app/tests/scripts/test_backend_pre_start.py index 631690fcf6..064fca9880 100644 --- a/backend/app/tests/scripts/test_backend_pre_start.py +++ b/backend/app/tests/scripts/test_backend_pre_start.py @@ -9,11 +9,13 @@ def test_init_successful_connection() -> None: engine_mock = MagicMock() session_mock = MagicMock() - exec_mock = MagicMock(return_value=True) - session_mock.configure_mock(**{"exec.return_value": exec_mock}) + session_mock.__enter__.return_value = session_mock + + select1 = select(1) with ( - patch("sqlmodel.Session", return_value=session_mock), + patch("app.backend_pre_start.Session", return_value=session_mock), + patch("app.backend_pre_start.select", return_value=select1), patch.object(logger, "info"), patch.object(logger, "error"), patch.object(logger, "warn"), @@ -28,6 +30,4 @@ def test_init_successful_connection() -> None: connection_successful ), "The database connection should be successful and not raise an exception." - assert session_mock.exec.called_once_with( - select(1) - ), "The session should execute a select statement once." + session_mock.exec.assert_called_once_with(select1) diff --git a/backend/app/tests/scripts/test_test_pre_start.py b/backend/app/tests/scripts/test_test_pre_start.py index a176f380de..a4947fdfbf 100644 --- a/backend/app/tests/scripts/test_test_pre_start.py +++ b/backend/app/tests/scripts/test_test_pre_start.py @@ -9,11 +9,13 @@ def test_init_successful_connection() -> None: engine_mock = MagicMock() session_mock = MagicMock() - exec_mock = MagicMock(return_value=True) - session_mock.configure_mock(**{"exec.return_value": exec_mock}) + session_mock.__enter__.return_value = session_mock + + select1 = select(1) with ( - patch("sqlmodel.Session", return_value=session_mock), + patch("app.tests_pre_start.Session", return_value=session_mock), + patch("app.tests_pre_start.select", return_value=select1), patch.object(logger, "info"), patch.object(logger, "error"), patch.object(logger, "warn"), @@ -28,6 +30,4 @@ def test_init_successful_connection() -> None: connection_successful ), "The database connection should be successful and not raise an exception." - assert session_mock.exec.called_once_with( - select(1) - ), "The session should execute a select statement once." + session_mock.exec.assert_called_once_with(select1)