Add some fixtures typing

Author: insspb

tests/conftest.py

@@ -1,4 +1,5 @@
 from datetime import datetime
+from typing import NoReturn
 
 import mongoengine
 import pytest
@@ -9,7 +10,7 @@
 
 
 @pytest.fixture(autouse=True, scope="session")
-def session_clean_up():
+def session_clean_up() -> NoReturn:
     """Mandatory tests environment clean up before/after test session."""
     client = MongoClient("localhost", 27017)
     client.drop_database("flask_mongoengine_test_db")
@@ -24,7 +25,7 @@ def session_clean_up():
 
 
 @pytest.fixture()
-def app():
+def app() -> Flask:
     app = Flask(__name__)
     app.config["TESTING"] = True
     app.config["WTF_CSRF_ENABLED"] = False
@@ -36,7 +37,7 @@ def app():
 
 
 @pytest.fixture()
-def db(app):
+def db(app) -> MongoEngine:
     app.config["MONGODB_HOST"] = "mongodb://localhost:27017/flask_mongoengine_test_db"
     test_db = MongoEngine(app)
     db_name = (

tests/test_debug_panel.py

@@ -3,7 +3,6 @@
 
 - Independent of global configuration by design.
 """
-
 import contextlib
 
 import jinja2
@@ -18,14 +17,15 @@
 from pytest_mock import MockerFixture
 
 from flask_mongoengine.panels import (
+    MongoCommandLogger,
     MongoDebugPanel,
     _maybe_patch_jinja_loader,
     mongo_command_logger,
 )
 
 
 @pytest.fixture()
-def app_no_mongo_monitoring():
+def app_no_mongo_monitoring() -> Flask:
     app = Flask(__name__)
     app.config["TESTING"] = True
     app.config["WTF_CSRF_ENABLED"] = False
@@ -39,7 +39,7 @@ def app_no_mongo_monitoring():
 
 
 @pytest.fixture(autouse=True)
-def registered_monitoring():
+def registered_monitoring() -> MongoCommandLogger:
     """Register/Unregister mongo_command_logger in required tests"""
     monitoring.register(mongo_command_logger)
     mongo_command_logger.reset_tracker()
@@ -69,7 +69,7 @@ class TestMongoDebugPanel:
     """Trivial tests to highlight any unexpected changes in namings or code."""
 
     @pytest.fixture
-    def toolbar_with_no_flask(self):
+    def toolbar_with_no_flask(self) -> MongoDebugPanel:
         """Simple instance of MongoDebugPanel without flask application"""
         jinja2_env = jinja2.Environment()
         return MongoDebugPanel(jinja2_env)
@@ -186,7 +186,7 @@ class TestMongoCommandLogger:
     """By design tested with raw pymongo."""
 
     @pytest.fixture(autouse=True)
-    def py_db(self, registered_monitoring):
+    def py_db(self, registered_monitoring) -> pymongo.MongoClient:
         """Clean up and returns special database for testing on pymongo driver level"""
         client = pymongo.MongoClient("localhost", 27017)
         db = client.pymongo_test_database