diff --git a/tests/test_grapple.py b/tests/test_grapple.py index 74e5c6fa..cb3ba133 100644 --- a/tests/test_grapple.py +++ b/tests/test_grapple.py @@ -10,7 +10,7 @@ from django.db import connection from django.test import RequestFactory, TestCase, override_settings from graphene.test import Client -from testapp.factories import AdvertFactory, BlogPageFactory +from testapp.factories import AdvertFactory, BlogPageFactory, PersonFactory from testapp.models import GlobalSocialMediaSettings, HomePage, SocialMediaSettings from wagtail.documents import get_document_model from wagtail.models import Page, Site @@ -1587,8 +1587,14 @@ def setUp(self): super().setUp() self.factory = RequestFactory() self.advert = AdvertFactory() + self.person = PersonFactory() def test_snippets(self): + """ + Query for snippets of different types, they should all be returned in + the same response. + """ + query = """ { snippets { @@ -1602,11 +1608,16 @@ def test_snippets(self): self.assertEqual(type(executed["data"]), dict) self.assertEqual(type(executed["data"]["snippets"]), list) + self.assertEqual(len(executed["data"]["snippets"]), 2) self.assertEqual(type(executed["data"]["snippets"][0]), dict) - snippets_data = executed["data"]["snippets"] + snippets_data = sorted( + executed["data"]["snippets"], key=lambda s: s["snippetType"] + ) self.assertEqual(snippets_data[0]["snippetType"], "Advert") self.assertEqual(snippets_data[0]["contentType"], "testapp.Advert") + self.assertEqual(snippets_data[1]["snippetType"], "Person") + self.assertEqual(snippets_data[1]["contentType"], "testapp.Person") def test_no_snippet_classes_registered(self): """