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)