fix: correct tests

airtai · May 21, 2024 · fc393b2 · fc393b2
1 parent 536026b
commit fc393b2
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/faststream/nats/opentelemetry/provider.py b/faststream/nats/opentelemetry/provider.py
@@ -110,7 +110,7 @@ def telemetry_attributes_provider_factory(
 ]:
     if isinstance(msg, Sequence):
         return NatsBatchTelemetrySettingsProvider()
-    elif isinstance(msg, Msg):
+    elif isinstance(msg, Msg) or msg is None:
         return NatsTelemetrySettingsProvider()
     else:
         # KeyValue and Object Storage watch cases

diff --git a/tests/brokers/nats/test_kv_declarer_cache.py b/tests/brokers/nats/test_kv_declarer_cache.py
@@ -8,6 +8,7 @@
 
 
 @pytest.mark.asyncio()
+@pytest.mark.nats()
 async def test_kv_storage_cache():
     broker = NatsBroker()
     await broker.connect()

diff --git a/tests/brokers/nats/test_os_declarer_cache.py b/tests/brokers/nats/test_os_declarer_cache.py
@@ -8,6 +8,7 @@
 
 
 @pytest.mark.asyncio()
+@pytest.mark.nats()
 async def test_object_storage_cache():
     broker = NatsBroker()
     await broker.connect()