def test_find_child_wildcard(self):
root = Node('root')
root.add_path('a')
root.add_path('{b}')
root.add_path('c')
self.assertEqual(root.find_child('a').name, 'a')
self.assertEqual(root.find_child('c').name, 'c')
self.assertEqual(root.find_child('x').name, '{b}')
self.assertFalse(root.find_child('x', wildcard=False))
class TestNode(unittest.TestCase):
def setUp(self):
# Make a tree like this:
# root
# a h
# b d i n
# c f e g k o
# l m p
self.root = Node('root')
self.root.add_path('a', 'b', 'c')
self.root.add_path('a', 'b', 'f')
self.root.add_path('a', 'd', 'e')
self.root.add_path('a', 'd', 'g')
self.root.add_path('h', 'i', 'k', 'l')
self.root.add_path('h', 'i', 'k', 'm')
self.root.add_path('h', 'i', 'k', 'p')
self.root.add_path('h', 'n', 'o')
def test_illegal_name(self):
self.assertRaises(ValueError, Node, '.')
self.assertRaises(ValueError, Node, '..')
def test_str(self):
node = Node('my node')
self.assertEqual(str(node), 'my node')
def test_cmp_same_type(self):
a = Node('a', data={'type': 'dir'})
b = Node('b', data={'type': 'dir'})
self.assertTrue(a < b)
def test_cmp_different_type(self):
a = Node('a', data={'type': 'file'})
b = Node('b', data={'type': 'dir'})
self.assertTrue(b < a)
def test_eq(self):
a = Node('a', data={'type': 'file'})
b = Node('b', data={'type': 'dir'})
self.assertNotEqual(a, b)
a = Node('a', data={'type': 'file'})
b = Node('a', data={'type': 'file'})
self.assertEqual(a, b)
def test_add_path_and_find_child(self):
# Level 1 (root)
self.assertEqual(set(c.name for c in self.root.children), set('ah'))
# Level 2
node_a = self.root.find_child('a')
node_h = self.root.find_child('h')
self.assertEqual(set(c.name for c in node_a.children), set('bd'))
self.assertEqual(set(c.name for c in node_h.children), set('in'))
# Level 3
node_b = node_a.find_child('b')
node_i = node_h.find_child('i')
self.assertEqual(set(c.name for c in node_b.children), set('cf'))
self.assertEqual(set(c.name for c in node_i.children), set('k'))
# Level 4
node_c = node_b.find_child('c')
node_k = node_i.find_child('k')
self.assertEqual(set(c.name for c in node_c.children), set())
self.assertEqual(set(c.name for c in node_k.children), set('lmp'))
# Return None if child can't be found
self.assertFalse(node_c.find_child('x'))
def test_find_child_wildcard(self):
root = Node('root')
root.add_path('a')
root.add_path('{b}')
root.add_path('c')
self.assertEqual(root.find_child('a').name, 'a')
self.assertEqual(root.find_child('c').name, 'c')
self.assertEqual(root.find_child('x').name, '{b}')
self.assertFalse(root.find_child('x', wildcard=False))
def test_ls(self):
self.assertEqual([n.name for n in self.root.ls('a')], list('bd'))
self.assertEqual([n.name for n in self.root.ls('a', 'b')], list('cf'))
self.assertEqual([n.name for n in self.root.ls('a', 'b', 'c')], [])
self.assertEqual([n.name for n in self.root.ls('h', 'i', 'k')],
list('lmp'))
def test_ls_root(self):
self.assertEqual([n.name for n in self.root.ls()], list('ah'))
def test_ls_non_existing(self):
self.assertEqual([n.name for n in self.root.ls('x')], [])
self.assertEqual([n.name for n in self.root.ls('a', 'b', 'x')], [])
def test_ls_parent(self):
self.assertEqual([n.name for n in self.root.ls('..')], list('ah'))
self.assertEqual([n.name for n in self.root.ls('..', '..', '..')],
list('ah'))
self.assertEqual([n.name for n in self.root.ls('..', '..', 'h')],
list('in'))
self.assertEqual(
[n.name for n in self.root.ls('..', '..', 'h', '..', 'a')],
list('bd'))
def test_ls_dot(self):
self.assertEqual([n.name for n in self.root.ls('.')], list('ah'))
self.assertEqual([n.name for n in self.root.ls('.', '.', '.')],
list('ah'))
self.assertEqual([n.name for n in self.root.ls('.', 'a', 'b')],
list('cf'))
self.assertEqual([n.name for n in self.root.ls('.', 'h', '.')],
list('in'))
self.assertEqual(
[n.name for n in self.root.ls('.', 'h', '.', '.', 'n')], ['o'])
def test_ls_sort_by_types(self):
self.root.add_path('q', 'r')
self.root.add_path('q', 's', node_type='file')
self.root.add_path('q', 't', node_type='file')
self.root.add_path('q', 'u')
self.root.add_path('q', 'v', node_type='file')
self.assertEqual([n.name for n in self.root.ls('q')], list('rustv'))