From 4136bb5053165076bdcb7ac5e19ed71ee18fb5cf Mon Sep 17 00:00:00 2001 From: Karl Hobley Date: Mon, 16 Jun 2014 15:15:38 +0100 Subject: [PATCH] Added tests for db6bb92 --- wagtail/wagtailadmin/tests/test_pages_views.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/wagtail/wagtailadmin/tests/test_pages_views.py b/wagtail/wagtailadmin/tests/test_pages_views.py index 12c42f3e4..31301f70f 100644 --- a/wagtail/wagtailadmin/tests/test_pages_views.py +++ b/wagtail/wagtailadmin/tests/test_pages_views.py @@ -188,6 +188,11 @@ class TestPageCreation(TestCase, WagtailTestUtils): self.assertTemplateUsed(response, 'tests/simple_page.html') self.assertContains(response, "New page!") + # Check that the treebeard attributes were set correctly on the page object + self.assertEqual(response.context['self'].depth, self.root_page.depth + 1) + self.assertTrue(response.context['self'].path.startswith(self.root_page.path)) + self.assertEqual(response.context['self'].get_parent(), self.root_page) + class TestPageEdit(TestCase, WagtailTestUtils): def setUp(self):