tests: fix NATS OS tes

airtai · Jun 14, 2024 · e23eccc · e23eccc
1 parent c445a11
commit e23eccc
Showing 1 changed file with 4 additions and 18 deletions.
diff --git a/tests/docs/nats/js/test_object.py b/tests/docs/nats/js/test_object.py
@@ -9,23 +9,9 @@
 async def test_basic():
     from docs.docs_src.nats.js.object import app, broker, handler
 
-    async with TestNatsBroker(broker, with_real=True):
-        await broker.start()
-
-        os = await broker.object_storage("example-bucket")
-        try:
-            existed_files = await os.list()
-        except Exception:
-            existed_files = ()
-
-        call = True
-        for file in existed_files:
-            if file.name == "file.txt":
-                call = False
-
-        if call:
-            async with TestApp(app):
-                pass
-
+    async with (
+        TestNatsBroker(broker, with_real=True, connect_only=True),
+        TestApp(app),
+    ):
         await handler.wait_call(3.0)
         handler.mock.assert_called_once_with("file.txt")