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))