Guest User

Untitled

a guest
May 26th, 2018
91
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 2.19 KB | None | 0 0
  1. diff --git a/cms/tests/menu.py b/cms/tests/menu.py
  2. index 3b7724b..76a5ee7 100644
  3. --- a/cms/tests/menu.py
  4. +++ b/cms/tests/menu.py
  5. @@ -34,6 +34,11 @@ class MenusTestCase(CMSTestCase):
  6. self.page3 = self.create_page(parent_page=self.page2, published=True, in_navigation=True)
  7. self.page4 = self.create_page(parent_page=None, published=True, in_navigation=True)
  8. self.page5 = self.create_page(parent_page=self.page4, published=True, in_navigation=True)
  9. + self.page6 = self.create_page(parent_page=None, published=True, in_navigation=False)
  10. + self.page7 = self.create_page(parent_page=self.page6, published=True, in_navigation=True)
  11. + self.all_pages = [self.page1, self.page2, self.page3, self.page4,
  12. + self.page5, self.page6, self.page7]
  13. + self.top_level_pages = [self.page1, self.page4, self.page6]
  14.  
  15. def test_01_basic_cms_menu(self):
  16. self.assertEqual(len(menu_pool.menus), 1)
  17. @@ -44,7 +49,7 @@ class MenusTestCase(CMSTestCase):
  18. # test the cms menu class
  19. menu = CMSMenu()
  20. nodes = menu.get_nodes(request)
  21. - self.assertEqual(len(nodes), 5)
  22. + self.assertEqual(len(nodes), len(self.all_pages))
  23.  
  24. def test_02_show_menu(self):
  25.  
  26. @@ -94,7 +99,7 @@ class MenusTestCase(CMSTestCase):
  27. context = self.get_context()
  28. # test standard show_menu
  29. nodes = show_menu(context, 1, 1, 100, 100)['children']
  30. - self.assertEqual(len(nodes), 2)
  31. + self.assertEqual(len(nodes), len(self.top_level_pages))
  32. for node in nodes:
  33. self.assertEqual(len(node.children), 0)
  34.  
  35. @@ -248,4 +253,9 @@ class MenusTestCase(CMSTestCase):
  36. self.assertEqual(len(nodes), 2)
  37. self.assertEqual(nodes[0].get_absolute_url(), page1.get_absolute_url())
  38. self.assertEqual(len(nodes[0].children[0].children), 0)
  39. -
  40. \ No newline at end of file
  41. +
  42. + def test_17_show_submenu_from_non_menu_page(self):
  43. + page6 = Page.objects.get(pk=self.page6.pk)
  44. + context = self.get_context(page6.get_absolute_url())
  45. + nodes = show_menu(context, 1, 100, 0, 1)['children']
  46. + self.assertEqual(len(nodes[0].children), 1)
Add Comment
Please, Sign In to add comment