fix new channels tests

strawberry-graphql · Jul 8, 2023 · e7b17a7 · e7b17a7
1 parent 6085526
commit e7b17a7
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/channels/test_layers.py b/tests/channels/test_layers.py
@@ -83,7 +83,7 @@ async def test_channel_listen(ws: WebsocketCommunicator):
     )
 
     response = await ws.receive_json_from()
-    del response["payload"]["extensions"]
+    response["payload"].pop("extensions", None)
     assert (
         response
         == NextMessage(
@@ -128,6 +128,7 @@ async def test_channel_listen_with_confirmation(ws: WebsocketCommunicator):
     )
 
     response = await ws.receive_json_from()
+    response["payload"].pop("extensions", None)
     assert (
         response
         == NextMessage(
@@ -296,7 +297,7 @@ async def test_channel_listen_group(ws: WebsocketCommunicator):
         },
     )
     response = await ws.receive_json_from()
-    del response["payload"]["extensions"]
+    response["payload"].pop("extensions", None)
     assert (
         response
         == NextMessage(
@@ -313,7 +314,7 @@ async def test_channel_listen_group(ws: WebsocketCommunicator):
     )
 
     response = await ws.receive_json_from()
-    del response["payload"]["extensions"]
+    response["payload"].pop("extensions", None)
     assert (
         response
         == NextMessage(
@@ -358,6 +359,7 @@ async def test_channel_listen_group_cm(ws: WebsocketCommunicator):
         },
     )
     response = await ws.receive_json_from()
+    response["payload"].pop("extensions", None)
     assert (
         response
         == NextMessage(
@@ -374,6 +376,7 @@ async def test_channel_listen_group_cm(ws: WebsocketCommunicator):
     )
 
     response = await ws.receive_json_from()
+    response["payload"].pop("extensions", None)
     assert (
         response
         == NextMessage(