Add wagtail_hooks tests

springload · Apr 6, 2017 · 75d0b32 · 75d0b32
1 parent 153c7f4
commit 75d0b32
Showing 1 changed file with 10 additions and 0 deletions.
diff --git 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(), '<link rel="stylesheet" href="/static/wagtaildraftail/wagtaildraftail.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(), '<script src="/static/wagtaildraftail/wagtaildraftail.js"></script>')
+
+    def test_insert_editor_js(self):
+        hooks = get_hooks('insert_editor_js')
+        self.assertIn(draftail_editor_js, hooks, 'Editor JS should be inserted automatically.')