FastAPI 依赖覆盖失败

Question

我正在使用 MongoMock 来测试我的 FastAPI 应用程序，但我无法覆盖 FastAPI 的依赖项。

create_test_data将使用 MongoMock 的客户端插入一些伪造的测试数据，并返回task_id请求所需的 task_id。

from mongomock import MongoClient
from fastapi.testclient import TestClient
from pytest import fixture
from app import app
from core.utils.db_utils import get_db
from models.tasks import TaskCreate
from models.frequency import Frequency


@fixture(name="conn")
def conn_fixture():
    with MongoClient() as conn:
        yield conn


@fixture(name="client")
def client_fixture(conn: MongoClient):
    def get_conn_override():
        return conn

    app.dependency_overrides[get_db] = get_conn_override

    client = TestClient(app)
    yield client
    app.dependency_overrides.clear()


def test_wrong_api_log(conn: MongoClient, client: TestClient):
    task_id = create_test_data(conn=conn, platform="test")
    assert task_id
    
    response = client.get(url=f"/api/tasks/{task_id}")
    assert response.text

在“test_wrong_api_log”function 中，第一个带有 task_id 的断言可以通过，但第二个不能通过似乎是覆盖 FastAPI 的依赖失败原因。

以下代码是get_db ， connect_to_mongo将返回 pymongo 的 MongoClient 而不是 MongoMock 的 MongoClient。

from db.mongodb_utils import connect_to_mongo


def get_db():
    db_client = connect_to_mongo()
    try:
        yield db_client
    finally:
        db_client.close()

无论如何要解决这个问题？

Answer 1

你永远不会打电话给你的client_fixture 。 此外，您的conn也应该调用您的夹具。

尝试这个

def test_wrong_api_log(conn_fixture: MongoClient, client_fixture: TestClient):
    task_id = create_test_data(conn=conn_fixture, platform="test")
    assert task_id
    
    response = client_fixture.get(url=f"/api/tasks/{task_id}")
    assert response.text