def test_add_node(self): """add_node added node""" master = Node(name='Apples', link='misago:index') child = Node(name='Oranges', link='misago:index') master.add_node(child) self.assertTrue(child in master.children())
def test_add_node_before(self): """add_node added node before specific node""" master = Node(name='Apples', link='misago:index') child = Node(name='Oranges', link='misago:index') master.add_node(child) test = Node(name='Potatoes', link='misago:index') master.add_node(test, before='misago:index') all_nodes = master.children() for i, node in enumerate(all_nodes): if node.name == test.name: self.assertEqual(all_nodes[i + 1].name, child.name)