add failing test for wagtail#557

RoPP · Oct 1, 2014 · 2c01c27 · 2c01c27
1 parent 21d3809
commit 2c01c27
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/wagtail/wagtailcore/tests/test_page_queryset.py b/wagtail/wagtailcore/tests/test_page_queryset.py
@@ -260,6 +260,19 @@ def test_type(self):
         event = Page.objects.get(url_path='/home/events/someone-elses-event/')
         self.assertTrue(pages.filter(id=event.id).exists())
 
+    def test_type_includes_subclasses(self):
+        from wagtail.wagtailforms.models import AbstractEmailForm
+        pages = Page.objects.type(AbstractEmailForm)
+
+        # Check that all objects are instances of AbstractEmailForm
+        for page in pages:
+            self.assertIsInstance(page.specific, AbstractEmailForm)
+
+        # Check that the contact form page is in the results
+        contact_us = Page.objects.get(url_path='/home/contact-us/')
+        self.assertTrue(pages.filter(id=contact_us.id).exists())
+
+
     def test_not_type(self):
         pages = Page.objects.not_type(EventPage)