def test_register_default_resolver(): Query = ObjectType("Query", [Field("id", String)]) schema = Schema(Query) def query_default(root, ctx, info): return 42 schema.register_default_resolver("Query", query_default) assert (cast(ObjectType, schema.get_type("Query")).default_resolver is query_default)
def test_register_default_resolver_allow_override(): Query = ObjectType("Query", [Field("id", String)]) schema = Schema(Query) def query_default(root, ctx, info): return 42 def query_default_2(root, ctx, info): return 84 schema.register_default_resolver("Query", query_default) schema.register_default_resolver("Query", query_default_2, allow_override=True) assert (cast(ObjectType, schema.get_type("Query")).default_resolver is query_default_2)
def test_register_default_resolver_already_set(): Query = ObjectType("Query", [Field("id", String)]) schema = Schema(Query) def query_default(root, ctx, info): return 42 def query_default_2(root, ctx, info): return 84 schema.register_default_resolver("Query", query_default) with pytest.raises(ValueError): schema.register_default_resolver("Query", query_default_2) assert (cast(ObjectType, schema.get_type("Query")).default_resolver is query_default)