graphql-python/graphene-django
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene-django.git synced 2025-07-04 12:23:13 +03:00
Code Issues Packages Projects Releases Wiki Activity

update query test for multi table inheritance

Browse Source
This commit is contained in:
Tom Dror 2023-01-10 13:04:59 -05:00
parent d009f3502e
commit 54f48b3e99
1 changed files with 43 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
graphene_django/tests/test_query.py
View File

@ -15,7 +15,7 @@ from ..compat import IntegerRangeField, MissingType
from ..fields import DjangoConnectionField from ..fields import DjangoConnectionField
from ..types import DjangoObjectType from ..types import DjangoObjectType
from ..utils import DJANGO_FILTER_INSTALLED from ..utils import DJANGO_FILTER_INSTALLED
from .models import Article, CNNReporter, Film, FilmDetails, Person, Pet, Reporter from .models import Article, CNNReporter, Film, FilmDetails, Person, Pet, Reporter, APNewsReporter
def test_should_query_only_fields(): def test_should_query_only_fields():
@ -1067,9 +1067,9 @@ def test_proxy_model_support():
assert result.data == expected assert result.data == expected
def test_proxy_model_support_reverse_relationships(): def test_model_inheritance_support_reverse_relationships():
""" """
This test asserts that we can query reverse relationships for all Reporters and proxied Reporters. This test asserts that we can query reverse relationships for all Reporters and proxied Reporters and multi table Reporters.
""" """
class FilmType(DjangoObjectType): class FilmType(DjangoObjectType):
@ -1091,6 +1091,13 @@ def test_proxy_model_support_reverse_relationships():
use_connection = True use_connection = True
fields = "__all__" fields = "__all__"
class APNewsReporterType(DjangoObjectType):
class Meta:
model = APNewsReporter
interfaces = (Node,)
use_connection = True
fields = "__all__"
film = Film.objects.create(genre="do") film = Film.objects.create(genre="do")
reporter = Reporter.objects.create( reporter = Reporter.objects.create(
@ -1105,12 +1112,17 @@ def test_proxy_model_support_reverse_relationships():
reporter_type=2, # set this guy to be CNN reporter_type=2, # set this guy to be CNN
) )
film.reporters.add(cnn_reporter) ap_news_reporter = APNewsReporter.objects.create(
first_name="John", last_name="Doe", email="johndoe@example.com", a_choice=1
)
film.reporters.add(cnn_reporter, ap_news_reporter)
film.save() film.save()
class Query(graphene.ObjectType): class Query(graphene.ObjectType):
all_reporters = DjangoConnectionField(ReporterType) all_reporters = DjangoConnectionField(ReporterType)
cnn_reporters = DjangoConnectionField(CNNReporterType) cnn_reporters = DjangoConnectionField(CNNReporterType)
ap_news_reporters = DjangoConnectionField(APNewsReporterType)
schema = graphene.Schema(query=Query) schema = graphene.Schema(query=Query)
query = """ query = """
@ -1135,9 +1147,20 @@ def test_proxy_model_support_reverse_relationships():
} }
} }
} }
apNewsReporters {
edges {
node {
id
films {
id
}
}
}
}
} }
""" """
import pdb; pdb.set_trace()
expected = { expected = {
"allReporters": { "allReporters": {
"edges": [ "edges": [
@ -1153,6 +1176,12 @@ def test_proxy_model_support_reverse_relationships():
"films": [{"id": f"{film.id}"}], "films": [{"id": f"{film.id}"}],
}, },
}, },
{
"node": {
"id": to_global_id("ReporterType", ap_news_reporter.id),
"films": [{"id": f"{film.id}"}],
},
},
] ]
}, },
"cnnReporters": { "cnnReporters": {
@ -1165,6 +1194,16 @@ def test_proxy_model_support_reverse_relationships():
} }
] ]
}, },
"apNewsReporters": {
"edges": [
{
"node": {
"id": to_global_id("APNewsReporterType", ap_news_reporter.id),
"films": [{"id": f"{film.id}"}],
}
}
]
},
} }
result = schema.execute(query) result = schema.execute(query)