Initial commit

env/lib/python3.10/site-packages/wagtail/tests/test_hooks.py

@@ -0,0 +1,36 @@
+from django.test import TestCase
+
+from wagtail import hooks
+from wagtail.test.utils import WagtailTestUtils
+
+
+def test_hook():
+    pass
+
+
+class TestLoginView(WagtailTestUtils, TestCase):
+    fixtures = ["test.json"]
+
+    @classmethod
+    def setUpClass(cls):
+        hooks.register("test_hook_name", test_hook)
+
+    @classmethod
+    def tearDownClass(cls):
+        del hooks._hooks["test_hook_name"]
+
+    def test_before_hook(self):
+        def before_hook():
+            pass
+
+        with self.register_hook("test_hook_name", before_hook, order=-1):
+            hook_fns = hooks.get_hooks("test_hook_name")
+            self.assertEqual(hook_fns, [before_hook, test_hook])
+
+    def test_after_hook(self):
+        def after_hook():
+            pass
+
+        with self.register_hook("test_hook_name", after_hook, order=1):
+            hook_fns = hooks.get_hooks("test_hook_name")
+            self.assertEqual(hook_fns, [test_hook, after_hook])