25 lines
543 B
Python
25 lines
543 B
Python
"""pytest 共享 fixture。"""
|
|
import fakeredis.aioredis
|
|
import pytest
|
|
import pytest_asyncio
|
|
|
|
from app.repository.agent_repo import AgentRepo
|
|
from app.repository.task_repo import TaskRepo
|
|
|
|
|
|
@pytest_asyncio.fixture
|
|
async def redis():
|
|
server = fakeredis.FakeServer()
|
|
client = fakeredis.aioredis.FakeRedis(server=server, decode_responses=True)
|
|
yield client
|
|
await client.aclose()
|
|
|
|
|
|
@pytest_asyncio.fixture
|
|
def task_repo(redis):
|
|
return TaskRepo(redis)
|
|
|
|
|
|
@pytest_asyncio.fixture
|
|
def agent_repo(redis):
|
|
return AgentRepo(redis) |