def test_unregister_unknown_account_page_class(self):
"""Testing unregister_account_page_class with unknown page."""
class MyPage(AccountPage):
page_id = 'test-page'
page_title = 'Test Page'
with self.assertRaises(AccountPage.registry.lookup_error_class):
unregister_account_page_class(MyPage)
def test_unregister_account_page_class(self):
"""Testing unregister_account_page_class."""
class MyPage(AccountPage):
page_id = 'test-page'
page_title = 'Test Page'
register_account_page_class(MyPage)
unregister_account_page_class(MyPage)
self.assertEqual(set(get_page_classes()), self.builtin_pages)
def test_unregister_account_page_class(self):
"""Testing unregister_account_page_class"""
class MyPage(AccountPage):
page_id = 'test-page'
page_title = 'Test Page'
register_account_page_class(MyPage)
unregister_account_page_class(MyPage)
page_classes = list(get_page_classes())
self.assertEqual(len(page_classes), 4)
def test_unregister_account_page_class(self):
"""Testing unregister_account_page_class"""
class MyPage(AccountPage):
page_id = 'test-page'
page_title = 'Test Page'
register_account_page_class(MyPage)
unregister_account_page_class(MyPage)
page_classes = list(get_page_classes())
self.assertEqual(len(page_classes), 4)
def test_default_form_classes_for_page(self):
"""Testing AccountPage._default_form_classes persistence"""
class MyForm(AccountPageForm):
form_id = 'test-form'
class MyPage(AccountPage):
page_id = 'test-page'
page_title = 'Test Page'
form_classes = [MyForm]
register_account_page_class(MyPage)
self.assertEqual(MyPage.form_classes, [MyForm])
unregister_account_page_class(MyPage)
self.assertEqual(MyPage.form_classes, [])
register_account_page_class(MyPage)
self.assertEqual(MyPage.form_classes, [MyForm])
def test_empty_default_form_classes_for_page(self):
"""Testing AccountPage._default_form_classes with no form_classes"""
class MyPage(AccountPage):
page_id = 'test-page'
page_title = 'Test Page'
class MyForm(AccountPageForm):
form_id = 'test-form'
register_account_page_class(MyPage)
self.assertEqual(MyPage.form_classes, [])
MyPage.add_form(MyForm)
self.assertEqual(MyPage.form_classes, [MyForm])
unregister_account_page_class(MyPage)
self.assertEqual(MyPage.form_classes, [])
register_account_page_class(MyPage)
self.assertEqual(MyPage.form_classes, [])
def test_unregister_unknown_account_page_class(self):
"""Testing unregister_account_page_class with unknown page."""
class MyPage(AccountPage):
page_id = 'test-page'
page_title = 'Test Page'
self.assertRaises(KeyError,
lambda: unregister_account_page_class(MyPage))
def shutdown(self):
super(AccountPagesHook, self).shutdown()
for page_class in self.page_classes:
unregister_account_page_class(page_class)
def shutdown(self):
super(AccountPagesHook, self).shutdown()
for page_class in self.page_classes:
unregister_account_page_class(page_class)
