Delete notifications older than 90 days

care/facility/tasks/notification/delete_older_notifications.py

@@ -0,0 +1,15 @@
+from datetime import timedelta
+
+from celery.decorators import periodic_task
+from celery.schedules import crontab
+from django.utils import timezone
+
+from care.facility.models.notification import Notification
+
+
+@periodic_task(
+    run_every=crontab(minute="0", hour="0")
+)  # Run the task daily at midnight
+def delete_old_notifications():
+    ninety_days_ago = timezone.now() - timedelta(days=90)
+    Notification.objects.filter(created_date__lte=ninety_days_ago).delete()

care/facility/tests/test_delete_older_notifications_task.py

@@ -0,0 +1,28 @@
+from datetime import timedelta
+
+from django.test import TestCase
+from django.utils import timezone
+from freezegun import freeze_time
+
+from care.facility.models.notification import Notification
+from care.facility.tasks.notification.delete_older_notifications import (
+    delete_old_notifications,
+)
+
+
+class DeleteOldNotificationsTest(TestCase):
+    def test_delete_old_notifications(self):
+        # notifications created 90 days ago
+        with freeze_time(timezone.now() - timedelta(days=90)):
+            notification1 = Notification.objects.create()
+            notification2 = Notification.objects.create()
+
+        # notification created now
+        notification3 = Notification.objects.create()
+
+        delete_old_notifications()
+
+        # Assert
+        self.assertFalse(Notification.objects.filter(pk=notification1.pk).exists())
+        self.assertFalse(Notification.objects.filter(pk=notification2.pk).exists())
+        self.assertTrue(Notification.objects.filter(pk=notification3.pk).exists())