Don't use a custom asyncio loop in test

There is really no need to create our own loop for these tests, we can
just use the loop provided by `pytest-asyncio` which already offers a
lot of tools for detecting issues and configure how we want the loop to
work.

Signed-off-by: Leandro Lucarella <[email protected]>

Author: llucax

tests/test_client.py

@@ -42,7 +42,6 @@ class SetupParams:  # pylint: disable=too-many-instance-attributes
 
     client: Client
     mock_stub: AsyncMock
-    loop: asyncio.AbstractEventLoop
     gridpool_id: int
     delivery_area: DeliveryArea
     delivery_period: DeliveryPeriod
@@ -62,10 +61,6 @@ def set_up() -> Generator[Any, Any, Any]:
     mock_stub = AsyncMock()
     client._stub = mock_stub  # pylint: disable=protected-access
 
-    # Create a new event loop for each test
-    loop = asyncio.new_event_loop()
-    asyncio.set_event_loop(loop)
-
     # Set up the parameters for the orders
     # Setting delivery start to the next day 12:00
     delivery_start = (datetime.now(timezone.utc) + timedelta(days=1)).replace(
@@ -87,7 +82,6 @@ def set_up() -> Generator[Any, Any, Any]:
     yield SetupParams(
         client=client,
         mock_stub=mock_stub,
-        loop=loop,
         gridpool_id=gridpool_id,
         delivery_area=delivery_area,
         delivery_period=delivery_period,
@@ -99,8 +93,6 @@ def set_up() -> Generator[Any, Any, Any]:
         valid_until=valid_until,
     )
 
-    loop.close()
-
 
 # pylint: disable=redefined-outer-name
 def set_up_order_detail_response(
@@ -189,7 +181,7 @@ async def test_receive_public_trades(
     ]
 
 
-def test_create_gridpool_order(
+async def test_create_gridpool_order(
     set_up: SetupParams,
 ) -> None:
     """
@@ -205,17 +197,15 @@ def test_create_gridpool_order(
     )
     set_up.mock_stub.CreateGridpoolOrder.return_value = mock_response
 
-    set_up.loop.run_until_complete(
-        set_up.client.create_gridpool_order(
-            gridpool_id=set_up.gridpool_id,
-            delivery_area=set_up.delivery_area,
-            delivery_period=set_up.delivery_period,
-            order_type=set_up.order_type,
-            side=set_up.side,
-            price=set_up.price,
-            quantity=set_up.quantity,
-            execution_option=set_up.order_execution_option,  # optional field
-        )
+    await set_up.client.create_gridpool_order(
+        gridpool_id=set_up.gridpool_id,
+        delivery_area=set_up.delivery_area,
+        delivery_period=set_up.delivery_period,
+        order_type=set_up.order_type,
+        side=set_up.side,
+        price=set_up.price,
+        quantity=set_up.quantity,
+        execution_option=set_up.order_execution_option,  # optional field
     )
 
     set_up.mock_stub.CreateGridpoolOrder.assert_called_once()
@@ -229,7 +219,7 @@ def test_create_gridpool_order(
     assert args[0].order.execution_option == set_up.order_execution_option.to_pb()
 
 
-def test_update_gridpool_order(
+async def test_update_gridpool_order(
     set_up: SetupParams,
 ) -> None:
     """Test the method updating a gridpool order."""
@@ -241,13 +231,11 @@ def test_update_gridpool_order(
     )
     set_up.mock_stub.UpdateGridpoolOrder.return_value = mock_response
 
-    set_up.loop.run_until_complete(
-        set_up.client.update_gridpool_order(
-            gridpool_id=set_up.gridpool_id,
-            order_id=1,
-            quantity=set_up.quantity,
-            valid_until=set_up.valid_until,
-        )
+    await set_up.client.update_gridpool_order(
+        gridpool_id=set_up.gridpool_id,
+        order_id=1,
+        quantity=set_up.quantity,
+        valid_until=set_up.valid_until,
     )
 
     valid_until_pb = timestamp_pb2.Timestamp()
@@ -263,7 +251,7 @@ def test_update_gridpool_order(
     ), "Price field should not be set."
 
 
-def test_cancel_gridpool_order(
+async def test_cancel_gridpool_order(
     set_up: SetupParams,
 ) -> None:
     """Test the method cancelling gridpool orders."""
@@ -279,10 +267,8 @@ def test_cancel_gridpool_order(
 
     set_up.mock_stub.CancelGridpoolOrder.return_value = mock_response
 
-    set_up.loop.run_until_complete(
-        set_up.client.cancel_gridpool_order(
-            gridpool_id=set_up.gridpool_id, order_id=order_id
-        )
+    await set_up.client.cancel_gridpool_order(
+        gridpool_id=set_up.gridpool_id, order_id=order_id
     )
 
     set_up.mock_stub.CancelGridpoolOrder.assert_called_once()
@@ -389,7 +375,7 @@ async def test_list_gridpool_orders(
         ),
     ],
 )
-def test_create_gridpool_order_with_invalid_params(
+async def test_create_gridpool_order_with_invalid_params(
     # pylint: disable=too-many-arguments, too-many-positional-arguments
     set_up: SetupParams,
     price: Price,
@@ -401,18 +387,16 @@ def test_create_gridpool_order_with_invalid_params(
 ) -> None:
     """Test creating an order with invalid input parameters."""
     with pytest.raises(expected_exception):
-        set_up.loop.run_until_complete(
-            set_up.client.create_gridpool_order(
-                gridpool_id=set_up.gridpool_id,
-                delivery_area=set_up.delivery_area,
-                delivery_period=delivery_period,
-                order_type=OrderType.LIMIT,
-                side=MarketSide.BUY,
-                price=price,
-                quantity=quantity,
-                execution_option=execution_option,
-                valid_until=valid_until,
-            )
+        await set_up.client.create_gridpool_order(
+            gridpool_id=set_up.gridpool_id,
+            delivery_area=set_up.delivery_area,
+            delivery_period=delivery_period,
+            order_type=OrderType.LIMIT,
+            side=MarketSide.BUY,
+            price=price,
+            quantity=quantity,
+            execution_option=execution_option,
+            valid_until=valid_until,
         )
 
 
@@ -442,7 +426,7 @@ def test_create_gridpool_order_with_invalid_params(
         ),
     ],
 )
-def test_update_gridpool_order_with_invalid_params(  # pylint: disable=too-many-arguments
+async def test_update_gridpool_order_with_invalid_params(  # pylint: disable=too-many-arguments
     set_up: SetupParams,
     price: Price,
     quantity: Power,
@@ -451,12 +435,10 @@ def test_update_gridpool_order_with_invalid_params(  # pylint: disable=too-many-
 ) -> None:
     """Test updating an order with invalid input parameters."""
     with pytest.raises(expected_exception):
-        set_up.loop.run_until_complete(
-            set_up.client.update_gridpool_order(
-                gridpool_id=set_up.gridpool_id,
-                order_id=1,
-                price=price,
-                quantity=quantity,
-                valid_until=valid_until,
-            )
+        await set_up.client.update_gridpool_order(
+            gridpool_id=set_up.gridpool_id,
+            order_id=1,
+            price=price,
+            quantity=quantity,
+            valid_until=valid_until,
         )