def test_ready_and_delete_tasks(self): task_1 = Task( owner=self.taskowner, source="@handle_a", dest="@handle_b", content="some message", ready_time=datetime.utcnow() + timedelta(minutes=10), ) task_1.save() task_2 = Task( owner=self.taskowner, source="*****@*****.**", dest="*****@*****.**", content="some message", ready_time=datetime.utcnow(), ) task_2.save() ready_tasks = pushQueue._get_ready_tasks() self.assertEqual(len(ready_tasks), 1) self.assertEqual(ready_tasks[0].source, "*****@*****.**") pushQueue._delete_processed_tasks(ready_tasks) ready_tasks_after = pushQueue._get_ready_tasks() self.assertEqual(len(ready_tasks_after), 0)
def test_queue_task_creation(self): task_1 = Task(owner=self.taskowner, source="@handle_a", dest="@handle_b", content="some message", ready_time=datetime.now()) task_1.save() ready_tasks = pushQueue._get_ready_tasks() queue_Task = pushQueue._create_task_from_db(ready_tasks[0]) self.assertTrue(queue_Task)
def test_ready_and_delete_tasks(self): task_1 = Task(owner=self.taskowner, source="@handle_a", dest="@handle_b", content="some message", ready_time=datetime.utcnow() + timedelta(minutes=10)) task_1.save() task_2 = Task(owner=self.taskowner, source="*****@*****.**", dest="*****@*****.**", content="some message", ready_time=datetime.utcnow()) task_2.save() ready_tasks = pushQueue._get_ready_tasks() self.assertEqual(len(ready_tasks), 1) self.assertEqual(ready_tasks[0].source, "*****@*****.**") pushQueue._delete_processed_tasks(ready_tasks) ready_tasks_after = pushQueue._get_ready_tasks() self.assertEqual(len(ready_tasks_after), 0)
def test_queue_task_creation(self): task_1 = Task( owner=self.taskowner, source="@handle_a", dest="@handle_b", content="some message", ready_time=datetime.now(), ) task_1.save() ready_tasks = pushQueue._get_ready_tasks() queue_Task = pushQueue._create_task_from_db(ready_tasks[0]) self.assertTrue(queue_Task)