Add @oneOf support to introspection query (#241)

Replicates graphql/graphql-js@7df786e

Author: erikwrede

src/graphql/utilities/build_client_schema.py

@@ -256,6 +256,7 @@ def build_input_object_def(
             fields=lambda: build_input_value_def_map(
                 input_object_introspection["inputFields"]
             ),
+            is_one_of=input_object_introspection.get("isOneOf"),
         )
 
     type_builders: dict[str, Callable[[IntrospectionType], GraphQLNamedType]] = {

src/graphql/utilities/get_introspection_query.py

@@ -17,7 +17,6 @@
 except ImportError:  # Python < 3.10
     from typing_extensions import TypeAlias
 
-
 __all__ = [
     "IntrospectionDirective",
     "IntrospectionEnumType",
@@ -44,6 +43,7 @@ def get_introspection_query(
     directive_is_repeatable: bool = False,
     schema_description: bool = False,
     input_value_deprecation: bool = False,
+    input_object_one_of: bool = False,
 ) -> str:
     """Get a query for introspection.
 
@@ -55,6 +55,7 @@ def get_introspection_query(
     maybe_specified_by_url = "specifiedByURL" if specified_by_url else ""
     maybe_directive_is_repeatable = "isRepeatable" if directive_is_repeatable else ""
     maybe_schema_description = maybe_description if schema_description else ""
+    maybe_input_object_one_of = "isOneOf" if input_object_one_of else ""
 
     def input_deprecation(string: str) -> str | None:
         return string if input_value_deprecation else ""
@@ -87,6 +88,7 @@ def input_deprecation(string: str) -> str | None:
           name
           {maybe_description}
           {maybe_specified_by_url}
+          {maybe_input_object_one_of}
           fields(includeDeprecated: true) {{
             name
             {maybe_description}
@@ -253,6 +255,7 @@ class IntrospectionEnumType(WithName):
 class IntrospectionInputObjectType(WithName):
     kind: Literal["input_object"]
     inputFields: list[IntrospectionInputValue]
+    isOneOf: bool
 
 
 IntrospectionType: TypeAlias = Union[
@@ -264,7 +267,6 @@ class IntrospectionInputObjectType(WithName):
     IntrospectionInputObjectType,
 ]
 
-
 IntrospectionOutputType: TypeAlias = Union[
     IntrospectionScalarType,
     IntrospectionObjectType,
@@ -273,7 +275,6 @@ class IntrospectionInputObjectType(WithName):
     IntrospectionEnumType,
 ]
 
-
 IntrospectionInputType: TypeAlias = Union[
     IntrospectionScalarType, IntrospectionEnumType, IntrospectionInputObjectType
 ]

src/graphql/utilities/introspection_from_schema.py

@@ -21,6 +21,7 @@ def introspection_from_schema(
     directive_is_repeatable: bool = True,
     schema_description: bool = True,
     input_value_deprecation: bool = True,
+    input_object_one_of: bool = True,
 ) -> IntrospectionQuery:
     """Build an IntrospectionQuery from a GraphQLSchema
 
@@ -37,6 +38,7 @@ def introspection_from_schema(
             directive_is_repeatable,
             schema_description,
             input_value_deprecation,
+            input_object_one_of,
         )
     )
 

tests/utilities/test_get_introspection_query.py

@@ -2,7 +2,6 @@
 
 import re
 from typing import Pattern
-
 from graphql.language import parse
 from graphql.utilities import build_schema, get_introspection_query
 from graphql.validation import validate
@@ -80,6 +79,11 @@ def includes_deprecation_reason_field_on_input_values():
             "deprecationReason", 2
         )
 
+    def includes_input_object_one_of_field():
+        ExcpectIntrospectionQuery().to_not_match("isOneOf")
+        ExcpectIntrospectionQuery(input_object_one_of=True).to_match("isOneOf")
+        ExcpectIntrospectionQuery(input_object_one_of=False).to_not_match("isOneOf")
+
     def includes_deprecated_input_field_and_args():
         ExcpectIntrospectionQuery().to_match("includeDeprecated: true", 2)
         ExcpectIntrospectionQuery(input_value_deprecation=True).to_match(