def _test_get_nav_tree_only_root_folders(self): from petrel.utils import get_nav_tree folder1 = self._makeFolder('folder1') expected = [('folder1', [])] self.assertTreesEqual(get_nav_tree(self.site), expected) self.assertTreesEqual(get_nav_tree(folder1), expected) folder2 = self._makeFolder('folder2') expected = [('folder1', []), ('folder2', [])] self.assertTreesEqual(get_nav_tree(self.site), expected) self.assertTreesEqual(get_nav_tree(folder1), expected) self.assertTreesEqual(get_nav_tree(folder2), expected)
def _test_get_nav_tree_deep_document(self): from petrel.utils import get_nav_tree self._makePath('folder1/folder11/doc111') self._makePath('folder1/folder11/doc112') self._makePath('folder1/folder12/doc121') self._makePath('folder2/doc21') folder1 = self.site['folder1'] folder11 = folder1['folder11'] doc111 = folder11['doc111'] doc112 = folder11['doc112'] expected = [('folder1', []), ('folder2', [])] self.assertTreesEqual(get_nav_tree(self.site), expected) self.assertTreesEqual(get_nav_tree(folder1), expected) expected = [('folder1', []), ('folder2', [('folder11', [('doc111', []), ('doc112', [])])])] self.assertTreesEqual(get_nav_tree(folder11), expected) self.assertTreesEqual(get_nav_tree(doc111), expected) self.assertTreesEqual(get_nav_tree(doc112), expected)
def _test_get_nav_tree_empty_site(self): from petrel.utils import get_nav_tree self.assertEqual(get_nav_tree(self.site), [])