@@ -219,13 +219,16 @@ async def test_execute_query_with_operation_name(self, mock_post):
219
219
220
220
221
221
class TestGraphqlClientSubscriptions (IsolatedAsyncioTestCase ):
222
+ """Test cases for subscribing GraphQL subscriptions."""
223
+
222
224
@patch ("websockets.connect" )
223
225
async def test_subscribe (self , mock_connect ):
226
+ """Subsribe a GraphQL subscription."""
224
227
mock_websocket = mock_connect .return_value .__aenter__ .return_value
225
228
mock_websocket .send = AsyncMock ()
226
229
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 "}}}' ,
229
232
]
230
233
231
234
client = GraphqlClient (endpoint = "ws://www.test-api.com/graphql" )
@@ -241,7 +244,7 @@ async def test_subscribe(self, mock_connect):
241
244
242
245
mock_handle .assert_has_calls (
243
246
[
244
- call ({"data" : {"messageAdded" : "test1 " }}),
245
- call ({"data" : {"messageAdded" : "test2 " }}),
247
+ call ({"data" : {"messageAdded" : "one " }}),
248
+ call ({"data" : {"messageAdded" : "two " }}),
246
249
]
247
250
)
0 commit comments