feat: configure pytest + Write first test

config.py

@@ -45,3 +45,28 @@ class ProductionConfig(Config):
         "pool_size": 10,
         "max_overflow": 20,
     }
+
+
+class TestingConfig(Config):
+    DEBUG = True
+    TESTING = True
+    SECRET_KEY = "test"
+    JWT_SECRET_KEY = "test"
+    ENABLE_SEMANTIC_SEARCH = False
+    SQLALCHEMY_DATABASE_URI = os.environ.get(
+        "DATABASE_URI_TESTING",
+        "mysql+mysqlconnector://root:@/onisep_testing?charset=utf8mb4&collation=utf8mb4_general_ci",
+    )
+
+
+def load_config(env: str) -> Config:
+    config_switch = {
+        "production": ProductionConfig,
+        "testing": TestingConfig,
+        "development": DevelopmentConfig,
+    }
+
+    config = config_switch.get(env, DevelopmentConfig)
+    config.ENV = env
+
+    return config

conftest.py

@@ -0,0 +1 @@
+pytest_plugins = ["src.tests.fixtures.service"]

src/__init__.py

@@ -1,3 +1,4 @@
+import os
 import sys
 
 from flasgger import Swagger
@@ -8,24 +9,22 @@
 from flask_sqlalchemy import SQLAlchemy
 from loguru import logger
 
-from config import DevelopmentConfig, ProductionConfig
+from config import load_config
 from src.config.swagger import swagger_config, template
-from src.constants.env import is_dev
 
 from .errors import register_error_handlers
 from .middlewares import after_request
 
 db = SQLAlchemy()
 
 
-def create_app(config_class=DevelopmentConfig):
+def create_app(environment=None):
     _setup_logging()
+    env = environment or os.environ.get("ENV")
+    config = load_config(env)
     app = Flask(__name__, instance_relative_config=True)
 
-    if not is_dev():
-        config_class = ProductionConfig
-
-    app.config.from_object(config_class)
+    app.config.from_object(config)
     CORS(app)
     db.init_app(app)
     Migrate().init_app(app, db)
@@ -53,13 +52,18 @@ def register_blueprints(app: Flask):
     app.register_blueprint(formations)
 
 
-def _setup_logging():
-    logger.remove()
-    logger.level("INFO", color="<green>")
-    logger.level("DEBUG", color="<blue>")
-    logger.level("WARNING", color="<yellow>")
-    logger.level("ERROR", color="<red>")
-    logger.level("CRITICAL", color="<red>")
+def _set_log_levels():
+    log_levels = {
+        "INFO": "<green>",
+        "DEBUG": "<blue>",
+        "WARNING": "<yellow>",
+        "ERROR": "<red>",
+        "CRITICAL": "<red>",
+    }
+    [logger.level(level, color=color) for level, color in log_levels.items()]
+
+
+def _add_stderr_logger():
     logger.add(
         sys.stderr,
         colorize=True,
@@ -70,3 +74,9 @@ def _setup_logging():
             "{extra}"
         ),
     )
+
+
+def _setup_logging():
+    logger.remove()
+    _set_log_levels()
+    _add_stderr_logger()

src/tests/fixtures/service.py

@@ -0,0 +1,35 @@
+from typing import Generator
+import pytest
+from flask import Flask
+from flask.testing import FlaskClient
+from flask_sqlalchemy import SQLAlchemy
+
+from src import create_app
+from src import db as _db
+
+
+@pytest.fixture(autouse=True, scope="session")
+def app():
+    app = create_app("testing")
+    return app
+
+
+@pytest.fixture(scope="module")
+def db(app):
+    with app.app_context():
+        _db.create_all()
+        yield _db
+        _db.drop_all()
+
+
+@pytest.fixture(scope="module")
+def db_session(db):
+    yield db.session
+    db.session.rollback()
+    db.session.close()
+
+
+@pytest.fixture()
+def client(app: Flask, db: SQLAlchemy) -> Generator[FlaskClient, None, None]:
+    with app.test_client() as client:
+        yield client

src/tests/test_config.py

@@ -0,0 +1,11 @@
+from config import TestingConfig
+
+
+def test_testing_config(db_session):
+    from flask import current_app as app
+
+    assert app.config["DEBUG"]
+    assert app.config["TESTING"]
+    assert (
+        app.config["SQLALCHEMY_DATABASE_URI"] == TestingConfig.SQLALCHEMY_DATABASE_URI
+    )