From 5ab94618723c488df640ae0777ac5a16ee1ab369 Mon Sep 17 00:00:00 2001
From: David Hotham <david.hotham@microsoft.com>
Date: Sat, 13 Apr 2024 05:27:01 +0100
Subject: [PATCH] unit test fixes (#1634)

---
 autobahn/wamp/test/test_wamp_component_aio.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/autobahn/wamp/test/test_wamp_component_aio.py b/autobahn/wamp/test/test_wamp_component_aio.py
index 168933ea5..2de9bf358 100644
--- 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
