def show_queues():
queues = get_queues(_get_properties())
total_messages = 0
for queue in queues:
print "queue %s: %s messages" % (queue, _queue_size(queue))
total_messages += _queue_size(queue)
print "total messages: %s" % total_messages
def show_queues():
queues = get_queues(_get_properties())
total_messages = 0
for queue in queues:
print "queue %s: %s messages" % (queue, _queue_size(queue))
total_messages += _queue_size(queue)
print "total messages: %s" % total_messages
def test_consume_from_2_queues(self):
"""
Check that worker consumes messages from 2 queues properly
"""
self.counter = 0
def show_queues():
queues = get_queues(_get_properties())
total_messages = 0
for queue in queues:
print "queue %s: %s messages" % (queue, _queue_size(queue))
total_messages += _queue_size(queue)
print "total messages: %s" % total_messages
def check_queue_size(*args, **kwargs):
"""
Closure to attach to _dispatch
counts how many messages processed
"""
self.counter += 1
#SetUp the TaskBroker but override _dispatch
task_broker = _task_broker_factory(dispatch_func=check_queue_size)
#Publish a Couple of Messages to both queues
channel = task_broker.channel
properties = {DEVICE_GROUP: "test", DEVICE_NAME: "testname"}
queues = get_queues(properties)
self.assertEquals(len(queues), 2)
for queue in queues:
foo_cmd_msg = CommandMessage(['foo'], '', 1, timeout=1)
foo_msg = pack_message(foo_cmd_msg)
channel.basic_publish(foo_msg,
mandatory=True,
exchange=queue,
routing_key=queue)
#
bar_cmd_msg = CommandMessage(['bar'], '', 1, timeout=1)
bar_msg = pack_message(bar_cmd_msg)
channel.basic_publish(bar_msg,
mandatory=True,
exchange=queue,
routing_key=queue)
#
baz_cmd_msg = CommandMessage(['bar'], '', 1, timeout=1)
baz_msg = pack_message(baz_cmd_msg)
channel.basic_publish(baz_msg,
mandatory=True,
exchange=queue,
routing_key=queue)
#Set to Consume
task_broker._start_consume()
while self.counter < 6: # Process all messages
channel.wait()
for queue in queues: # Make sure all queues are empty
self.assertEquals(_queue_size(queue), 0)
def test_get_queues_device_group_name_and_id(self):
properties = dict()
properties["devicegroup"] = "test"
properties["devicename"] = "testname"
properties["deviceid"] = "hw1"
queues = get_queues(properties)
self.assertEquals(queues, [ "test.testname.hw1", "test.testname", "test"])
def test_get_queues_device_group_name_and_id(self):
properties = dict()
properties["devicegroup"] = "test"
properties["devicename"] = "testname"
properties["deviceid"] = "hw1"
queues = get_queues(properties)
self.assertEquals(queues,
["test.testname.hw1", "test.testname", "test"])
def test_get_queues_device_group_only(self):
properties = dict()
properties["devicegroup"] = "test"
queues = get_queues(properties)
self.assertEquals(queues, ["test"])
def test_consume_from_2_queues(self):
"""
Check that worker consumes messages from 2 queues properly
"""
self.counter = 0
def show_queues():
queues = get_queues(_get_properties())
total_messages = 0
for queue in queues:
print "queue %s: %s messages" % (queue, _queue_size(queue))
total_messages += _queue_size(queue)
print "total messages: %s" % total_messages
def check_queue_size(*args,**kwargs):
"""
Closure to attach to _dispatch
counts how many messages processed
"""
self.counter += 1
#SetUp the TaskBroker but override _dispatch
task_broker = _task_broker_factory(dispatch_func = check_queue_size)
#Publish a Couple of Messages to both queues
channel = task_broker.channel
properties = {DEVICE_GROUP : "test",
DEVICE_NAME : "testname"}
queues = get_queues(properties)
self.assertEquals(len(queues), 2)
for queue in queues:
foo_cmd_msg = CommandMessage(['foo'],'', 1, timeout = 1)
foo_msg = pack_message(foo_cmd_msg)
channel.basic_publish(foo_msg,
mandatory = True,
exchange = queue,
routing_key = queue)
#
bar_cmd_msg = CommandMessage(['bar'],'', 1, timeout = 1)
bar_msg = pack_message(bar_cmd_msg)
channel.basic_publish(bar_msg,
mandatory = True,
exchange = queue,
routing_key = queue)
#
baz_cmd_msg = CommandMessage(['bar'],'', 1, timeout = 1)
baz_msg = pack_message(baz_cmd_msg)
channel.basic_publish(baz_msg,
mandatory = True,
exchange = queue,
routing_key = queue)
#Set to Consume
task_broker._start_consume()
while self.counter < 6: # Process all messages
channel.wait()
for queue in queues: # Make sure all queues are empty
self.assertEquals(_queue_size(queue), 0)
def tearDown(self):
queues = get_queues(_get_properties())
queues.append("test_response_queue")
for queue in queues:
_queue_delete(queue)
def setUp(self):
queues = get_queues(_get_properties())
queues.append("test_response_queue")
for queue in queues:
_queue_delete(queue)
self.assertTrue(_queue_size(queue) is None)
def tearDown(self):
queues = get_queues(_get_properties())
queues.append("test_response_queue")
for queue in queues:
_queue_delete(queue)
def setUp(self):
queues = get_queues(_get_properties())
queues.append("test_response_queue")
for queue in queues:
_queue_delete(queue)
self.assertTrue(_queue_size(queue) is None)
def test_get_queues_device_group_only(self):
properties = dict()
properties["devicegroup"] = "test"
queues = get_queues(properties)
self.assertEquals(queues, ["test"])
