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_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_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))