Add tests for flush command

tests/test_management_commands.py

@@ -1,4 +1,5 @@
-from datetime import timedelta
+from datetime import datetime, timedelta
+from io import StringIO
 
 from django.conf import settings
 from django.core import mail
@@ -9,7 +10,7 @@
 
 from timeline_logger.models import TimelineLog
 
-from .factories import ArticleFactory, UserFactory
+from .factories import ArticleFactory, TimelineLogFactory, UserFactory
 
 
 class ReportMailingTestCase(TestCase):
@@ -152,3 +153,32 @@ def test_timeline_digest_from_email_setting(self):
 
         self.assertEqual(len(mail.outbox), 1)
         self.assertEqual(mail.outbox[0].from_email, settings.TIMELINE_DIGEST_FROM_EMAIL)
+
+
+class PruneTimelineLogsTestCase(TestCase):
+    def setUp(self):
+
+        self.log_1 = TimelineLogFactory.create(
+            timestamp=datetime(2024, 3, 1, 12, 0, 0),
+        )
+
+        self.log_2 = TimelineLogFactory.create(timestamp=datetime(2024, 3, 1, 14, 0, 0))
+
+    def test_prune_timeline_logs_no_date(self):
+        call_command(
+            "prune_timeline_logs", interactive=False, verbosity=0, stdout=StringIO()
+        )
+
+        self.assertEqual(TimelineLog.objects.count(), 0)
+
+    def test_prune_timeline_logs_date(self):
+        call_command(
+            "prune_timeline_logs",
+            before=datetime(2024, 3, 1, 13, 0, 0),
+            interactive=False,
+            verbosity=0,
+            stdout=StringIO(),
+        )
+
+        self.assertEqual(TimelineLog.objects.count(), 1)
+        self.assertEqual(TimelineLog.objects.first().pk, self.log_2.pk)