File tree Expand file tree Collapse file tree 2 files changed +6
-13
lines changed
backend/app/tests/scripts Expand file tree Collapse file tree 2 files changed +6
-13
lines changed Original file line number Diff line number Diff line change 11from unittest .mock import MagicMock , patch
22
3- from sqlmodel import select
4-
5-
6-
73
84def test_init_successful_connection () -> None :
95 engine_mock = MagicMock ()
@@ -14,11 +10,10 @@ def test_init_successful_connection() -> None:
1410 enter_mock .configure_mock (** {"exec.return_value" : exec_mock })
1511 session_mock .__enter__ .return_value = enter_mock
1612
17- with (
18- patch ("sqlmodel.Session" , return_value = session_mock )
19- ):
13+ with patch ("sqlmodel.Session" , return_value = session_mock ):
2014 # causing effort if it is not here.. It seams Patch should happen before the import
2115 from app .backend_pre_start import init
16+
2217 try :
2318 init (engine_mock )
2419 connection_successful = True
Original file line number Diff line number Diff line change 11from unittest .mock import MagicMock , patch
22
3- from sqlmodel import select
43
54def test_init_successful_connection () -> None :
65 engine_mock = MagicMock ()
@@ -11,10 +10,9 @@ def test_init_successful_connection() -> None:
1110 enter_mock .configure_mock (** {"exec.return_value" : exec_mock })
1211 session_mock .__enter__ .return_value = enter_mock
1312
14- with (
15- patch ("sqlmodel.Session" , return_value = session_mock )
16- ):
17- from app .tests_pre_start import init , logger
13+ with patch ("sqlmodel.Session" , return_value = session_mock ):
14+ from app .tests_pre_start import init
15+
1816 try :
1917 init (engine_mock )
2018 connection_successful = True
@@ -24,6 +22,6 @@ def test_init_successful_connection() -> None:
2422 assert (
2523 connection_successful
2624 ), "The database connection should be successful and not raise an exception."
27-
25+
2826 session_mock .__enter__ .assert_called_once ()
2927 enter_mock .exec .assert_called_once ()
You can’t perform that action at this time.
0 commit comments