Fixed #6225 -- Added get_root method to Page model (#6226)

czpython · web-flow · commit cfee4ecdf548 · 2018-01-10T11:41:06.000-05:00
diff --git a/cms/models/pagemodel.py b/cms/models/pagemodel.py
@@ -1024,6 +1024,13 @@ def get_descendant_pages(self):
         )
         return pages
 
+    def get_root(self):
+        node = self.node
+        return self.__class__.objects.get(
+            node__path=node.path[0:node.steplen],
+            publisher_is_draft=self.publisher_is_draft,
+        )
+
     def get_parent_page(self):
         if not self.node.parent_id:
             return None
diff --git a/cms/tests/test_page.py b/cms/tests/test_page.py
@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 import datetime
 import os.path
+import functools
 from unittest import skipIf
 
 from django.conf import settings
@@ -66,6 +67,26 @@ def test_absolute_url(self):
             self.assertEqual(page_2.get_absolute_url(language='en'), '/en/inner/')
             self.assertEqual(page_2.get_absolute_url(language='fr'), '/fr/french-inner/')
 
+    def test_get_root_page(self):
+        _create = functools.partial(
+            create_page,
+            template='nav_playground.html',
+            language='en',
+            published=True,
+        )
+        page_a = _create('page_a')
+        page_a_a = _create('page_a_a_a', parent=page_a)
+        page_a_a_a = _create('page_a_a_a', parent=page_a_a)
+        page_tree_with_root = [
+            (page_a, page_a),
+            (page_a_a, page_a),
+            (page_a_a_a, page_a),
+        ]
+
+        for page, root in page_tree_with_root:
+            self.assertEqual(page.get_root(), root)
+            self.assertEqual(page.publisher_public.get_root(), root.publisher_public)
+
     def test_treebeard_delete(self):
         """
         This is a test for #4102
