34 lines
939 B
Python
34 lines
939 B
Python
from unittest.mock import MagicMock, patch
|
|
|
|
from sqlmodel import select
|
|
|
|
from app.backend_pre_start import init, logger
|
|
|
|
|
|
def test_init_successful_connection() -> None:
|
|
engine_mock = MagicMock()
|
|
|
|
session_mock = MagicMock()
|
|
session_mock.__enter__.return_value = session_mock
|
|
|
|
select1 = select(1)
|
|
|
|
with (
|
|
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"),
|
|
):
|
|
try:
|
|
init(engine_mock)
|
|
connection_successful = True
|
|
except Exception:
|
|
connection_successful = False
|
|
|
|
assert (
|
|
connection_successful
|
|
), "The database connection should be successful and not raise an exception."
|
|
|
|
session_mock.exec.assert_called_once_with(select1)
|