unit test fixes (#1634)

crossbario · Apr 13, 2024 · 5ab9461 · 5ab9461
1 parent 359f868
commit 5ab9461
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/autobahn/wamp/test/test_wamp_component_aio.py b/autobahn/wamp/test/test_wamp_component_aio.py
@@ -35,7 +35,7 @@
 
     @pytest.mark.skipif(sys.version_info < (3, 5), reason="requires Python 3.5+")
     @pytest.mark.asyncio(forbid_global_loop=True)
-    def test_asyncio_component(event_loop):
+    async def test_asyncio_component(event_loop):
         orig_loop = txaio.config.loop
         txaio.config.loop = event_loop
 
@@ -72,11 +72,11 @@ def done(f):
             txaio.config.loop = orig_loop
             assert comp._done_f is None
         f.add_done_callback(done)
-        return finished
+        await finished
 
     @pytest.mark.skipif(sys.version_info < (3, 5), reason="requires Python 3.5+")
     @pytest.mark.asyncio(forbid_global_loop=True)
-    def test_asyncio_component_404(event_loop):
+    async def test_asyncio_component_404(event_loop):
         """
         If something connects but then gets aborted, it should still try
         to re-connect (in real cases this could be e.g. wrong path,
@@ -151,4 +151,4 @@ def done(f):
                 finished.set_result(None)
                 txaio.config.loop = orig_loop
             f.add_done_callback(done)
-            return finished
+            await finished