def test_menuitem_is_none_because_not_published(self): MenuItem.add_root(uri='/a/b/', title='found', site=Site.objects.get_current(), is_published=False) req = self.rf.get('/a/b/') self.mw.process_request(req) with self.assertNumQueries(1): with self.assertRaises(AttributeError): self.assertIsNone(req.menuitem.uri)
def test_menuitem_is_published(self): MenuItem.add_root(uri='/a/b/', title='found', site=Site.objects.get_current(), is_published=True) req = self.rf.get('/a/b/') self.mw.process_request(req) with self.assertNumQueries(1): self.assertEqual(req.menuitem.uri, '/a/b/') self.assertEqual(req.menuitem.title, 'found')