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)