1

我正在使用 FastAPI 和 Tortoise-ORM 开发服务。

当我使用 Swagger UI 或者 curl 生成的界面时，可以成功添加和读取数据。

但是，当我运行 pytest 时，测试失败并显示以下错误消息：tortoise.exceptions.ConfigurationError: No DB associated to model

请记住，该错误仅在使用 pytest 时发生，我认为问题是某些配置错误或测试脚本中缺少，但我找不到原因。

有没有人有任何想法？

我的结构如下：

src /
     +--api /
     |      +-__ init__.py
     |      +-app.py
     |      +-main.py
     |      +-models.py
     |      +-routers.py
     |      +-schemas.py
     +--tests /
              +-__ init__.py
              +-test_subjects.py

test_1.py 文件如下：

import pytest
from fastapi.testclient import TestClient
from api.main import app

client = TestClient(app)


def test_create_subject():
    response = await client.post(
        '/api/subject/',
        json={
            'name': 'Programming',
        },
    )

def test_read_subjects():
    response = client.get("/api/subjects/")
    assert response.status_code == 200

应用程序.py：

from fastapi import FastAPI
from tortoise.contrib.fastapi import register_tortoise
from tortoise import Tortoise

def get_application():
    _app = FastAPI(title='MyProject')

    _app.add_middleware(
        CORSMiddleware,
        allow_credentials=True,
        allow_methods=['*'],
        allow_headers=['*'],
    )
    return _app

app = get_application()


@app.on_event('startup')
async def startup():
   register_tortoise(
        app,
        db_url='sqlite://db.sqlite3',
        modules={'models': ['api.models']},
        generate_schemas=True,
        add_exception_handlers=True,
    )

主应用程序：

import uvicorn
from .app import app
from .routers import subjects
from .schemas.subjects import SubjectSchema


app.include_router(subjects.router)


if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=8000)
4

1 回答 1

0

在我的情况下，它不是需要在 on_event('startup') 函数中的 register_tortoise 函数，而是我的代码的另一部分在初始化之前试图使用数据库。我将这段代码（具有查询的类的实例化）移动到 on_event('startup') 块中，一切都开始工作了。基本上，如果您有任何在 register_tortoise 之前触发的数据库查询，它将失败。（这就是它与 Swagger 一起使用的原因）

于 2021-11-28T18:15:13.257 回答