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()