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

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

2 

3import pytest 

4 

5from kwai_core.events.stream import RedisStream 

6from redis.asyncio import Redis 

7 

8 

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

10async def stream(redis: Redis) -> RedisStream: 

11 """Fixture for a redis stream. 

12 

13 The stream will be deleted after the tests. 

14 """ 

15 stream = RedisStream(redis, "kwai_test") 

16 yield stream 

17 await stream.delete()