def test_queue_message_handler_stops(self):
received = {"msgs": []}
def cb(msg):
received["msgs"].append(msg)
handler = subscribe.QueueMessageHandler(subscribe.MessageHandler(None, cb))
self.assertTrue(handler.is_alive())
handler.finish()
self.assertFalse(handler.is_alive())
def test_queue_message_handler_passes_msgs(self):
received = {"msgs": []}
def cb(msg):
received["msgs"].append(msg)
handler = subscribe.QueueMessageHandler(subscribe.MessageHandler(None, cb))
handler.queue_length = 1000
msgs = range(1000)
for x in msgs:
handler.handle_message(x)
time.sleep(0.1)
handler.finish()
self.assertEqual(msgs, received["msgs"])
def test_queue_message_handler_passes_msgs(self):
handler = subscribe.QueueMessageHandler(
subscribe.MessageHandler(None, self.dummy_cb))
self.help_test_queue(handler, 1000)
handler.finish()
