@@ -219,13 +219,16 @@ async def test_execute_query_with_operation_name(self, mock_post):
219219
220220
221221class TestGraphqlClientSubscriptions (IsolatedAsyncioTestCase ):
222+ """Test cases for subscribing GraphQL subscriptions."""
223+
222224 @patch ("websockets.connect" )
223225 async def test_subscribe (self , mock_connect ):
226+ """Subsribe a GraphQL subscription."""
224227 mock_websocket = mock_connect .return_value .__aenter__ .return_value
225228 mock_websocket .send = AsyncMock ()
226229 mock_websocket .__aiter__ .return_value = [
227- '{"type": "data", "id": "1", "payload": {"data": {"messageAdded": "test1 "}}}' ,
228- '{"type": "data", "id": "1", "payload": {"data": {"messageAdded": "test2 "}}}' ,
230+ '{"type": "data", "id": "1", "payload": {"data": {"messageAdded": "one "}}}' ,
231+ '{"type": "data", "id": "1", "payload": {"data": {"messageAdded": "two "}}}' ,
229232 ]
230233
231234 client = GraphqlClient (endpoint = "ws://www.test-api.com/graphql" )
@@ -241,7 +244,7 @@ async def test_subscribe(self, mock_connect):
241244
242245 mock_handle .assert_has_calls (
243246 [
244- call ({"data" : {"messageAdded" : "test1 " }}),
245- call ({"data" : {"messageAdded" : "test2 " }}),
247+ call ({"data" : {"messageAdded" : "one " }}),
248+ call ({"data" : {"messageAdded" : "two " }}),
246249 ]
247250 )
0 commit comments