def test0CreateAndDeleteQueue(self):
rabbit_url = os.environ['RABBITMQ_URL']
queue_name = os.environ['QUEUE_NAME']
messageQ1 = initializeMessageQueue(rabbit_url)
messageQ1.createQueue(queue_name)
messageQ1.deleteQueue(queue_name)
def test1RunMessageQueue(self):
#10.8.50.235:5671/%2f
rabbit_url = os.environ['RABBITMQ_URL']
queue_name = os.environ['QUEUE_NAME']
testMessages = []
messageQ1 = initializeMessageQueue(rabbit_url)
messageQ1.createQueue(queue_name)
for i in range(0,5):
message = Message(value = i)
testMessages.append(message)
messageQ1.sendMessage(queue_name,message)
messageQ2 = initializeMessageQueue(rabbit_url)
sentMsgs = messageQ2.getMessages(queue_name,5)
self.assertTrue(len(sentMsgs) == len(testMessages))
messageQ1.deleteQueue(queue_name)
def test2GetAndProcessMessages(self):
rabbit_url = os.environ['RABBITMQ_URL']
queue_name = os.environ['QUEUE_NAME']
testMessages = []
messageQ1 = initializeMessageQueue(rabbit_url)
messageQ1.createQueue(queue_name)
for i in range(0,5):
message = Message(value = i)
testMessages.append(message)
messageQ1.sendMessage(queue_name,message)
messageQ2 = initializeMessageQueue(rabbit_url)
def processMessage(message):
valid = sets.Set({0,1,2,3,4})
if message.messageKey not in valid:
raise Exception('invalid value '%message.value)
messageQ2.getAndProcessMessages(queue_name,processMessage,True)
