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)