python - 如何使用 pytest、fastapi 和 tortoise-orm 回滚每个测试?
问题描述
我正在为 crud 编写单元测试,我使用的框架是 FastAPI,ORM 是乌龟,测试模块是 pytest。
我有这个配置文件:
import os
import pytest
from starlette.testclient import TestClient
from tortoise.contrib.fastapi import register_tortoise
from app.config import Settings, get_settings
from app.main import create_application
def get_settings_override():
return Settings(
testing=1, database_url=os.environ.get("DATABASE_TEST_URL")
)
@pytest.fixture(scope="module")
def test_app_with_db():
# set up
app = create_application()
app.dependency_overrides[get_settings] = get_settings_override
# Link with DB for testing
register_tortoise(
app,
db_url=os.environ.get("DATABASE_TEST_URL"),
modules={"models": ["app.infra.postgres.models"]},
generate_schemas=True,
add_exception_handlers=True,
)
with TestClient(app) as test_client:
# testing
yield test_client
# tear down
另外,我有这些测试,第一个创建一个新的载体。第二个,创建一个运营商,然后搜索数据库中所有现有的运营商。
import json
from app.infra.postgres.crud.cft import cft as pg_cft
from app.services.cft import CFTService
cft_crud = CFTService(pg_cft)
PREFIX = "/api/cft/"
def test_create_carrier(test_app_with_db):
response = test_app_with_db.post(
f"{PREFIX}",
data=json.dumps(
{
"broker_id": 1,
"carrier_id": 1,
"lower_limit": 10,
"upper_limit": 100,
"fee": 50,
"is_active": True,
}
),
)
assert response.status_code == 201
assert type(response.json()["id"]) == int
def test_get_carriers(test_app_with_db):
response = test_app_with_db.post(
f"{PREFIX}",
data=json.dumps(
{
"broker_id": 1,
"carrier_id": 1,
"lower_limit": 10,
"upper_limit": 100,
"fee": 50,
"is_active": True,
}
),
)
summary_id = response.json()["id"]
response = test_app_with_db.get(f"{PREFIX}")
assert response.status_code == 200
response_list = response.json()
assert (
len(list(filter(lambda d: d["id"] == summary_id, response_list))) == 1
)
问题是,当我使用命令运行测试时docker-compose -f Docker-compose.dev.yml exec web python -m pytest
(容器的名称是web),我收到一个错误,因为已经有 broker_id 和 carrier_id 的组合。我想要的是为每个测试恢复数据库。我怎样才能做到这一点?
编辑:
这就是我设法做我想做的事:
import os
import pytest
from starlette.testclient import TestClient
from tortoise.contrib.test import finalizer, initializer
from app.config import Settings, get_settings
from app.main import create_application
def get_settings_override():
return Settings(testing=1, database_dev_url=os.environ.get("DATABASE_TEST_URL"))
@pytest.fixture(scope="function")
def test_app():
# set up
app = create_application()
app.dependency_overrides[get_settings] = get_settings_override
with TestClient(app) as test_client:
# testing
yield test_client
# tear down
@pytest.fixture(scope="function")
def test_app_with_db():
# set up
app = create_application()
app.dependency_overrides[get_settings] = get_settings_override
# Link with DB for testing
initializer(
["app.infra.postgres.models"],
db_url=os.environ.get("DATABASE_TEST_URL"),
)
with TestClient(app) as test_client:
# testing
yield test_client
# tear down
finalizer()
解决方案
正如 MrBean 所说,您需要为每个测试而不是每个模块运行夹具。
您可以像这样更改范围:
@pytest.fixture(scope="function")
def test_app_with_db():
# set up
...
或者您可以完全删除该scope
参数,因为它默认为function
:
这是假设您# teardown
在灯具的部分中确实有一些东西。否则没有什么可以清理数据库。
推荐阅读
- c# - 如何在.cshtml文件中以调试和发布模式声明变量
- angular - Observables 数组请求一个字符串?
- hibernate - HQL 查询返回单个最新记录,但查看两个(不是一个)日期列
- ios - 有没有办法确定 UILabel 中的文本结束坐标?(不是标签框矩形)
- java - 是否可以根据 @Autowire 的路径变量来决定使用哪个 bean?
- angularjs - 创建实际排序的AngularJS自定义过滤器
- sql - 如何修复:无法将值 NULL 插入列,插入失败
- cryptography - 找不到请求的对象 - System.Security.Cryptography.X509Certificates
- python - 为什么python中的方法不起作用,但代码之外的方法起作用?
- python - 如何设置垂直网格线matplotlib