def test_sendNow(self):
consumer = RabbitQueue(global_session, next(queue_names))
routing_key = RabbitRoutingKey(global_session, next(key_names))
routing_key.associateConsumerNow(consumer)
for data in range(50, 60):
routing_key.sendNow(data)
received_data = consumer.receive(timeout=2)
self.assertEqual(data, received_data)
def test_sendNow(self):
consumer = RabbitQueue(global_session, next(queue_names))
routing_key = RabbitRoutingKey(global_session, next(key_names))
routing_key.associateConsumerNow(consumer)
for data in range(50, 60):
routing_key.sendNow(data)
received_data = consumer.receive(timeout=2)
self.assertEqual(data, received_data)
def test_abort(self):
consumer = RabbitQueue(global_session, next(queue_names))
routing_key = RabbitRoutingKey(global_session, next(key_names))
routing_key.associateConsumerNow(consumer)
for data in range(90, 100):
routing_key.send(data)
# Messages sent using send() are forgotten on abort.
transaction.abort()
self.assertRaises(QueueEmpty, consumer.receive, timeout=2)
def test_abort(self):
consumer = RabbitQueue(global_session, next(queue_names))
routing_key = RabbitRoutingKey(global_session, next(key_names))
routing_key.associateConsumerNow(consumer)
for data in range(90, 100):
routing_key.send(data)
# Messages sent using send() are forgotten on abort.
transaction.abort()
self.assertRaises(QueueEmpty, consumer.receive, timeout=2)
def test_associateConsumerNow(self):
# associateConsumerNow() associates the consumer right away.
consumer = RabbitQueue(global_session, next(queue_names))
routing_key = RabbitRoutingKey(global_session, next(key_names))
routing_key.associateConsumerNow(consumer)
routing_key.sendNow('now')
routing_key.send('later')
# There is already something in the queue.
self.assertEqual('now', consumer.receive(timeout=2))
transaction.commit()
# Now that the transaction has been committed there is another item in
# the queue.
self.assertEqual('later', consumer.receive(timeout=2))
def test_associateConsumerNow(self):
# associateConsumerNow() associates the consumer right away.
consumer = RabbitQueue(global_session, next(queue_names))
routing_key = RabbitRoutingKey(global_session, next(key_names))
routing_key.associateConsumerNow(consumer)
routing_key.sendNow('now')
routing_key.send('later')
# There is already something in the queue.
self.assertEqual('now', consumer.receive(timeout=2))
transaction.commit()
# Now that the transaction has been committed there is another item in
# the queue.
self.assertEqual('later', consumer.receive(timeout=2))
def test_receive(self):
consumer = RabbitQueue(global_session, next(queue_names))
routing_key = RabbitRoutingKey(global_session, next(key_names))
routing_key.associateConsumerNow(consumer)
for data in range(55, 65):
routing_key.sendNow(data)
self.assertEqual(data, consumer.receive(timeout=2))
# All the messages received were consumed.
self.assertRaises(QueueEmpty, consumer.receive, timeout=2)
# New connections to the queue see an empty queue too.
consumer.session.disconnect()
consumer = RabbitQueue(global_session, next(queue_names))
routing_key = RabbitRoutingKey(global_session, next(key_names))
routing_key.associateConsumerNow(consumer)
self.assertRaises(QueueEmpty, consumer.receive, timeout=2)
def test_receive(self):
consumer = RabbitQueue(global_session, next(queue_names))
routing_key = RabbitRoutingKey(global_session, next(key_names))
routing_key.associateConsumerNow(consumer)
for data in range(55, 65):
routing_key.sendNow(data)
self.assertEqual(data, consumer.receive(timeout=2))
# All the messages received were consumed.
self.assertRaises(QueueEmpty, consumer.receive, timeout=2)
# New connections to the queue see an empty queue too.
consumer.session.disconnect()
consumer = RabbitQueue(global_session, next(queue_names))
routing_key = RabbitRoutingKey(global_session, next(key_names))
routing_key.associateConsumerNow(consumer)
self.assertRaises(QueueEmpty, consumer.receive, timeout=2)
def test_send(self):
consumer = RabbitQueue(global_session, next(queue_names))
routing_key = RabbitRoutingKey(global_session, next(key_names))
routing_key.associateConsumerNow(consumer)
for data in range(90, 100):
routing_key.send(data)
routing_key.sendNow('sync')
# There is nothing in the queue except the sync we just sent.
self.assertEqual('sync', consumer.receive(timeout=2))
# Messages get sent on commit
transaction.commit()
for data in range(90, 100):
self.assertEqual(data, consumer.receive())
# There are no more messages. They have all been consumed.
routing_key.sendNow('sync')
self.assertEqual('sync', consumer.receive(timeout=2))
def test_send(self):
consumer = RabbitQueue(global_session, next(queue_names))
routing_key = RabbitRoutingKey(global_session, next(key_names))
routing_key.associateConsumerNow(consumer)
for data in range(90, 100):
routing_key.send(data)
routing_key.sendNow('sync')
# There is nothing in the queue except the sync we just sent.
self.assertEqual('sync', consumer.receive(timeout=2))
# Messages get sent on commit
transaction.commit()
for data in range(90, 100):
self.assertEqual(data, consumer.receive())
# There are no more messages. They have all been consumed.
routing_key.sendNow('sync')
self.assertEqual('sync', consumer.receive(timeout=2))