"""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)