mirror of
https://github.com/fastapi-practices/fastapi_best_architecture.git
synced 2026-03-13 09:31:31 +08:00
78dad3021a
* feat: add postgresql db supports * change: change mysql conn str create way * fix: Modify the default alembic migration file to meet multi-database support * Update settings and lint * update models * Simplify database config * Simplify the get db method * Update create db url * Updated model type adaptation * Update sql scripts * Fix models type * Adaptation to postgresql code generation * Update README.md * Fix alembic file template * Update docker scripts
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
from typing import Generator
|
|
|
|
import pytest
|
|
|
|
from starlette.testclient import TestClient
|
|
|
|
from backend.app.admin.tests.utils.db import override_get_db
|
|
from backend.core.conf import settings
|
|
from backend.database.db import get_db
|
|
from backend.main import app
|
|
|
|
# 重载数据库
|
|
app.dependency_overrides[get_db] = override_get_db
|
|
|
|
|
|
# Test data
|
|
PYTEST_USERNAME = 'admin'
|
|
PYTEST_PASSWORD = '123456'
|
|
PYTEST_BASE_URL = f'http://testserver{settings.FASTAPI_API_V1_PATH}'
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def client() -> Generator:
|
|
with TestClient(app, base_url=PYTEST_BASE_URL) as c:
|
|
yield c
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def token_headers(client: TestClient) -> dict[str, str]:
|
|
params = {
|
|
'username': PYTEST_USERNAME,
|
|
'password': PYTEST_PASSWORD,
|
|
}
|
|
response = client.post('/auth/login/swagger', params=params)
|
|
response.raise_for_status()
|
|
token_type = response.json()['token_type']
|
|
access_token = response.json()['access_token']
|
|
headers = {'Authorization': f'{token_type} {access_token}'}
|
|
return headers
|