运行pytest时出现异常“没有与模型关联的数据库”

2024-06-01 09:30:45 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在用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

app.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,
    )

main.app:

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)


Tags: frompytestimportapiapppytestmain