def test_get_edit_handler(self):
"""
Forms for pages should have a base class of WagtailAdminPageForm.
"""
EditHandler = EventPage.get_edit_handler()
EventPageForm = EditHandler.get_form_class(EventPage)
# The generated form should inherit from WagtailAdminPageForm
self.assertTrue(issubclass(EventPageForm, WagtailAdminPageForm))
def test_get_edit_handler(self):
"""
Forms for pages should have a base class of WagtailAdminPageForm.
"""
EditHandler = EventPage.get_edit_handler()
EventPageForm = EditHandler.get_form_class(EventPage)
# The generated form should inherit from WagtailAdminPageForm
self.assertTrue(issubclass(EventPageForm, WagtailAdminPageForm))
