|
2 | 2 |
|
3 | 3 | import pytest |
4 | 4 |
|
5 | | -from gql import Client, GraphQLRequest, gql |
| 5 | +from gql import Client, GraphQLRequest |
6 | 6 | from gql.transport.exceptions import ( |
7 | 7 | TransportClosed, |
8 | 8 | TransportProtocolError, |
@@ -118,10 +118,10 @@ def test_code(): |
118 | 118 | batch_interval=0.01, |
119 | 119 | ) as session: |
120 | 120 |
|
121 | | - query = gql(query1_str) |
| 121 | + request = GraphQLRequest(query1_str) |
122 | 122 |
|
123 | 123 | # Execute query synchronously |
124 | | - result = session.execute(query) |
| 124 | + result = session.execute(request) |
125 | 125 |
|
126 | 126 | continents = result["continents"] |
127 | 127 |
|
@@ -172,10 +172,10 @@ def test_code(): |
172 | 172 | ) as session: |
173 | 173 |
|
174 | 174 | def test_thread(): |
175 | | - query = gql(query1_str) |
| 175 | + request = GraphQLRequest(query1_str) |
176 | 176 |
|
177 | 177 | # Execute query synchronously |
178 | | - result = session.execute(query) |
| 178 | + result = session.execute(request) |
179 | 179 |
|
180 | 180 | continents = result["continents"] |
181 | 181 |
|
@@ -306,10 +306,10 @@ def test_code(): |
306 | 306 | batch_interval=0.01, |
307 | 307 | ) as session: |
308 | 308 |
|
309 | | - query = gql(query1_str) |
| 309 | + request = GraphQLRequest(query1_str) |
310 | 310 |
|
311 | 311 | with pytest.raises(TransportServerError) as exc_info: |
312 | | - session.execute(query) |
| 312 | + session.execute(request) |
313 | 313 |
|
314 | 314 | assert "401 Client Error: Unauthorized" in str(exc_info.value) |
315 | 315 |
|
@@ -562,22 +562,21 @@ def test_requests_sync_batch_auto(): |
562 | 562 | batch_max=3, |
563 | 563 | ) |
564 | 564 |
|
565 | | - query = gql( |
566 | | - """ |
| 565 | + query = """ |
567 | 566 | query getContinentName($continent_code: ID!) { |
568 | 567 | continent(code: $continent_code) { |
569 | 568 | name |
570 | 569 | } |
571 | 570 | } |
572 | | - """ |
573 | | - ) |
| 571 | + """ |
574 | 572 |
|
575 | 573 | 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 | + ) |
579 | 578 |
|
580 | | - result = session.execute(query, variable_values=variables) |
| 579 | + result = session.execute(request) |
581 | 580 |
|
582 | 581 | name = result["continent"]["name"] |
583 | 582 | print(f"The continent with the code {continent_code} has the name: '{name}'") |
|
0 commit comments