Coverage for src/tests/core/events/conftest.py: 100%

1"""Define common fixtures for Redis.""" 

2 

3from typing import AsyncGenerator 

4 

5import pytest 

6 

7from kwai_core.events.stream import RedisStream 

8from redis.asyncio import Redis 

9 

10 

11@pytest.fixture(scope="module") 

12async def stream(redis: Redis) -> AsyncGenerator[RedisStream, None]: 

13 """Fixture for a redis stream. 

14 

15 The stream will be deleted after the tests. 

16 """ 

17 stream = RedisStream(redis, "kwai_test") 

18 yield stream 

19 await stream.delete()