예제 #1
0
    def test_submenu(self):
        """
        Test making a simple menu.
        """

        s = Site.objects.get(pk=1) # we always have a site: from django
        u = User()
        m = Menu(name="parent", url="/sitemap/", tree_id=s.id, user_id=1)
        m.save()
        m2 = Menu(name="main", url="/", tree_id=s.id, parent=m, user_id=1)
        m2.save()
        m3 = Menu(name="forum", url="/forum/", tree_id=s.id, parent=m, user_id=1)
        m3.save()
        self.assertEqual([m2, m3], m3.submenu(u))