def with_extensions():
object_extensions = {"SomeObjectExt": "object"}
field_extensions = {"SomeFieldExt": "field"}
arg_extensions = {"SomeArgExt": "arg"}
some_object = GraphQLObjectType(
"SomeObject",
{
"someField": GraphQLField(
dummy_type,
{
"someArg": GraphQLArgument(
dummy_type, extensions=arg_extensions
)
},
extensions=field_extensions,
)
},
extensions=object_extensions,
)
assert some_object.extensions is object_extensions
some_field = some_object.fields["someField"]
assert some_field.extensions is field_extensions
some_arg = some_field.args["someArg"]
assert some_arg.extensions is arg_extensions
assert some_object.to_kwargs()["extensions"] is object_extensions
assert some_field.to_kwargs()["extensions"] is field_extensions
assert some_arg.to_kwargs()["extensions"] is arg_extensions
def without_extensions():
some_object = GraphQLObjectType(
"SomeObject",
{
"someField": GraphQLField(
dummy_type, {"someArg": GraphQLArgument(dummy_type)}
)
},
)
assert some_object.extensions is None
some_field = some_object.fields["someField"]
assert some_field.extensions is None
some_arg = some_field.args["someArg"]
assert some_arg.extensions is None
assert some_object.to_kwargs()["extensions"] is None
assert some_field.to_kwargs()["extensions"] is None
assert some_arg.to_kwargs()["extensions"] is None
