From 805e0785562d7080c8c6f828c639432cac18e4a2 Mon Sep 17 00:00:00 2001
From: Eric Drechsel <eric@pointnineseven.com>
Date: Fri, 21 Nov 2014 13:24:29 -0800
Subject: [PATCH] test that add button is present only if page allows subpages

---
 wagtail/wagtailadmin/tests/test_userbar.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

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 = '<a href="%s" target="_parent" class="action icon icon-plus" title="Add a child page">Add</a>' % 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 = '<a href="%s" target="_parent" class="action icon icon-plus" title="Add another page at this level">Add</a>' % expected_url
-        self.assertContains(response, expected_link)
+        expected_link = '<a href="%s" target="_parent" class="action icon icon-plus" title="Add a child page">Add</a>' % expected_url
+        self.assertNotContains(response, expected_link)
 
 
 class TestUserbarModeration(TestCase, WagtailTestUtils):