Added tests for signal firing from publish_scheduled_pages

wagtail/wagtailcore/tests/test_management_commands.py

@@ -7,6 +7,7 @@ from django.core import management
 from django.utils import timezone
 
 from wagtail.wagtailcore.models import Page, PageRevision
+from wagtail.wagtailcore.signals import page_published, page_unpublished
 from wagtail.tests.models import SimplePage
 
 
@@ -96,6 +97,15 @@ class TestPublishScheduledPagesCommand(TestCase):
         self.root_page = Page.objects.get(id=2)
 
     def test_go_live_page_will_be_published(self):
+        # Connect a mock signal handler to page_published signal
+        signal_fired = [False]
+        signal_page = [None]
+        def page_published_handler(sender, instance, **kwargs):
+            signal_fired[0] = True
+            signal_page[0] = instance
+        page_published.connect(page_published_handler)
+
+
         page = SimplePage(
             title="Hello world!",
             slug="hello-world",
@@ -116,6 +126,11 @@ class TestPublishScheduledPagesCommand(TestCase):
         self.assertTrue(p.live)
         self.assertFalse(PageRevision.objects.filter(page=p).exclude(approved_go_live_at__isnull=True).exists())
 
+        # Check that the page_published signal was fired
+        self.assertTrue(signal_fired[0])
+        self.assertEqual(signal_page[0], page)
+        self.assertEqual(signal_page[0], signal_page[0].specific)
+
     def test_future_go_live_page_will_not_be_published(self):
         page = SimplePage(
             title="Hello world!",
@@ -138,6 +153,15 @@ class TestPublishScheduledPagesCommand(TestCase):
         self.assertTrue(PageRevision.objects.filter(page=p).exclude(approved_go_live_at__isnull=True).exists())
 
     def test_expired_page_will_be_unpublished(self):
+        # Connect a mock signal handler to page_unpublished signal
+        signal_fired = [False]
+        signal_page = [None]
+        def page_unpublished_handler(sender, instance, **kwargs):
+            signal_fired[0] = True
+            signal_page[0] = instance
+        page_unpublished.connect(page_unpublished_handler)
+
+
         page = SimplePage(
             title="Hello world!",
             slug="hello-world",
@@ -155,6 +179,11 @@ class TestPublishScheduledPagesCommand(TestCase):
         self.assertFalse(p.live)
         self.assertTrue(p.expired)
 
+        # Check that the page_published signal was fired
+        self.assertTrue(signal_fired[0])
+        self.assertEqual(signal_page[0], page)
+        self.assertEqual(signal_page[0], signal_page[0].specific)
+
     def test_future_expired_page_will_not_be_unpublished(self):
         page = SimplePage(
             title="Hello world!",