Skip to content

Commit d80d387

Browse files
committed
fix pre-commit error
1 parent 9f3d21a commit d80d387

File tree

2 files changed

+6
-13
lines changed

2 files changed

+6
-13
lines changed

backend/app/tests/scripts/test_backend_pre_start.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,5 @@
11
from unittest.mock import MagicMock, patch
22

3-
from sqlmodel import select
4-
5-
6-
73

84
def 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
Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
from unittest.mock import MagicMock, patch
22

3-
from sqlmodel import select
43

54
def 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()

0 commit comments

Comments
 (0)