def test_graphql_options():
class Book(GraphQLModel):
class GraphQL:
gql_type = "Book"
class User_2(GraphQLModel):
class GraphQL:
gql_type = "User"
include = ["username"]
exclude = ["profile"]
class User_3(GraphQLModel):
class GraphQL:
gql_type = "User"
include = ["unknow"]
bindable = PydanticBindable({"User_2": User_2, "Book": Book})
with pytest.raises(PydanticBindError):
make_executable_schema(
schema,
base_scalars_resolvers,
snake_case_fallback_resolvers,
bindable,
)
bindable = PydanticBindable({"User_3": User_3, "Book": Book})
with pytest.raises(PydanticBindError):
make_executable_schema(
schema,
base_scalars_resolvers,
snake_case_fallback_resolvers,
bindable,
)
def test_register_type():
class Foo(GraphQLModel):
class GraphQL:
gql_type = "Foo"
bindable = PydanticBindable({"Foo": Foo})
# Remove JSON type
bindable._type_map.pop("JSON")
with pytest.raises(PydanticBindError):
make_executable_schema(
schema,
base_scalars_resolvers,
snake_case_fallback_resolvers,
bindable,
)
# Bring it back through `register` method
bindable.register_scalar("JSON", dict)
make_executable_schema(
schema,
base_scalars_resolvers,
snake_case_fallback_resolvers,
bindable,
)
def test_validator():
class Book(GraphQLModel):
class GraphQL:
gql_type = "Book"
class User(GraphQLModel):
class GraphQL:
gql_type = "User"
@validator("username")
def check_username(value):
if len(value) <= 3:
raise ValueError("Username length must be greater than 3")
return value
bindable = PydanticBindable({"User": User, "Book": Book})
make_executable_schema(
schema,
base_scalars_resolvers,
snake_case_fallback_resolvers,
bindable,
)
User(username="******")
with pytest.raises(ValidationError):
User(username="******")
def test_graphql_types():
"""Test pydantic bindings."""
class GraphQLTypes(GraphQLModel):
class GraphQL:
gql_type = "GraphQLTypes"
bindable = PydanticBindable({"GraphQLTypes": GraphQLTypes})
make_executable_schema(
schema,
base_scalars_resolvers,
snake_case_fallback_resolvers,
bindable,
)
book_schema = GraphQLTypes.schema()
book_schema.pop("description")
assert book_schema == {
"properties": {
"id": {
"title": "Id",
"anyOf": [{
"type": "integer"
}, {
"type": "string"
}],
},
"string": {
"title": "String",
"type": "string"
},
"int": {
"title": "Int",
"type": "integer"
},
"float": {
"title": "Float",
"type": "number"
},
"bool": {
"title": "Bool",
"type": "boolean"
},
},
"required": ["int"],
"title": "GraphQLTypes",
"type": "object",
}
GraphQLTypes(int=1)
with pytest.raises(ValueError):
GraphQLTypes()
def test_exclude():
class Book(GraphQLModel):
class GraphQL:
gql_type = "Book"
exclude = ["title"]
bindable = PydanticBindable({"Book": Book})
make_executable_schema(
schema,
base_scalars_resolvers,
snake_case_fallback_resolvers,
bindable,
)
assert "title" not in Book.schema()["properties"]
def test_binding_errors(book_type, user_type):
class Book(GraphQLModel):
class GraphQL:
gql_type = book_type
class User(GraphQLModel):
class GraphQL:
gql_type = user_type
bindable = PydanticBindable({book_type: Book, user_type: User})
with pytest.raises(PydanticBindError):
make_executable_schema(
schema,
base_scalars_resolvers,
snake_case_fallback_resolvers,
bindable,
)
def test_referencing_error(has_borrowed, fav_book):
class Book(GraphQLModel):
class GraphQL:
gql_type = "Book"
class User(GraphQLModel):
class GraphQL:
gql_type = "User"
bindable = PydanticBindable({"Book": Book, "User": User})
make_executable_schema(
schema,
base_scalars_resolvers,
snake_case_fallback_resolvers,
bindable,
)
with pytest.raises(ValueError):
User(has_borrowed=has_borrowed, fav_book=fav_book)
def test_type_override():
class Book(GraphQLModel):
class GraphQL:
gql_type = "Book"
fields = {"id": UUID4}
bindable = PydanticBindable({"Book": Book})
make_executable_schema(
schema,
base_scalars_resolvers,
snake_case_fallback_resolvers,
bindable,
)
assert Book.schema()["properties"]["id"] == {
"title": "Id",
"type": "string",
"format": "uuid4",
}
def test_include():
class Book(GraphQLModel):
class GraphQL:
gql_type = "Book"
include = ["title"]
bindable = PydanticBindable({"Book": Book})
make_executable_schema(
schema,
base_scalars_resolvers,
snake_case_fallback_resolvers,
bindable,
)
assert Book.schema()["properties"] == {
"title": {
"title": "Title",
"type": "string"
}
}
def test_referencing():
class Book(GraphQLModel):
class GraphQL:
gql_type = "Book"
class User(GraphQLModel):
class GraphQL:
gql_type = "User"
bindable = PydanticBindable({"Book": Book, "User": User})
make_executable_schema(
schema,
base_scalars_resolvers,
snake_case_fallback_resolvers,
bindable,
)
book = {"title": "random"}
User(has_borrowed=[book], favBook=book)
with pytest.raises(ValueError):
User(has_borrowed=book, favBook=[book])
with pytest.raises(ValueError):
User(has_borrowed=1, favBook=1)