Pass explicitly created tasks when calling asyncio.await

This is required because as of Python 3.11, passing coroutine objects
to wait() directly is forbidden.

Signed-off-by: Sahas Subramanian <[email protected]>

Author: shsms

tests/actor/test_power_distributing.py

@@ -266,7 +266,8 @@ async def test_power_distributor_one_user(
             await client_handle.send(request)
 
             done, pending = await asyncio.wait(
-                [client_handle.receive()], timeout=SAFETY_TIMEOUT
+                [asyncio.create_task(client_handle.receive())],
+                timeout=SAFETY_TIMEOUT,
             )
             await distributor._stop()  # type: ignore # pylint: disable=no-member
 
@@ -337,7 +338,11 @@ async def test_power_distributor_two_users(
             await asyncio.gather(*[task1, task2])
 
             done, pending = await asyncio.wait(
-                [user1_handle.receive(), user2_handle.receive()], timeout=SAFETY_TIMEOUT
+                [
+                    asyncio.create_task(user1_handle.receive()),
+                    asyncio.create_task(user2_handle.receive()),
+                ],
+                timeout=SAFETY_TIMEOUT,
             )
             await distributor._stop()  # type: ignore # pylint: disable=no-member
 
@@ -394,7 +399,8 @@ async def test_power_distributor_invalid_battery_id(self) -> None:
             await user1_handle.send(request)
 
             done, _ = await asyncio.wait(
-                [user1_handle.receive()], timeout=SAFETY_TIMEOUT
+                [asyncio.create_task(user1_handle.receive())],
+                timeout=SAFETY_TIMEOUT,
             )
             await distributor._stop()  # type: ignore # pylint: disable=no-member
 
@@ -476,9 +482,9 @@ async def test_power_distributor_overlapping_batteries(self) -> None:
 
             done, _ = await asyncio.wait(
                 [
-                    user1_handle.receive(),
-                    user2_handle.receive(),
-                    user3_handle.receive(),
+                    asyncio.create_task(user1_handle.receive()),
+                    asyncio.create_task(user2_handle.receive()),
+                    asyncio.create_task(user3_handle.receive()),
                 ],
                 timeout=SAFETY_TIMEOUT,
             )
@@ -552,7 +558,8 @@ async def test_power_distributor_one_user_adjust_power_consume(
             await user1_handle.send(request)
 
             done, pending = await asyncio.wait(
-                [user1_handle.receive()], timeout=SAFETY_TIMEOUT
+                [asyncio.create_task(user1_handle.receive())],
+                timeout=SAFETY_TIMEOUT,
             )
             await distributor._stop()  # type: ignore # pylint: disable=no-member
 
@@ -618,7 +625,8 @@ async def test_power_distributor_one_user_adjust_power_supply(
             await user1_handle.send(request)
 
             done, pending = await asyncio.wait(
-                [user1_handle.receive()], timeout=SAFETY_TIMEOUT
+                [asyncio.create_task(user1_handle.receive())],
+                timeout=SAFETY_TIMEOUT,
             )
             await distributor._stop()  # type: ignore # pylint: disable=no-member
 
@@ -684,7 +692,8 @@ async def test_power_distributor_one_user_adjust_power_success(
             await user1_handle.send(request)
 
             done, pending = await asyncio.wait(
-                [user1_handle.receive()], timeout=SAFETY_TIMEOUT
+                [asyncio.create_task(user1_handle.receive())],
+                timeout=SAFETY_TIMEOUT,
             )
             await distributor._stop()  # type: ignore # pylint: disable=no-member
 
@@ -743,7 +752,8 @@ async def test_not_all_batteries_are_working(
             await client_handle.send(request)
 
             done, pending = await asyncio.wait(
-                [client_handle.receive()], timeout=SAFETY_TIMEOUT
+                [asyncio.create_task(client_handle.receive())],
+                timeout=SAFETY_TIMEOUT,
             )
             await distributor._stop()  # type: ignore # pylint: disable=no-member
 

tests/microgrid/test_microgrid_api.py

@@ -115,8 +115,8 @@ async def test_microgrid_api(
                 _microgrid.get()
 
             tasks = [
-                _microgrid.initialize("127.0.0.1", 10001),
-                _microgrid.initialize("127.0.0.1", 10001),
+                asyncio.create_task(_microgrid.initialize("127.0.0.1", 10001)),
+                asyncio.create_task(_microgrid.initialize("127.0.0.1", 10001)),
             ]
             initialize_task = asyncio.wait(tasks, return_when=ALL_COMPLETED)