无法使用pytest模拟get_redis函数

Question

我在应用程序上使用 fakeredis 和 pytest

我的文件上的 get_redis 函数

app/helpers/providers.py

：

from redis import ConnectionPool, Redis

redis_pool = None


def get_redis() -> Redis:
    global redis_pool
    if redis_pool is None:
        redis_pool = ConnectionPool()
    return Redis.from_pool(redis_pool)

然后我在我的代码中使用它，位于

app/endpoints/secrets.py

:

from app.helpers.providers import get_redis

@router.get("/secrets/{secret_id}")
async def get_secret(
    request: Request,
    secret_id: UUID,
    credentials: Annotated[HTTPBasicCredentials, Depends(security)],
    redis: Redis = Depends(get_redis),
):
    with redis.pipeline() as pipe: # I use redis from get_redis here
        pipe.get(key)
        pipe.delete(key)
        results = pipe.execute()

    return results # just to simplify

然后在我的测试中我得到以下结果：

import sys
from unittest.mock import patch

import fakeredis
import pytest
from fastapi import HTTPException, status
from fastapi.exceptions import RequestValidationError
from fastapi.testclient import TestClient


@pytest.fixture
def fake_redis():
    return fakeredis.FakeStrictRedis()


@pytest.fixture(autouse=True)
def mock_redis_dependencies(fake_redis):
    with patch('app.endpoints.secrets.get_redis', return_value=fake_redis):
        yield


sys.path.append(".")

from app.endpoints.secrets import router


def test_secret_not_found(fake_redis):
    client = TestClient(router)
    with pytest.raises(HTTPException) as exc:
        client.get(
            "/api/secrets/aaaaaaaa-bbbb-4ccc-aaaa-eeeeeeeeeef1",
            auth=("admin", "admin"),
        )

    assert exc.value.status_code == status.HTTP_404_NOT_FOUND

测试失败，因为它尝试使用真正的redis，这会引发异常。

我做错了什么？我还有另一个测试，效果很好。

Answer 1

这解决了我的问题

@pytest.fixture(autouse=True)
def mock_redis_dependencies(monkeypatch, fake_redis):
    monkeypatch.setattr("app.decorators.rate_limit.redis", fake_redis)
    monkeypatch.setattr("app.endpoints.secrets.redis", fake_redis)