def test_get_homepage(self): menu = Menu(id=1, parent=None, weight=1) self.session.add(menu) page_1 = Page(id=2, parent=menu, weight=1, home=True) self.session.add(page_1) page_2 = Page(id=3, parent=menu, weight=2) self.session.add(page_2) page_3 = Page(id=4, parent=menu, weight=3) self.session.add(page_3) page_4 = Page(id=5, parent=menu, weight=4) self.session.add(page_4) self.assertEqual(page_1, Page.get_homepage(self.session)) page_2.home = True with self.assertRaises(MultipleResultsFound): Page.get_homepage(self.session) page_1.home = False self.assertEqual(page_2, Page.get_homepage(self.session)) page_2.home = False self.assertEqual(page_1, Page.get_homepage(self.session)) page_1.home = False section = Section(id=6, parent=menu, weight=5) self.session.add(section) page_1.parent = section page_2.parent = section page_3.parent = section page_4.parent = section self.assertIn(Page.get_homepage(self.session), (page_1, page_2, page_3, page_4))
def test_get_homepage(self): menu = Menu(id=1, parent=None, weight=1) self.session.add(menu) page_1 = Page(id=2, parent=menu, weight=1) self.session.add(page_1) page_2 = Page(id=3, parent=menu, weight=2) self.session.add(page_2) page_3 = Page(id=4, parent=menu, weight=3) self.session.add(page_3) page_4 = Page(id=5, parent=menu, weight=4, home=True) self.session.add(page_4) it = Language(lang=u'it', country=u'it') self.session.add(it) en = Language(lang=u'en', country=u'gb') self.session.add(en) page_info_1 = PageInfo(id=1, label='Home', title='Pagina Principale', url_part='index', content='', node=page_4, lang=it) self.session.add(page_info_1) page_info_2 = PageInfo(id=2, label='Home', title='Main Page', url_part='index', content='', node=page_4, lang=en) self.session.add(page_info_2) page_info_3 = PageInfo(id=3, label='Home 2', title='Pagina Principale 2', url_part='index', content='', node=page_1, lang=it) self.session.add(page_info_1) page_info_4 = PageInfo(id=4, label='Home 2', title='Main Page 2', url_part='index', content='', node=page_1, lang=en) self.session.add(page_info_2) self.session.flush() self.assertEqual(PageInfo.get_homepage(self.session, it), page_info_1) self.assertEqual(PageInfo.get_homepage(self.session, en), page_info_2) page_4.home = False self.assertEqual(PageInfo.get_homepage(self.session, it), page_info_3) self.assertEqual(PageInfo.get_homepage(self.session, en), page_info_4)