def tearDown(self): from wagtail.tests.testapp.models import DefaultRichBlockFieldPage from wagtail.tests.testapp.models import DefaultRichTextFieldPage DefaultRichTextFieldPage.get_edit_handler()._form_class = None block_page_edit_handler = DefaultRichBlockFieldPage.get_edit_handler() if block_page_edit_handler._form_class: rich_text_block = block_page_edit_handler._form_class.base_fields['body'].block.child_blocks['rich_text'] if hasattr(rich_text_block, 'field'): del rich_text_block.field block_page_edit_handler._form_class = None
def tearDown(self): from wagtail.tests.testapp.models import DefaultRichBlockFieldPage from wagtail.tests.testapp.models import DefaultRichTextFieldPage DefaultRichTextFieldPage.get_edit_handler()._form_class = None block_page_edit_handler = DefaultRichBlockFieldPage.get_edit_handler() if block_page_edit_handler._form_class: rich_text_block = block_page_edit_handler._form_class.base_fields[ 'body'].block.child_blocks['rich_text'] if hasattr(rich_text_block, 'field'): del rich_text_block.field block_page_edit_handler._form_class = None
def _clear_edit_handler_cache(self): """ These tests generate new EditHandlers with different settings. The cached edit handlers should be cleared before and after each test run to ensure that no changes leak through to other tests. """ from wagtail.tests.testapp.models import DefaultRichBlockFieldPage rich_text_block = (DefaultRichBlockFieldPage.get_edit_handler( ).get_form_class().base_fields['body'].block.child_blocks['rich_text']) if hasattr(rich_text_block, 'field'): del rich_text_block.field for page_class in get_page_models(): page_class.get_edit_handler.cache_clear()
def _clear_edit_handler_cache(self): """ These tests generate new EditHandlers with different settings. The cached edit handlers should be cleared before and after each test run to ensure that no changes leak through to other tests. """ from wagtail.tests.testapp.models import DefaultRichBlockFieldPage rich_text_block = (DefaultRichBlockFieldPage.get_edit_handler() .get_form_class().base_fields['body'].block .child_blocks['rich_text']) if hasattr(rich_text_block, 'field'): del rich_text_block.field for page_class in get_page_models(): page_class.get_edit_handler.cache_clear()