Misago/misago/admin/tests/test_admin_hierarchy.py

from django.test import TestCase

from misago.admin.hierarchy import Node


class NodeTests(TestCase):
    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_after(self):
        """add_node added node after 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, after="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)

    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)