mirror of
https://github.com/graphql-python/graphene-django.git
synced 2024-11-24 18:44:08 +03:00
9a773b9d7b
* Use ruff in pre-commit * Add pyupgrade * Add isort * Add bugbear * Fix B015 Pointless comparison * Fix B026 * B018 false positive * Remove flake8 and isort config from setup.cfg * Remove black and flake8 from dev dependencies * Update black * Show list of fixes applied with autofix on * Fix typo * Add C4 flake8-comprehensions * Add ruff to dev dependencies * Fix up
315 lines
8.8 KiB
Python
315 lines
8.8 KiB
Python
import pytest
|
|
|
|
import graphene
|
|
from graphene.relay import Node
|
|
from graphene_django import DjangoConnectionField, DjangoObjectType
|
|
|
|
from ...tests.models import Reporter
|
|
from ..middleware import DjangoDebugMiddleware
|
|
from ..types import DjangoDebug
|
|
|
|
|
|
class context:
|
|
pass
|
|
|
|
|
|
def test_should_query_field():
|
|
r1 = Reporter(last_name="ABA")
|
|
r1.save()
|
|
r2 = Reporter(last_name="Griffin")
|
|
r2.save()
|
|
|
|
class ReporterType(DjangoObjectType):
|
|
class Meta:
|
|
model = Reporter
|
|
interfaces = (Node,)
|
|
fields = "__all__"
|
|
|
|
class Query(graphene.ObjectType):
|
|
reporter = graphene.Field(ReporterType)
|
|
debug = graphene.Field(DjangoDebug, name="_debug")
|
|
|
|
def resolve_reporter(self, info, **args):
|
|
return Reporter.objects.first()
|
|
|
|
query = """
|
|
query ReporterQuery {
|
|
reporter {
|
|
lastName
|
|
}
|
|
_debug {
|
|
sql {
|
|
rawSql
|
|
}
|
|
}
|
|
}
|
|
"""
|
|
expected = {
|
|
"reporter": {"lastName": "ABA"},
|
|
"_debug": {"sql": [{"rawSql": str(Reporter.objects.order_by("pk")[:1].query)}]},
|
|
}
|
|
schema = graphene.Schema(query=Query)
|
|
result = schema.execute(
|
|
query, context_value=context(), middleware=[DjangoDebugMiddleware()]
|
|
)
|
|
assert not result.errors
|
|
assert result.data == expected
|
|
|
|
|
|
@pytest.mark.parametrize("max_limit", [None, 100])
|
|
def test_should_query_nested_field(graphene_settings, max_limit):
|
|
graphene_settings.RELAY_CONNECTION_MAX_LIMIT = max_limit
|
|
|
|
r1 = Reporter(last_name="ABA")
|
|
r1.save()
|
|
r2 = Reporter(last_name="Griffin")
|
|
r2.save()
|
|
r2.pets.add(r1)
|
|
r1.pets.add(r2)
|
|
|
|
class ReporterType(DjangoObjectType):
|
|
class Meta:
|
|
model = Reporter
|
|
interfaces = (Node,)
|
|
fields = "__all__"
|
|
|
|
class Query(graphene.ObjectType):
|
|
reporter = graphene.Field(ReporterType)
|
|
debug = graphene.Field(DjangoDebug, name="_debug")
|
|
|
|
def resolve_reporter(self, info, **args):
|
|
return Reporter.objects.first()
|
|
|
|
query = """
|
|
query ReporterQuery {
|
|
reporter {
|
|
lastName
|
|
pets { edges { node {
|
|
lastName
|
|
pets { edges { node { lastName } } }
|
|
} } }
|
|
}
|
|
_debug {
|
|
sql {
|
|
rawSql
|
|
}
|
|
}
|
|
}
|
|
"""
|
|
expected = {
|
|
"reporter": {
|
|
"lastName": "ABA",
|
|
"pets": {
|
|
"edges": [
|
|
{
|
|
"node": {
|
|
"lastName": "Griffin",
|
|
"pets": {"edges": [{"node": {"lastName": "ABA"}}]},
|
|
}
|
|
}
|
|
]
|
|
},
|
|
}
|
|
}
|
|
schema = graphene.Schema(query=Query)
|
|
result = schema.execute(
|
|
query, context_value=context(), middleware=[DjangoDebugMiddleware()]
|
|
)
|
|
assert not result.errors
|
|
query = str(Reporter.objects.order_by("pk")[:1].query)
|
|
assert result.data["_debug"]["sql"][0]["rawSql"] == query
|
|
assert "COUNT" in result.data["_debug"]["sql"][1]["rawSql"]
|
|
assert "tests_reporter_pets" in result.data["_debug"]["sql"][2]["rawSql"]
|
|
assert "COUNT" in result.data["_debug"]["sql"][3]["rawSql"]
|
|
assert "tests_reporter_pets" in result.data["_debug"]["sql"][4]["rawSql"]
|
|
assert len(result.data["_debug"]["sql"]) == 5
|
|
|
|
assert result.data["reporter"] == expected["reporter"]
|
|
|
|
|
|
def test_should_query_list():
|
|
r1 = Reporter(last_name="ABA")
|
|
r1.save()
|
|
r2 = Reporter(last_name="Griffin")
|
|
r2.save()
|
|
|
|
class ReporterType(DjangoObjectType):
|
|
class Meta:
|
|
model = Reporter
|
|
interfaces = (Node,)
|
|
fields = "__all__"
|
|
|
|
class Query(graphene.ObjectType):
|
|
all_reporters = graphene.List(ReporterType)
|
|
debug = graphene.Field(DjangoDebug, name="_debug")
|
|
|
|
def resolve_all_reporters(self, info, **args):
|
|
return Reporter.objects.all()
|
|
|
|
query = """
|
|
query ReporterQuery {
|
|
allReporters {
|
|
lastName
|
|
}
|
|
_debug {
|
|
sql {
|
|
rawSql
|
|
}
|
|
}
|
|
}
|
|
"""
|
|
expected = {
|
|
"allReporters": [{"lastName": "ABA"}, {"lastName": "Griffin"}],
|
|
"_debug": {"sql": [{"rawSql": str(Reporter.objects.all().query)}]},
|
|
}
|
|
schema = graphene.Schema(query=Query)
|
|
result = schema.execute(
|
|
query, context_value=context(), middleware=[DjangoDebugMiddleware()]
|
|
)
|
|
assert not result.errors
|
|
assert result.data == expected
|
|
|
|
|
|
@pytest.mark.parametrize("max_limit", [None, 100])
|
|
def test_should_query_connection(graphene_settings, max_limit):
|
|
graphene_settings.RELAY_CONNECTION_MAX_LIMIT = max_limit
|
|
|
|
r1 = Reporter(last_name="ABA")
|
|
r1.save()
|
|
r2 = Reporter(last_name="Griffin")
|
|
r2.save()
|
|
|
|
class ReporterType(DjangoObjectType):
|
|
class Meta:
|
|
model = Reporter
|
|
interfaces = (Node,)
|
|
fields = "__all__"
|
|
|
|
class Query(graphene.ObjectType):
|
|
all_reporters = DjangoConnectionField(ReporterType)
|
|
debug = graphene.Field(DjangoDebug, name="_debug")
|
|
|
|
def resolve_all_reporters(self, info, **args):
|
|
return Reporter.objects.all()
|
|
|
|
query = """
|
|
query ReporterQuery {
|
|
allReporters(first:1) {
|
|
edges {
|
|
node {
|
|
lastName
|
|
}
|
|
}
|
|
}
|
|
_debug {
|
|
sql {
|
|
rawSql
|
|
}
|
|
}
|
|
}
|
|
"""
|
|
expected = {"allReporters": {"edges": [{"node": {"lastName": "ABA"}}]}}
|
|
schema = graphene.Schema(query=Query)
|
|
result = schema.execute(
|
|
query, context_value=context(), middleware=[DjangoDebugMiddleware()]
|
|
)
|
|
assert not result.errors
|
|
assert result.data["allReporters"] == expected["allReporters"]
|
|
assert len(result.data["_debug"]["sql"]) == 2
|
|
assert "COUNT" in result.data["_debug"]["sql"][0]["rawSql"]
|
|
query = str(Reporter.objects.all()[:1].query)
|
|
assert result.data["_debug"]["sql"][1]["rawSql"] == query
|
|
|
|
|
|
@pytest.mark.parametrize("max_limit", [None, 100])
|
|
def test_should_query_connectionfilter(graphene_settings, max_limit):
|
|
graphene_settings.RELAY_CONNECTION_MAX_LIMIT = max_limit
|
|
|
|
from ...filter import DjangoFilterConnectionField
|
|
|
|
r1 = Reporter(last_name="ABA")
|
|
r1.save()
|
|
r2 = Reporter(last_name="Griffin")
|
|
r2.save()
|
|
|
|
class ReporterType(DjangoObjectType):
|
|
class Meta:
|
|
model = Reporter
|
|
interfaces = (Node,)
|
|
fields = "__all__"
|
|
|
|
class Query(graphene.ObjectType):
|
|
all_reporters = DjangoFilterConnectionField(ReporterType, fields=["last_name"])
|
|
s = graphene.String(resolver=lambda *_: "S")
|
|
debug = graphene.Field(DjangoDebug, name="_debug")
|
|
|
|
def resolve_all_reporters(self, info, **args):
|
|
return Reporter.objects.all()
|
|
|
|
query = """
|
|
query ReporterQuery {
|
|
allReporters(first:1) {
|
|
edges {
|
|
node {
|
|
lastName
|
|
}
|
|
}
|
|
}
|
|
_debug {
|
|
sql {
|
|
rawSql
|
|
}
|
|
}
|
|
}
|
|
"""
|
|
expected = {"allReporters": {"edges": [{"node": {"lastName": "ABA"}}]}}
|
|
schema = graphene.Schema(query=Query)
|
|
result = schema.execute(
|
|
query, context_value=context(), middleware=[DjangoDebugMiddleware()]
|
|
)
|
|
assert not result.errors
|
|
assert result.data["allReporters"] == expected["allReporters"]
|
|
assert len(result.data["_debug"]["sql"]) == 2
|
|
assert "COUNT" in result.data["_debug"]["sql"][0]["rawSql"]
|
|
query = str(Reporter.objects.all()[:1].query)
|
|
assert result.data["_debug"]["sql"][1]["rawSql"] == query
|
|
|
|
|
|
def test_should_query_stack_trace():
|
|
class ReporterType(DjangoObjectType):
|
|
class Meta:
|
|
model = Reporter
|
|
interfaces = (Node,)
|
|
fields = "__all__"
|
|
|
|
class Query(graphene.ObjectType):
|
|
reporter = graphene.Field(ReporterType)
|
|
debug = graphene.Field(DjangoDebug, name="_debug")
|
|
|
|
def resolve_reporter(self, info, **args):
|
|
raise Exception("caught stack trace")
|
|
|
|
query = """
|
|
query ReporterQuery {
|
|
reporter {
|
|
lastName
|
|
}
|
|
_debug {
|
|
exceptions {
|
|
message
|
|
stack
|
|
}
|
|
}
|
|
}
|
|
"""
|
|
schema = graphene.Schema(query=Query)
|
|
result = schema.execute(
|
|
query, context_value=context(), middleware=[DjangoDebugMiddleware()]
|
|
)
|
|
assert result.errors
|
|
assert len(result.data["_debug"]["exceptions"])
|
|
debug_exception = result.data["_debug"]["exceptions"][0]
|
|
assert debug_exception["stack"].count("\n") > 1
|
|
assert "test_query.py" in debug_exception["stack"]
|
|
assert debug_exception["message"] == "caught stack trace"
|