diff --git a/tests/test_wagtail_hooks.py b/tests/test_wagtail_hooks.py index 717fce9..631dc06 100644 --- a/tests/test_wagtail_hooks.py +++ b/tests/test_wagtail_hooks.py @@ -2,6 +2,8 @@ import unittest +from wagtail.wagtailcore.hooks import get_hooks + from wagtaildraftail.wagtail_hooks import draftail_editor_css, draftail_editor_js @@ -10,6 +12,14 @@ def test_editor_css(self): self.assertEqual( draftail_editor_css(), '') + def test_insert_editor_css_hook(self): + hooks = get_hooks('insert_editor_css') + self.assertIn(draftail_editor_css, hooks, 'Editor CSS should be inserted automatically.') + def test_editor_js(self): self.assertEqual( draftail_editor_js(), '') + + def test_insert_editor_js(self): + hooks = get_hooks('insert_editor_js') + self.assertIn(draftail_editor_js, hooks, 'Editor JS should be inserted automatically.')