def test_retry_notifications(self, mock_notify): # day_old = timezone.now() - datetime.timedelta(days=1) mock_notify.return_value = 'notified' n = Notification.objects.first() n.timestamp = timezone.now() n.save() msg = notifications.retry_notifications() self.assertTrue(mock_notify.call_count == 1) n.timestamp = n.timestamp - datetime.timedelta(days=4) n.save() msg = notifications.retry_notifications() self.assertTrue('found' in msg)
def test_retry_notifications(self, mock_notify): day_old = timezone.now() - datetime.timedelta(days=1) mock_notify.return_value = 'notified' n = Notification.objects.first() n.timestamp = timezone.now() n.save() msg = notifications.retry_notifications() self.assertTrue(mock_notify.call_count == 1) n.timestamp = n.timestamp - datetime.timedelta(days=4) n.save() msg = notifications.retry_notifications() self.assertTrue('found' in msg)
def handle(self, *args, **options): msg = retry_notifications(days=options['days']) self.stdout.write(msg)