def test_PikaPublisher(self):
options = {"rabbitmq_user": self.rabbitmq_user,
"rabbitmq_pwd": self.rabbitmq_pwd}
publisher = PikaPublisher(self.base_routing_key, **options)
publisher.connect()
publisher.register(self.metric_name, self.num_channels,
self.buffer_size)
for metric_name in publisher.metrics_to_num_channels().keys():
routing_key = "%s:%s" % (self.base_routing_key, metric_name)
self.assertEqual(publisher.channels[routing_key].published_count, 0)
self.assertEqual(publisher.channels[routing_key].message_published,
None)
publisher.publish(metric_name, self.message)
self.assertEqual(publisher.channels[routing_key].published_count, 0)
publisher.publish(metric_name, self.message)
self.assertEqual(publisher.channels[routing_key].published_count, 1)
expected_message = [{
"channel_5": 5, "channel_4": 4, "channel_7": 7,
"channel_6": 6, "channel_1": 1, "channel_0": 0,
"channel_3": 3, "channel_2": 2, "timestamp": 100
}, {
"channel_5": 5, "channel_4": 4, "channel_7": 7,
"channel_6": 6, "channel_1": 1, "channel_0": 0,
"channel_3": 3, "channel_2": 2, "timestamp": 100
}]
published_message = json.loads(
publisher.channels[routing_key].message_published)
self.assertEqual(published_message, expected_message)
def test_PikaPublisher(self):
options = {"rabbitmq_address": self.rabbitmq_address,
"rabbitmq_user": self.rabbitmq_user,
"rabbitmq_pwd": self.rabbitmq_pwd}
publisher = PikaPublisher(self.base_routing_key, **options)
publisher.connect()
publisher.register(self.metric_name, self.num_channels, self.buffer_size)
for metric_name in publisher.metrics_to_num_channels().keys():
routing_key = "%s:%s" % (self.base_routing_key, metric_name)
self.assertEqual(publisher.channels[routing_key].published_count, 0)
self.assertEqual(publisher.channels[routing_key].message_published,
None)
publisher.publish(metric_name, self.message)
self.assertEqual(publisher.channels[routing_key].published_count, 0)
publisher.publish(metric_name, self.message)
self.assertEqual(publisher.channels[routing_key].published_count, 1)
expectedMessage = json.dumps([{
"channel_5": 5, "channel_4": 4, "channel_7": 7,
"channel_6": 6, "channel_1": 1, "channel_0": 0,
"channel_3": 3, "channel_2": 2, "timestamp": 100
}, {
"channel_5": 5, "channel_4": 4, "channel_7": 7,
"channel_6": 6, "channel_1": 1, "channel_0": 0,
"channel_3": 3, "channel_2": 2, "timestamp": 100
}])
self.assertEqual(publisher.channels[routing_key].message_published,
expectedMessage)
