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)