test_extend_schema: replace 'GraphQL*' types with SDL

Replicates graphql/graphql-js@57e7e1b

Author: Cito

tests/utilities/test_extend_schema.py

@@ -8,120 +8,78 @@
 from graphql.type import (
     GraphQLArgument,
     GraphQLBoolean,
-    GraphQLDirective,
-    GraphQLEnumType,
     GraphQLEnumValue,
     GraphQLField,
     GraphQLFloat,
     GraphQLID,
     GraphQLInputField,
-    GraphQLInputObjectType,
     GraphQLInt,
-    GraphQLInterfaceType,
-    GraphQLList,
     GraphQLNamedType,
-    GraphQLNonNull,
     GraphQLObjectType,
-    GraphQLScalarType,
     GraphQLSchema,
     GraphQLString,
-    GraphQLUnionType,
     assert_directive,
     assert_enum_type,
     assert_input_object_type,
     assert_interface_type,
     assert_object_type,
     assert_scalar_type,
     assert_union_type,
-    specified_directives,
     validate_schema,
 )
 from graphql.utilities import build_schema, extend_schema, print_schema
 
 # Test schema.
 
-SomeScalarType = GraphQLScalarType(name="SomeScalar")
-
-SomeInterfaceType: GraphQLInterfaceType = GraphQLInterfaceType(
-    name="SomeInterface",
-    fields=lambda: {
-        "name": GraphQLField(GraphQLString),
-        "some": GraphQLField(SomeInterfaceType),
-    },
-)
-
-FooType: GraphQLObjectType = GraphQLObjectType(
-    name="Foo",
-    interfaces=[SomeInterfaceType],
-    fields=lambda: {
-        "name": GraphQLField(GraphQLString),
-        "some": GraphQLField(SomeInterfaceType),
-        "tree": GraphQLField(GraphQLNonNull(GraphQLList(FooType))),
-    },
-)
-
-BarType = GraphQLObjectType(
-    name="Bar",
-    interfaces=[SomeInterfaceType],
-    fields=lambda: {
-        "name": GraphQLField(GraphQLString),
-        "some": GraphQLField(SomeInterfaceType),
-        "foo": GraphQLField(FooType),
-    },
-)
-
-BizType = GraphQLObjectType(
-    name="Biz", fields=lambda: {"fizz": GraphQLField(GraphQLString)}
-)
-
-SomeUnionType = GraphQLUnionType(name="SomeUnion", types=[FooType, BizType])
-
-SomeEnumType = GraphQLEnumType(
-    name="SomeEnum", values={"ONE": GraphQLEnumValue(1), "TWO": GraphQLEnumValue(2)}
-)
-
-SomeInputType = GraphQLInputObjectType(
-    "SomeInput", lambda: {"fooArg": GraphQLInputField(GraphQLString)}
-)
-
-FooDirective = GraphQLDirective(
-    name="foo",
-    args={"input": GraphQLArgument(SomeInputType)},
-    is_repeatable=True,
-    locations=[
-        DirectiveLocation.SCHEMA,
-        DirectiveLocation.SCALAR,
-        DirectiveLocation.OBJECT,
-        DirectiveLocation.FIELD_DEFINITION,
-        DirectiveLocation.ARGUMENT_DEFINITION,
-        DirectiveLocation.INTERFACE,
-        DirectiveLocation.UNION,
-        DirectiveLocation.ENUM,
-        DirectiveLocation.ENUM_VALUE,
-        DirectiveLocation.INPUT_OBJECT,
-        DirectiveLocation.INPUT_FIELD_DEFINITION,
-    ],
-)
-
-test_schema = GraphQLSchema(
-    query=GraphQLObjectType(
-        name="Query",
-        fields=lambda: {
-            "foo": GraphQLField(FooType),
-            "someScalar": GraphQLField(SomeScalarType),
-            "someUnion": GraphQLField(SomeUnionType),
-            "someEnum": GraphQLField(SomeEnumType),
-            "someInterface": GraphQLField(
-                SomeInterfaceType,
-                args={"id": GraphQLArgument(GraphQLNonNull(GraphQLID))},
-            ),
-            "someInput": GraphQLField(
-                GraphQLString, args={"input": GraphQLArgument(SomeInputType)}
-            ),
-        },
-    ),
-    types=[FooType, BarType],
-    directives=specified_directives + [FooDirective],
+test_schema = build_schema(
+    """
+    scalar SomeScalar
+
+    interface SomeInterface {
+      name: String
+      some: SomeInterface
+    }
+
+    type Foo implements SomeInterface {
+      name: String
+      some: SomeInterface
+      tree: [Foo]!
+    }
+
+    type Bar implements SomeInterface {
+      name: String
+      some: SomeInterface
+      foo: Foo
+    }
+
+    type Biz {
+      fizz: String
+    }
+
+    union SomeUnion = Foo | Biz
+
+    enum SomeEnum {
+      ONE
+      TWO
+    }
+
+    input SomeInput {
+      fooArg: String
+    }
+
+    directive @foo(input: SomeInput) repeatable on SCHEMA | SCALAR | OBJECT |
+      FIELD_DEFINITION | ARGUMENT_DEFINITION | INTERFACE | UNION | ENUM | ENUM_VALUE |
+      INPUT_OBJECT | INPUT_FIELD_DEFINITION
+
+    type Query {
+      foo: Foo
+      someScalar: SomeScalar
+      someUnion: SomeUnion
+      someEnum: SomeEnum
+      someInterface(id: ID!): SomeInterface
+      someInput(input: SomeInput): String
+    }
+    """
 )
 
 
@@ -1132,7 +1090,12 @@ def does_not_automatically_include_common_root_type_names():
             assert schema.mutation_type is None
 
         def adds_schema_definition_missing_in_the_original_schema():
-            schema = GraphQLSchema(directives=[FooDirective], types=[FooType])
+            schema = build_schema(
+                """
+                directive @foo on SCHEMA
+                type Foo
+                """
+            )
             assert schema.query_type is None
 
             extension_sdl = dedent(