Skip to content

Commit 102c67c

Browse files
committed
Fix linting errors
1 parent 774db0f commit 102c67c

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

tests/test_graphql_client.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -219,13 +219,16 @@ async def test_execute_query_with_operation_name(self, mock_post):
219219

220220

221221
class 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

Comments
 (0)