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), [])
