graphene-django/graphene_django/debug/tests/test_query.py

258 lines
7.0 KiB
Python
Raw Normal View History

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(object):
pass
2018-07-20 02:51:33 +03:00
def test_should_query_field():
2018-07-20 02:51:33 +03:00
r1 = Reporter(last_name="ABA")
r1.save()
2018-07-20 02:51:33 +03:00
r2 = Reporter(last_name="Griffin")
r2.save()
class ReporterType(DjangoObjectType):
class Meta:
model = Reporter
2018-07-20 02:51:33 +03:00
interfaces = (Node,)
class Query(graphene.ObjectType):
reporter = graphene.Field(ReporterType)
debug = graphene.Field(DjangoDebug, name="_debug")
2017-07-28 19:43:27 +03:00
def resolve_reporter(self, info, **args):
return Reporter.objects.first()
2018-07-20 02:51:33 +03:00
query = """
query ReporterQuery {
reporter {
lastName
}
_debug {
sql {
rawSql
}
}
}
2018-07-20 02:51:33 +03:00
"""
expected = {
2018-07-20 02:51:33 +03:00
"reporter": {"lastName": "ABA"},
"_debug": {"sql": [{"rawSql": str(Reporter.objects.order_by("pk")[:1].query)}]},
}
2016-09-20 10:08:14 +03:00
schema = graphene.Schema(query=Query)
2018-07-20 02:51:33 +03:00
result = schema.execute(
query, context_value=context(), middleware=[DjangoDebugMiddleware()]
)
assert not result.errors
assert result.data == expected
def test_should_query_nested_field():
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,)
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():
2018-07-20 02:51:33 +03:00
r1 = Reporter(last_name="ABA")
r1.save()
2018-07-20 02:51:33 +03:00
r2 = Reporter(last_name="Griffin")
r2.save()
class ReporterType(DjangoObjectType):
class Meta:
model = Reporter
2018-07-20 02:51:33 +03:00
interfaces = (Node,)
class Query(graphene.ObjectType):
all_reporters = graphene.List(ReporterType)
debug = graphene.Field(DjangoDebug, name="_debug")
2017-07-28 19:43:27 +03:00
def resolve_all_reporters(self, info, **args):
return Reporter.objects.all()
2018-07-20 02:51:33 +03:00
query = """
query ReporterQuery {
allReporters {
lastName
}
_debug {
sql {
rawSql
}
}
}
2018-07-20 02:51:33 +03:00
"""
expected = {
2018-07-20 02:51:33 +03:00
"allReporters": [{"lastName": "ABA"}, {"lastName": "Griffin"}],
"_debug": {"sql": [{"rawSql": str(Reporter.objects.all().query)}]},
}
2016-09-20 10:08:14 +03:00
schema = graphene.Schema(query=Query)
2018-07-20 02:51:33 +03:00
result = schema.execute(
query, context_value=context(), middleware=[DjangoDebugMiddleware()]
)
assert not result.errors
assert result.data == expected
def test_should_query_connection():
2018-07-20 02:51:33 +03:00
r1 = Reporter(last_name="ABA")
r1.save()
2018-07-20 02:51:33 +03:00
r2 = Reporter(last_name="Griffin")
r2.save()
class ReporterType(DjangoObjectType):
class Meta:
model = Reporter
2018-07-20 02:51:33 +03:00
interfaces = (Node,)
class Query(graphene.ObjectType):
all_reporters = DjangoConnectionField(ReporterType)
debug = graphene.Field(DjangoDebug, name="_debug")
2017-07-28 19:43:27 +03:00
def resolve_all_reporters(self, info, **args):
return Reporter.objects.all()
2018-07-20 02:51:33 +03:00
query = """
query ReporterQuery {
allReporters(first:1) {
edges {
node {
lastName
}
}
}
_debug {
sql {
rawSql
}
}
}
2018-07-20 02:51:33 +03:00
"""
expected = {"allReporters": {"edges": [{"node": {"lastName": "ABA"}}]}}
2016-09-20 10:08:14 +03:00
schema = graphene.Schema(query=Query)
2018-07-20 02:51:33 +03:00
result = schema.execute(
query, context_value=context(), middleware=[DjangoDebugMiddleware()]
)
assert not result.errors
2018-07-20 02:51:33 +03:00
assert result.data["allReporters"] == expected["allReporters"]
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_connectionfilter():
from ...filter import DjangoFilterConnectionField
2018-07-20 02:51:33 +03:00
r1 = Reporter(last_name="ABA")
r1.save()
2018-07-20 02:51:33 +03:00
r2 = Reporter(last_name="Griffin")
r2.save()
class ReporterType(DjangoObjectType):
class Meta:
model = Reporter
2018-07-20 02:51:33 +03:00
interfaces = (Node,)
class Query(graphene.ObjectType):
2018-07-20 02:51:33 +03:00
all_reporters = DjangoFilterConnectionField(ReporterType, fields=["last_name"])
s = graphene.String(resolver=lambda *_: "S")
debug = graphene.Field(DjangoDebug, name="_debug")
2017-07-28 19:43:27 +03:00
def resolve_all_reporters(self, info, **args):
return Reporter.objects.all()
2018-07-20 02:51:33 +03:00
query = """
query ReporterQuery {
allReporters(first:1) {
edges {
node {
lastName
}
}
}
_debug {
sql {
rawSql
}
}
}
2018-07-20 02:51:33 +03:00
"""
expected = {"allReporters": {"edges": [{"node": {"lastName": "ABA"}}]}}
2016-09-20 10:08:14 +03:00
schema = graphene.Schema(query=Query)
2018-07-20 02:51:33 +03:00
result = schema.execute(
query, context_value=context(), middleware=[DjangoDebugMiddleware()]
)
assert not result.errors
2018-07-20 02:51:33 +03:00
assert result.data["allReporters"] == expected["allReporters"]
assert "COUNT" in result.data["_debug"]["sql"][0]["rawSql"]
query = str(Reporter.objects.all()[:1].query)
assert result.data["_debug"]["sql"][1]["rawSql"] == query