fix: add pytestmark bus

fbraem · Jul 28, 2023 · b1cde47 · b1cde47
1 parent a8f4e55
commit b1cde47
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/backend/src/tests/core/events/test_consumer.py b/backend/src/tests/core/events/test_consumer.py
@@ -6,6 +6,8 @@
 from kwai.core.events.consumer import RedisConsumer
 from kwai.core.events.stream import RedisMessage, RedisStream
 
+pytestmark = pytest.mark.bus
+
 
 @pytest.mark.asyncio
 async def test_consumer(stream: RedisStream):

diff --git a/backend/src/tests/core/events/test_stream.py b/backend/src/tests/core/events/test_stream.py
@@ -4,6 +4,8 @@
 
 from kwai.core.events.stream import RedisMessage, RedisStream
 
+pytestmark = pytest.mark.bus
+
 
 @pytest.mark.asyncio
 async def test_add(stream: RedisStream):