Skip to content

Commit 21963e6

Browse files
committed
Fix DeprecationWarning in test_requests_batch tests
1 parent f07c89f commit 21963e6

1 file changed

Lines changed: 14 additions & 15 deletions

File tree

tests/test_requests_batch.py

Lines changed: 14 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
import pytest
44

5-
from gql import Client, GraphQLRequest, gql
5+
from gql import Client, GraphQLRequest
66
from gql.transport.exceptions import (
77
TransportClosed,
88
TransportProtocolError,
@@ -118,10 +118,10 @@ def test_code():
118118
batch_interval=0.01,
119119
) as session:
120120

121-
query = gql(query1_str)
121+
request = GraphQLRequest(query1_str)
122122

123123
# Execute query synchronously
124-
result = session.execute(query)
124+
result = session.execute(request)
125125

126126
continents = result["continents"]
127127

@@ -172,10 +172,10 @@ def test_code():
172172
) as session:
173173

174174
def test_thread():
175-
query = gql(query1_str)
175+
request = GraphQLRequest(query1_str)
176176

177177
# Execute query synchronously
178-
result = session.execute(query)
178+
result = session.execute(request)
179179

180180
continents = result["continents"]
181181

@@ -306,10 +306,10 @@ def test_code():
306306
batch_interval=0.01,
307307
) as session:
308308

309-
query = gql(query1_str)
309+
request = GraphQLRequest(query1_str)
310310

311311
with pytest.raises(TransportServerError) as exc_info:
312-
session.execute(query)
312+
session.execute(request)
313313

314314
assert "401 Client Error: Unauthorized" in str(exc_info.value)
315315

@@ -562,22 +562,21 @@ def test_requests_sync_batch_auto():
562562
batch_max=3,
563563
)
564564

565-
query = gql(
566-
"""
565+
query = """
567566
query getContinentName($continent_code: ID!) {
568567
continent(code: $continent_code) {
569568
name
570569
}
571570
}
572-
"""
573-
)
571+
"""
574572

575573
def get_continent_name(session, continent_code):
576-
variables = {
577-
"continent_code": continent_code,
578-
}
574+
request = GraphQLRequest(
575+
query,
576+
variable_values={"continent_code": continent_code},
577+
)
579578

580-
result = session.execute(query, variable_values=variables)
579+
result = session.execute(request)
581580

582581
name = result["continent"]["name"]
583582
print(f"The continent with the code {continent_code} has the name: '{name}'")

0 commit comments

Comments
 (0)