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.')