Fix tests and bump protos

Author: AHGIJMKLKKZNPJKQR

fishjam/_ws_notifier.py

@@ -130,7 +130,7 @@ async def _receive_loop(self):
             raise RuntimeError("Notification handler is not defined")
 
         while True:
-            message = cast(bytes, await self._websocket.recv())
+            message = await self._websocket.recv(decode=False)
             message = ServerMessage().parse(message)
             _which, message = betterproto.which_one_of(message, "content")
 

protos

@@ -57,14 +57,16 @@ def handle_notitifcation(_notification):
             pass
 
         async with asyncio.TaskGroup() as tg:
-            tg.create_task(notifier.connect())
+            notifier_task = tg.create_task(notifier.connect())
             await notifier.wait_ready()
 
             assert (
                 notifier._websocket
                 and notifier._websocket.state == websockets.State.OPEN
             )
 
+            notifier_task.cancel()
+
 
 @pytest.fixture
 def room_api():
@@ -91,7 +93,7 @@ async def test_room_created_deleted(
         async with asyncio.TaskGroup() as tg:
             assert_task = tg.create_task(assert_events(notifier, event_checks.copy()))
 
-            tg.create_task(notifier.connect())
+            notifier_task = tg.create_task(notifier.connect())
             await notifier.wait_ready()
 
             options = RoomOptions(webhook_url=WEBHOOK_URL)
@@ -101,6 +103,8 @@ async def test_room_created_deleted(
 
             await assert_task
 
+            notifier_task.cancel()
+
         for event in event_checks:
             self.assert_event(event)
 
@@ -120,15 +124,15 @@ async def test_peer_connected_disconnected(
         async with asyncio.TaskGroup() as tg:
             assert_task = tg.create_task(assert_events(notifier, event_checks.copy()))
 
-            tg.create_task(notifier.connect())
+            notifier_task = tg.create_task(notifier.connect())
             await notifier.wait_ready()
 
             options = RoomOptions(webhook_url=WEBHOOK_URL)
             room = room_api.create_room(options=options)
 
             peer, token = room_api.create_peer(room.id)
             peer_socket = PeerSocket(fishjam_url=FISHJAM_ID)
-            tg.create_task(peer_socket.connect(token))
+            peer_socket_task = tg.create_task(peer_socket.connect(token))
 
             await peer_socket.wait_ready()
 
@@ -137,6 +141,9 @@ async def test_peer_connected_disconnected(
 
             await assert_task
 
+            notifier_task.cancel()
+            peer_socket_task.cancel()
+
         for event in event_checks:
             self.assert_event(event)
 
@@ -155,22 +162,25 @@ async def test_peer_connected_room_deleted(
         async with asyncio.TaskGroup() as tg:
             assert_task = tg.create_task(assert_events(notifier, event_checks.copy()))
 
-            tg.create_task(notifier.connect())
+            notifier_task = tg.create_task(notifier.connect())
             await notifier.wait_ready()
 
             options = RoomOptions(webhook_url=WEBHOOK_URL)
             room = room_api.create_room(options=options)
             _peer, token = room_api.create_peer(room.id)
 
             peer_socket = PeerSocket(fishjam_url=FISHJAM_ID)
-            tg.create_task(peer_socket.connect(token))
+            peer_socket_task = tg.create_task(peer_socket.connect(token))
 
             await peer_socket.wait_ready()
 
             room_api.delete_room(room.id)
 
             await assert_task
 
+            notifier_task.cancel()
+            peer_socket_task.cancel()
+
         for event in event_checks:
             self.assert_event(event)