🏷️ Add mypy to the GitHub Action for tests and fixed types in the whole project (#655)

Co-authored-by: Sebastián Ramírez <tiangolo@gmail.com>

backend/app/tests/scripts/test_backend_pre_start.py

@@ -1,21 +1,22 @@
 from unittest.mock import MagicMock
 
+from pytest_mock import MockerFixture
 from sqlmodel import select
 
 from app.backend_pre_start import init, logger
 
 
-def test_init_successful_connection(mocker):
+def test_init_successful_connection(mocker: MockerFixture) -> None:
     engine_mock = MagicMock()
 
     session_mock = MagicMock()
     exec_mock = MagicMock(return_value=True)
-    session_mock.configure_mock(**{'exec.return_value': exec_mock})
-    mocker.patch('sqlmodel.Session', return_value=session_mock)
+    session_mock.configure_mock(**{"exec.return_value": exec_mock})
+    mocker.patch("sqlmodel.Session", return_value=session_mock)
 
-    mocker.patch.object(logger, 'info')
-    mocker.patch.object(logger, 'error')
-    mocker.patch.object(logger, 'warn')
+    mocker.patch.object(logger, "info")
+    mocker.patch.object(logger, "error")
+    mocker.patch.object(logger, "warn")
 
     try:
         init(engine_mock)
@@ -23,6 +24,10 @@ def test_init_successful_connection(mocker):
     except Exception:
         connection_successful = False
 
-    assert connection_successful, "The database connection should be successful and not raise an exception."
+    assert (
+        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."
+    assert session_mock.exec.called_once_with(
+        select(1)
+    ), "The session should execute a select statement once."

backend/app/tests/scripts/test_celery_pre_start.py

@@ -1,21 +1,22 @@
 from unittest.mock import MagicMock
 
+from pytest_mock import MockerFixture
 from sqlmodel import select
 
 from app.celeryworker_pre_start import init, logger
 
 
-def test_init_successful_connection(mocker):
+def test_init_successful_connection(mocker: MockerFixture) -> None:
     engine_mock = MagicMock()
 
     session_mock = MagicMock()
     exec_mock = MagicMock(return_value=True)
-    session_mock.configure_mock(**{'exec.return_value': exec_mock})
-    mocker.patch('sqlmodel.Session', return_value=session_mock)
+    session_mock.configure_mock(**{"exec.return_value": exec_mock})
+    mocker.patch("sqlmodel.Session", return_value=session_mock)
 
-    mocker.patch.object(logger, 'info')
-    mocker.patch.object(logger, 'error')
-    mocker.patch.object(logger, 'warn')
+    mocker.patch.object(logger, "info")
+    mocker.patch.object(logger, "error")
+    mocker.patch.object(logger, "warn")
 
     try:
         init(engine_mock)
@@ -23,6 +24,10 @@ def test_init_successful_connection(mocker):
     except Exception:
         connection_successful = False
 
-    assert connection_successful, "The database connection should be successful and not raise an exception."
+    assert (
+        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."
+    assert session_mock.exec.called_once_with(
+        select(1)
+    ), "The session should execute a select statement once."

backend/app/tests/scripts/test_test_pre_start.py

@@ -1,21 +1,22 @@
 from unittest.mock import MagicMock
 
+from pytest_mock import MockerFixture
 from sqlmodel import select
 
 from app.tests_pre_start import init, logger
 
 
-def test_init_successful_connection(mocker):
+def test_init_successful_connection(mocker: MockerFixture) -> None:
     engine_mock = MagicMock()
 
     session_mock = MagicMock()
     exec_mock = MagicMock(return_value=True)
-    session_mock.configure_mock(**{'exec.return_value': exec_mock})
-    mocker.patch('sqlmodel.Session', return_value=session_mock)
+    session_mock.configure_mock(**{"exec.return_value": exec_mock})
+    mocker.patch("sqlmodel.Session", return_value=session_mock)
 
-    mocker.patch.object(logger, 'info')
-    mocker.patch.object(logger, 'error')
-    mocker.patch.object(logger, 'warn')
+    mocker.patch.object(logger, "info")
+    mocker.patch.object(logger, "error")
+    mocker.patch.object(logger, "warn")
 
     try:
         init(engine_mock)
@@ -23,6 +24,10 @@ def test_init_successful_connection(mocker):
     except Exception:
         connection_successful = False
 
-    assert connection_successful, "The database connection should be successful and not raise an exception."
+    assert (
+        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."
+    assert session_mock.exec.called_once_with(
+        select(1)
+    ), "The session should execute a select statement once."