Ok

Author: antonpirker

tests/integrations/strawberry/test_strawberry.py

@@ -198,7 +198,7 @@ def test_capture_request_if_available_and_send_pii_is_on(
     client = client_factory(schema)
 
     query = "query ErrorQuery { error }"
-    client.post("/graphql", json={"query": query, "operationName": "ErrorQuery"}).get_data()
+    client.post("/graphql", json={"query": query, "operationName": "ErrorQuery"}).close()
 
     assert len(events) == 1
 
@@ -253,7 +253,7 @@ def test_do_not_capture_request_if_send_pii_is_off(
     client = client_factory(schema)
 
     query = "query ErrorQuery { error }"
-    client.post("/graphql", json={"query": query, "operationName": "ErrorQuery"}).get_data()
+    client.post("/graphql", json={"query": query, "operationName": "ErrorQuery"}).close()
 
     assert len(events) == 1
 
@@ -293,7 +293,7 @@ def test_breadcrumb_no_operation_name(
     client = client_factory(schema)
 
     query = "{ error }"
-    client.post("/graphql", json={"query": query}).get_data()
+    client.post("/graphql", json={"query": query}).close()
 
     assert len(events) == 1
 
@@ -332,7 +332,7 @@ def test_capture_transaction_on_error(
     client = client_factory(schema)
 
     query = "query ErrorQuery { error }"
-    client.post("/graphql", json={"query": query, "operationName": "ErrorQuery"}).get_data()
+    client.post("/graphql", json={"query": query, "operationName": "ErrorQuery"}).close()
 
     assert len(events) == 2
     (_, transaction_event) = events
@@ -409,7 +409,7 @@ def test_capture_transaction_on_success(
     client = client_factory(schema)
 
     query = "query GreetingQuery { hello }"
-    client.post("/graphql", json={"query": query, "operationName": "GreetingQuery"}).get_data()
+    client.post("/graphql", json={"query": query, "operationName": "GreetingQuery"}).close()
 
     assert len(events) == 1
     (transaction_event,) = events
@@ -486,7 +486,7 @@ def test_transaction_no_operation_name(
     client = client_factory(schema)
 
     query = "{ hello }"
-    client.post("/graphql", json={"query": query}).get_data()
+    client.post("/graphql", json={"query": query}).close()
 
     assert len(events) == 1
     (transaction_event,) = events
@@ -566,7 +566,7 @@ def test_transaction_mutation(
     client = client_factory(schema)
 
     query = 'mutation Change { change(attribute: "something") }'
-    client.post("/graphql", json={"query": query}).get_data()
+    client.post("/graphql", json={"query": query}).close()
 
     assert len(events) == 1
     (transaction_event,) = events
@@ -641,7 +641,7 @@ def test_handle_none_query_gracefully(
     client_factory = request.getfixturevalue(client_factory)
     client = client_factory(schema)
 
-    client.post("/graphql", json={}).get_data()
+    client.post("/graphql", json={}).close()
 
     assert len(events) == 0, "expected no events to be sent to Sentry"
 
@@ -673,7 +673,7 @@ def test_span_origin(
     client = client_factory(schema)
 
     query = 'mutation Change { change(attribute: "something") }'
-    client.post("/graphql", json={"query": query}).get_data()
+    client.post("/graphql", json={"query": query}).close()
 
     (event,) = events
 
@@ -715,7 +715,7 @@ def test_span_origin2(
     client = client_factory(schema)
 
     query = "query GreetingQuery { hello }"
-    client.post("/graphql", json={"query": query, "operationName": "GreetingQuery"}).get_data()
+    client.post("/graphql", json={"query": query, "operationName": "GreetingQuery"}).close()
 
     (event,) = events
 
@@ -757,7 +757,7 @@ def test_span_origin3(
     client = client_factory(schema)
 
     query = "subscription { messageAdded { content } }"
-    client.post("/graphql", json={"query": query}).get_data()
+    client.post("/graphql", json={"query": query}).close()
 
     (event,) = events