def test_get_all_due_events(self):
patient = Patient()
patient.save()
# The fixtures already contain two due events
self.assertEquals(scheduler.get_all_due_events().count(), 2)
sendable = InfoMessage(text="Test Message",
way_of_communication = get_woc("sms"))
sendable.recipient = patient
sendable.save()
schedule1 = ScheduledEvent(sendable=sendable, send_time=datetime.now())
schedule1.save()
schedule2 = ScheduledEvent(sendable=sendable,
send_time=(datetime.now() - timedelta(days=1)))
schedule2.save()
schedule3 = ScheduledEvent(sendable=sendable,
send_time=(datetime.now() + timedelta(days=1)))
schedule3.save()
self.assertEquals(scheduler.get_all_due_events().count(), 4)
self.assertTrue(schedule1 in scheduler.get_all_due_events())
self.assertTrue(schedule2 in scheduler.get_all_due_events())
self.assertFalse(schedule3 in scheduler.get_all_due_events())
schedule4 = ScheduledEvent(sendable=sendable,
send_time=datetime.now(),
state = "failed")
schedule4.save()
schedule5 = ScheduledEvent(sendable=sendable,
send_time=(datetime.now() - timedelta(days=1)),
state = "sent")
schedule5.save()
self.assertEquals(scheduler.get_all_due_events().count(), 4)
schedule1.delete()
schedule2.delete()
schedule3.delete()
schedule4.delete()
schedule5.delete()
def test_get_all_queued_events(self):
patient = Patient()
patient.save()
sendable = InfoMessage(text="Test Message",
way_of_communication = get_woc("sms"))
sendable.recipient = patient
sendable.save()
self.assertEquals(scheduler.get_all_queued_events().count(), 0)
schedule1 = ScheduledEvent(sendable=sendable,
send_time=datetime.now(),
state = "queued")
schedule1.save()
self.assertEquals(scheduler.get_all_queued_events().count(), 1)
schedule2 = ScheduledEvent(sendable=sendable,
send_time=(datetime.now() - timedelta(days=1)),
state = "sent")
schedule2.save()
self.assertTrue(schedule1 in scheduler.get_all_queued_events())
self.assertFalse(schedule2 in scheduler.get_all_queued_events())
self.assertEquals(scheduler.get_all_queued_events().count(), 1)
schedule2.state = "queued"
schedule2.save()
self.assertEquals(scheduler.get_all_queued_events().count(), 2)
schedule1.delete()
schedule2.delete()
def test_check_spool_files(self):
def get_mock_status(filename):
"""
This is a mock method to represent check_spoolfile_status
It is told which status to return by using the (in this case
useless) filename parameter
"""
return filename
original_get_spoolfile_status = scheduler.get_spoolfile_status
scheduler.get_spoolfile_status = get_mock_status
patient = Patient()
patient.save()
sendable = InfoMessage(text="Test Message",
way_of_communication = get_woc("sms"))
sendable.recipient = patient
sendable.save()
event1 = ScheduledEvent(sendable=sendable,
send_time=datetime.now(),
state = "queued",
filename = "Completed")
event1.save()
event2 = ScheduledEvent(sendable=sendable,
send_time=datetime.now(),
state = "queued",
filename = "Expired")
event2.save()
# now: the real testing
scheduler.check_spool_files()
self.assertEquals(ScheduledEvent.objects.get(pk = event1.pk).state, \
"done")
self.assertEquals(ScheduledEvent.objects.get(pk = event2.pk).state, \
"new")
event2.filename = "Expired"
event2.save()
scheduler.check_spool_files()
self.assertEquals(ScheduledEvent.objects.get(pk = event2.pk).state, \
"new")
self.assertEquals(ScheduledEvent.objects.get(pk = event2.pk).retry, 1)
event2.retry = settings.ASTERISK_RETRY
event2.filename = "Expired"
event2.save()
scheduler.check_spool_files()
self.assertEquals(ScheduledEvent.objects.get(pk = event2.pk).state, \
"failed")
event3 = ScheduledEvent(sendable=sendable,
send_time=datetime.now(),
state = "queued",
filename = "Failed")
event3.save()
scheduler.check_spool_files()
self.assertEquals(ScheduledEvent.objects.get(pk = event3.pk).state, \
"failed")
# change everything back to normal
event1.delete()
event2.delete()
event3.delete()
scheduler.get_spoolfile_status = original_get_spoolfile_status