Co-authored-by: Yurii Motov <yurii.motov.monte@gmail.com>
This commit is contained in:
@@ -9,11 +9,13 @@ def test_init_successful_connection() -> None:
|
|||||||
engine_mock = MagicMock()
|
engine_mock = MagicMock()
|
||||||
|
|
||||||
session_mock = MagicMock()
|
session_mock = MagicMock()
|
||||||
exec_mock = MagicMock(return_value=True)
|
session_mock.__enter__.return_value = session_mock
|
||||||
session_mock.configure_mock(**{"exec.return_value": exec_mock})
|
|
||||||
|
select1 = select(1)
|
||||||
|
|
||||||
with (
|
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, "info"),
|
||||||
patch.object(logger, "error"),
|
patch.object(logger, "error"),
|
||||||
patch.object(logger, "warn"),
|
patch.object(logger, "warn"),
|
||||||
@@ -28,6 +30,4 @@ def test_init_successful_connection() -> None:
|
|||||||
connection_successful
|
connection_successful
|
||||||
), "The database connection should be successful and not raise an exception."
|
), "The database connection should be successful and not raise an exception."
|
||||||
|
|
||||||
assert session_mock.exec.called_once_with(
|
session_mock.exec.assert_called_once_with(select1)
|
||||||
select(1)
|
|
||||||
), "The session should execute a select statement once."
|
|
||||||
|
|||||||
@@ -9,11 +9,13 @@ def test_init_successful_connection() -> None:
|
|||||||
engine_mock = MagicMock()
|
engine_mock = MagicMock()
|
||||||
|
|
||||||
session_mock = MagicMock()
|
session_mock = MagicMock()
|
||||||
exec_mock = MagicMock(return_value=True)
|
session_mock.__enter__.return_value = session_mock
|
||||||
session_mock.configure_mock(**{"exec.return_value": exec_mock})
|
|
||||||
|
select1 = select(1)
|
||||||
|
|
||||||
with (
|
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, "info"),
|
||||||
patch.object(logger, "error"),
|
patch.object(logger, "error"),
|
||||||
patch.object(logger, "warn"),
|
patch.object(logger, "warn"),
|
||||||
@@ -28,6 +30,4 @@ def test_init_successful_connection() -> None:
|
|||||||
connection_successful
|
connection_successful
|
||||||
), "The database connection should be successful and not raise an exception."
|
), "The database connection should be successful and not raise an exception."
|
||||||
|
|
||||||
assert session_mock.exec.called_once_with(
|
session_mock.exec.assert_called_once_with(select1)
|
||||||
select(1)
|
|
||||||
), "The session should execute a select statement once."
|
|
||||||
|
|||||||
Reference in New Issue
Block a user