def test_connection(self):
l = CarrotListener(self.ready_queue, self.eta_scheduler, self.logger)
c = l.reset_connection()
self.assertTrue(isinstance(l.amqp_connection, BrokerConnection))
l.close_connection()
self.assertTrue(l.amqp_connection is None)
self.assertTrue(l.task_consumer is None)
c = l.reset_connection()
self.assertTrue(isinstance(l.amqp_connection, BrokerConnection))
l.stop()
self.assertTrue(l.amqp_connection is None)
self.assertTrue(l.task_consumer is None)
def test_receieve_message_eta(self):
l = CarrotListener(self.ready_queue, self.eta_scheduler, self.logger,
send_events=False)
backend = MockBackend()
m = create_message(backend, task=foo_task.name,
args=[2, 4, 8], kwargs={},
eta=(datetime.now() +
timedelta(days=1)).isoformat())
l.reset_connection()
l.receive_message(m.decode(), m)
in_hold = self.eta_scheduler.queue[0]
self.assertEquals(len(in_hold), 4)
eta, priority, task, on_accept = in_hold
self.assertTrue(isinstance(task, TaskWrapper))
self.assertTrue(callable(on_accept))
self.assertEquals(task.task_name, foo_task.name)
self.assertEquals(task.execute(), 2 * 4 * 8)
self.assertRaises(Empty, self.ready_queue.get_nowait)
