diff --git a/wagtail/wagtailadmin/tests/test_userbar.py b/wagtail/wagtailadmin/tests/test_userbar.py index 04056f87a..4daf9370a 100644 --- a/wagtail/wagtailadmin/tests/test_userbar.py +++ b/wagtail/wagtailadmin/tests/test_userbar.py @@ -78,18 +78,18 @@ class TestUserbarAddLink(TestCase, WagtailTestUtils): def test_page_allowing_subpages(self): response = self.client.get(reverse('wagtailadmin_userbar_frontend', args=(self.event_index.id, ))) - # page allows subpages, so the 'add page' button should add a subpage + # page allows subpages, so the 'add page' button should show expected_url = reverse('wagtailadmin_pages_add_subpage', args=(self.event_index.id, )) expected_link = 'Add' % expected_url self.assertContains(response, expected_link) def test_page_disallowing_subpages(self): response = self.client.get(reverse('wagtailadmin_userbar_frontend', args=(self.business_child.id, ))) - print response - # page disallows subpages, so the 'add page' button should add a page at the same level + + # page disallows subpages, so the 'add page' button shouldn't show expected_url = reverse('wagtailadmin_pages_add_subpage', args=(self.business_index.id, )) - expected_link = 'Add' % expected_url - self.assertContains(response, expected_link) + expected_link = 'Add' % expected_url + self.assertNotContains(response, expected_link) class TestUserbarModeration(TestCase, WagtailTestUtils):