formatting

Author: quinna-h

tests/contrib/graphene/test_graphene.py

@@ -89,19 +89,6 @@ def test_schema_execute(test_schema, test_source_str):
     assert result.data == {"patron": {"id": "1", "name": "Syrus", "age": 27}}
 
 
-@pytest.mark.snapshot(
-    ignores=["meta.events", "meta.error.stack"], variants={"v2": graphene.VERSION < (3,), "": graphene.VERSION >= (3,)}
-)
-@pytest.mark.snapshot
-def test_schema_failing_extensions():
-    schema = graphene.Schema(query=Query)
-    os.environ["DD_TRACE_GRAPHQL_ERROR_EXTENSIONS"] = "code, status"
-
-    query_string = '{ user(id: "999") }'
-    result = schema.execute(query_string)
-    assert result.errors
-
-
 @pytest.mark.asyncio
 @pytest.mark.snapshot(token="tests.contrib.graphene.test_graphene.test_schema_execute")
 @pytest.mark.skipif(graphene.VERSION < (3, 0, 0), reason="execute_async is only supported in graphene>=3.0")
@@ -127,6 +114,15 @@ async def test_schema_execute_async_with_resolvers(test_schema, test_source_str,
     assert result.data == {"patron": {"id": "1", "name": "Syrus", "age": 27}}
 
 
+@pytest.mark.snapshot
+def test_schema_failing_extensions(test_schema, test_source_str, enable_graphql_resolvers):
+    schema = graphene.Schema(query=Query)
+    os.environ["DD_TRACE_GRAPHQL_ERROR_EXTENSIONS"] = "code, status"
+    query_string = '{ user(id: "999") }'
+    result = schema.execute(query_string)
+    assert result.errors
+
+
 @pytest.mark.snapshot(
     ignores=["meta.events", "meta.error.stack"], variants={"v2": graphene.VERSION < (3,), "": graphene.VERSION >= (3,)}
 )

tests/contrib/graphql/test_graphql.py

@@ -1,6 +1,8 @@
 import os
 
 import graphql
+from graphql import build_schema
+from graphql import graphql_sync
 import pytest
 
 from ddtrace.contrib.internal.graphql.patch import _graphql_version as graphql_version
@@ -75,7 +77,6 @@ async def test_graphql_with_traced_resolver(test_schema, test_source_str, snapsh
         assert result.data == {"hello": "friend"}
 
 
-
 def resolve_fail(root, info):
     undefined_var = None
     return undefined_var.property
@@ -101,13 +102,16 @@ def test_graphql_fail(enable_graphql_resolvers):
     """
 
     test_schema = build_schema(schema_definition)
-    result = graphql_sync(test_schema, query, root_value=None, field_resolver=lambda _type, _field: resolvers[_type.name][_field.name])
+    result = graphql_sync(
+        test_schema, query, root_value=None, field_resolver=lambda _type, _field: resolvers[_type.name][_field.name]
+    )
 
     assert result.errors is not None
     assert len(result.errors) == 1
     assert isinstance(result.errors[0], graphql.error.GraphQLError)
     assert "'NoneType' object has no attribute 'name'" in result.errors[0].message
 
+
 @pytest.mark.asyncio
 async def test_graphql_error(test_schema, snapshot_context):
     with snapshot_context(ignores=["meta.error.type", "meta.error.message", "meta.events"]):