Update tests structure. (#68)

* Update tests structure.
* Unit tests use the test database
* Add function for creating database engine and session.

Author: downdawn

README.md

@@ -84,15 +84,21 @@ Execute the `backend/app/init_test_data.py` file
 
 Perform tests via pytest
 
-**Tip**: Before the test starts, please execute init the test data first, also, the fastapi service needs to be started
+1. Create a database `fba_test`, choose utf8mb4 encode
 
-1. First, go to the app directory
+2. First, go to the app directory
 
    ```shell
    cd backend/app/
    ```
+   
+3. Init the test data
+
+   ```shell
+   python tests/init_test_data.py
+   ```
 
-2. Execute the test command
+4. Execute the test command
 
    ```shell
    pytest -vs --disable-warnings

README.zh-CN.md

@@ -77,15 +77,21 @@ git clone https://github.com/wu-clan/fastapi_best_architecture.git
 
 通过 pytest 进行测试
 
-**提示**： 在测试开始前，请先执行初始化测试数据，同时，需要启动 fastapi 服务。
+1. 创建一个数据库`fba_test`，选择 utf8mb4 编码
 
-1. 首先，进入app目录
+2. 首先，进入app目录
 
    ```shell
    cd backend/app/
    ```
+   
+3. 初始化测试数据
+
+   ```shell
+   python tests/init_test_data.py
+   ```
 
-2. 执行测试命令
+4. 执行测试命令
 
    ```shell
    pytest -vs --disable-warnings

backend/__init__.py

@@ -0,0 +1,2 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-

backend/app/api/v1/auth/__init__.py

@@ -5,4 +5,4 @@
 
 router = APIRouter(prefix='/auth', tags=['认证'])
 
-router.include_router(auth_router, prefix='/users')
+router.include_router(auth_router)

backend/app/core/conf.py

@@ -32,12 +32,13 @@ class Settings(BaseSettings):
     TOKEN_WHITE_LIST: list[str]  # 白名单用户ID，可多点登录
 
     # FastAPI
+    API_V1_STR: str = '/v1'
     TITLE: str = 'FastAPI'
     VERSION: str = '0.0.1'
     DESCRIPTION: str = 'FastAPI Best Architecture'
-    DOCS_URL: str | None = '/v1/docs'
-    REDOCS_URL: str | None = '/v1/redocs'
-    OPENAPI_URL: str | None = '/v1/openapi'
+    DOCS_URL: str | None = f'{API_V1_STR}/docs'
+    REDOCS_URL: str | None = f'{API_V1_STR}/redocs'
+    OPENAPI_URL: str | None = f'{API_V1_STR}/openapi'
 
     @root_validator
     def validator_api_url(cls, values):
@@ -54,7 +55,7 @@ def validator_api_url(cls, values):
     STATIC_FILES: bool = False
 
     # MySQL
-    DB_ECHO: bool = True
+    DB_ECHO: bool = False
     DB_DATABASE: str = 'fba'
     DB_CHARSET: str = 'utf8mb4'
 
@@ -72,7 +73,7 @@ def validator_api_url(cls, values):
     # Token
     TOKEN_ALGORITHM: str = 'HS256'  # 算法
     TOKEN_EXPIRE_SECONDS: int = 60 * 60 * 24 * 1  # 过期时间，单位：秒
-    TOKEN_URL_SWAGGER: str = '/v1/auth/users/swagger_login'
+    TOKEN_URL_SWAGGER: str = f'{API_V1_STR}/auth/swagger_login'
     TOKEN_REDIS_PREFIX: str = 'fba_token'
 
     # Log
@@ -86,10 +87,10 @@ def validator_api_url(cls, values):
     # Casbin
     CASBIN_RBAC_MODEL_NAME: str = 'rbac_model.conf'
     CASBIN_EXCLUDE: list[dict[str, str], dict[str, str]] = [
-        {'method': 'POST', 'path': '/api/v1/auth/users/swagger_login'},
-        {'method': 'POST', 'path': '/api/v1/auth/users/login'},
-        {'method': 'POST', 'path': '/api/v1/auth/users/register'},
-        {'method': 'POST', 'path': '/api/v1/auth/users/password/reset'},
+        {'method': 'POST', 'path': '/api/v1/auth/swagger_login'},
+        {'method': 'POST', 'path': '/api/v1/auth/login'},
+        {'method': 'POST', 'path': '/api/v1/auth/register'},
+        {'method': 'POST', 'path': '/api/v1/auth/password/reset'},
     ]
 
     class Config:

backend/app/database/db_mysql.py

@@ -3,6 +3,7 @@
 import sys
 
 from fastapi import Depends
+from sqlalchemy import URL
 from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sessionmaker
 from typing_extensions import Annotated
 
@@ -14,20 +15,26 @@
 说明：SqlAlchemy
 """
 
+
+def create_engine_and_session(url: str | URL):
+    try:
+        # 数据库引擎
+        engine = create_async_engine(url, echo=settings.DB_ECHO, future=True, pool_pre_ping=True)
+        # log.success('数据库连接成功')
+    except Exception as e:
+        log.error('❌ 数据库链接失败 {}', e)
+        sys.exit()
+    else:
+        db_session = async_sessionmaker(bind=engine, autoflush=False, expire_on_commit=False)
+        return engine, db_session
+
+
 SQLALCHEMY_DATABASE_URL = (
     f'mysql+asyncmy://{settings.DB_USER}:{settings.DB_PASSWORD}@{settings.DB_HOST}:'
     f'{settings.DB_PORT}/{settings.DB_DATABASE}?charset={settings.DB_CHARSET}'
 )
 
-try:
-    # 数据库引擎
-    async_engine = create_async_engine(SQLALCHEMY_DATABASE_URL, echo=settings.DB_ECHO, future=True, pool_pre_ping=True)
-    # log.success('数据库连接成功')
-except Exception as e:
-    log.error('❌ 数据库链接失败 {}', e)
-    sys.exit()
-else:
-    async_db_session = async_sessionmaker(bind=async_engine, autoflush=False, expire_on_commit=False)
+async_engine, async_db_session = create_engine_and_session(SQLALCHEMY_DATABASE_URL)
 
 
 async def get_db() -> AsyncSession:

backend/app/init_test_data.py

@@ -14,28 +14,26 @@
 class InitTestData:
     """初始化测试数据"""
 
-    def __init__(self):
+    def __init__(self, session):
         self.fake = Faker('zh_CN')
+        self.session = session
 
-    @staticmethod
-    async def create_dept():
+    async def create_dept(self):
         """自动创建部门"""
-        async with async_db_session.begin() as db:
+        async with self.session.begin() as db:
             department_obj = Dept(name='test', create_user=1)
             db.add(department_obj)
         log.info('部门 test 创建成功')
 
-    @staticmethod
-    async def create_role():
+    async def create_role(self):
         """自动创建角色"""
-        async with async_db_session.begin() as db:
+        async with self.session.begin() as db:
             role_obj = Role(name='test', create_user=1)
             role_obj.menus.append(Menu(name='test', create_user=1))
             db.add(role_obj)
         log.info('角色 test 创建成功')
 
-    @staticmethod
-    async def create_test_user():
+    async def create_test_user(self):
         """创建测试用户"""
         username = 'test'
         password = 'test'
@@ -48,13 +46,12 @@ async def create_test_user():
             is_superuser=True,
             dept_id=1,
         )
-        async with async_db_session.begin() as db:
+        async with self.session.begin() as db:
             user_obj.roles.append(await db.get(Role, 1))
             db.add(user_obj)
         log.info(f'测试用户创建成功，账号：{username}，密码：{password}')
 
-    @staticmethod
-    async def create_superuser_by_yourself():
+    async def create_superuser_by_yourself(self):
         """手动创建管理员账户"""
         log.info('开始创建自定义管理员用户')
         print('请输入用户名:')
@@ -78,7 +75,7 @@ async def create_superuser_by_yourself():
             is_superuser=True,
             dept_id=1,
         )
-        async with async_db_session.begin() as db:
+        async with self.session.begin() as db:
             user_obj.roles.append(await db.get(Role, 1))
             db.add(user_obj)
         log.info(f'自定义管理员用户创建成功，账号：{username}，密码：{password}')
@@ -96,7 +93,7 @@ async def fake_user(self):
             is_superuser=False,
             dept_id=1,
         )
-        async with async_db_session.begin() as db:
+        async with self.session.begin() as db:
             user_obj.roles.append(await db.get(Role, 1))
             db.add(user_obj)
         log.info(f'普通用户创建成功，账号：{username}，密码：{password}')
@@ -115,7 +112,7 @@ async def fake_no_active_user(self):
             is_superuser=False,
             dept_id=1,
         )
-        async with async_db_session.begin() as db:
+        async with self.session.begin() as db:
             user_obj.roles.append(await db.get(Role, 1))
             db.add(user_obj)
         log.info(f'普通锁定用户创建成功，账号：{username}，密码：{password}')
@@ -133,7 +130,7 @@ async def fake_superuser(self):
             is_superuser=True,
             dept_id=1,
         )
-        async with async_db_session.begin() as db:
+        async with self.session.begin() as db:
             user_obj.roles.append(await db.get(Role, 1))
             db.add(user_obj)
         log.info(f'管理员用户创建成功，账号：{username}，密码：{password}')
@@ -152,7 +149,7 @@ async def fake_no_active_superuser(self):
             is_superuser=True,
             dept_id=1,
         )
-        async with async_db_session.begin() as db:
+        async with self.session.begin() as db:
             user_obj.roles.append(await db.get(Role, 1))
             db.add(user_obj)
         log.info(f'管理员锁定用户创建成功，账号：{username}，密码：{password}')
@@ -172,6 +169,6 @@ async def init_data(self):
 
 
 if __name__ == '__main__':
-    init = InitTestData()
+    init = InitTestData(session=async_db_session)
     loop = asyncio.get_event_loop()
     loop.run_until_complete(init.init_data())

backend/app/tests/api_v1/__init__.py

@@ -0,0 +1,2 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-

backend/app/tests/api_v1/test_auth.py

@@ -0,0 +1,21 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+from starlette.testclient import TestClient
+
+from backend.app.core.conf import settings
+
+
+def test_login(client: TestClient) -> None:
+    data = {
+        'username': 'test',
+        'password': 'test',
+    }
+    response = client.post(f'{settings.API_V1_STR}/auth/login', json=data)
+    assert response.status_code == 200
+    assert response.json()['data']['access_token_type'] == 'Bearer'
+
+
+def test_logout(client: TestClient, token_headers: dict[str, str]) -> None:
+    response = client.post(f'{settings.API_V1_STR}/auth/logout', headers=token_headers)
+    assert response.status_code == 200
+    assert response.json()['code'] == 200

backend/app/tests/conftest.py

@@ -2,30 +2,27 @@
 # -*- coding: utf-8 -*-
 import sys
 
+sys.path.append('../../')
+
 import pytest
-from httpx import AsyncClient
+from typing import Generator, Dict
 
-sys.path.append('../../')
+from starlette.testclient import TestClient
+
+from backend.app.main import app
+from backend.app.tests.utils.get_headers import get_token_headers
+from backend.app.database.db_mysql import get_db
+from backend.app.tests.utils.db_mysql import override_get_db
 
-from backend.app.common.redis import redis_client  # noqa: E402
-from backend.app.core.conf import settings  # noqa: E402
+app.dependency_overrides[get_db] = override_get_db
 
 
-@pytest.fixture(scope='session')
-def anyio_backend():
-    return 'asyncio'
+@pytest.fixture(scope='module')
+def client() -> Generator:
+    with TestClient(app) as c:
+        yield c
 
 
-@pytest.fixture(scope='package', autouse=True)
-async def function_fixture(anyio_backend):
-    auth_data = {
-        'url': f'http://{settings.UVICORN_HOST}:{settings.UVICORN_PORT}/v1/auth/users/login',
-        'headers': {'accept': 'application/json', 'Content-Type': 'application/json'},
-        'json': {'username': 'test', 'password': 'test'},
-    }
-    async with AsyncClient() as client:
-        response = await client.post(**auth_data)
-        token = response.json()['data']['access_token']
-        test_token = await redis_client.get('test_token')
-        if not test_token:
-            await redis_client.set('test_token', token, ex=86400)
+@pytest.fixture(scope='module')
+def token_headers(client: TestClient) -> Dict[str, str]:
+    return get_token_headers(client=client, username='test', password='test')