Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
@@ -1,5 +1,13 @@
|
||||
from datetime import timedelta
|
||||
from io import StringIO
|
||||
from unittest.mock import patch
|
||||
|
||||
from django.core.management import call_command
|
||||
from django.test import Client, TestCase, override_settings
|
||||
from django.urls import reverse
|
||||
from django.utils import timezone
|
||||
|
||||
from social.models import SocialPost
|
||||
|
||||
|
||||
class HealthzTests(TestCase):
|
||||
@@ -27,3 +35,43 @@ class PublicSmokeTests(TestCase):
|
||||
client = Client()
|
||||
self.assertEqual(client.get(reverse("public:about")).status_code, 200)
|
||||
self.assertEqual(client.get(reverse("public:contact")).status_code, 200)
|
||||
|
||||
|
||||
class DispatchDueTests(TestCase):
|
||||
def test_enqueues_past_scheduled_social_post(self):
|
||||
post = SocialPost.objects.create(
|
||||
body="Open house",
|
||||
status=SocialPost.Status.SCHEDULED,
|
||||
scheduled_for=timezone.now() - timedelta(minutes=5),
|
||||
)
|
||||
mock_task = patch(
|
||||
"core.management.commands.dispatch_due.publish_social_post"
|
||||
).start()
|
||||
self.addCleanup(patch.stopall)
|
||||
out = StringIO()
|
||||
call_command("dispatch_due", stdout=out)
|
||||
post.refresh_from_db()
|
||||
self.assertEqual(post.status, SocialPost.Status.QUEUED)
|
||||
mock_task.enqueue.assert_called_once_with(post_id=str(post.pk))
|
||||
self.assertIn("Enqueued 1 due item(s).", out.getvalue())
|
||||
|
||||
def test_skips_future_scheduled_social_post(self):
|
||||
post = SocialPost.objects.create(
|
||||
body="Later",
|
||||
status=SocialPost.Status.SCHEDULED,
|
||||
scheduled_for=timezone.now() + timedelta(hours=1),
|
||||
)
|
||||
mock_task = patch(
|
||||
"core.management.commands.dispatch_due.publish_social_post"
|
||||
).start()
|
||||
self.addCleanup(patch.stopall)
|
||||
call_command("dispatch_due", stdout=StringIO())
|
||||
post.refresh_from_db()
|
||||
self.assertEqual(post.status, SocialPost.Status.SCHEDULED)
|
||||
mock_task.enqueue.assert_not_called()
|
||||
|
||||
def test_quiet_when_nothing_due(self):
|
||||
out = StringIO()
|
||||
call_command("dispatch_due", stdout=out)
|
||||
self.assertEqual(out.getvalue(), "")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user