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 mock_process_events():
for event in scheduler.get_all_queued_events():
event.state = "done"
event.save()
def queued_events_count(state = 'queued'):
return scheduler.get_all_queued_events().count()
