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