def test_publish_metric(self):
datapoint = ("vumi.test.v1", 1.0, 1234)
client = WorkerHelper.get_fake_amqp_client(self.worker_helper.broker)
channel = yield client.get_channel()
pub = metrics_workers.GraphitePublisher()
pub.start(channel)
pub.publish_metric(*datapoint)
self._check_msg(channel, *datapoint)
def test_publish_metric(self):
datapoint = ("vumi.test.v1", 1.0, 1234)
client = WorkerHelper.get_fake_amqp_client(self.worker_helper.broker)
channel = yield client.get_channel()
pub = metrics_workers.GraphitePublisher()
pub.start(channel)
pub.publish_metric(*datapoint)
self._check_msg(channel, *datapoint)
def test_publish_heartbeat(self):
broker = FakeAMQPBroker()
client = WorkerHelper.get_fake_amqp_client(broker)
channel = yield client.get_channel()
pub = MockHeartBeatPublisher(self.gen_fake_attrs)
pub.start(channel)
pub._beat()
[msg] = broker.get_dispatched("vumi.health", "heartbeat.inbound")
self.assertEqual(json.loads(msg.body), self.gen_fake_attrs())
def test_publish_heartbeat(self):
broker = FakeAMQPBroker()
client = WorkerHelper.get_fake_amqp_client(broker)
channel = yield client.get_channel()
pub = MockHeartBeatPublisher(self.gen_fake_attrs)
pub.start(channel)
pub._beat()
[msg] = broker.get_dispatched("vumi.health", "heartbeat.inbound")
self.assertEqual(json.loads(msg.body), self.gen_fake_attrs())
def start_manager_as_publisher(self, manager):
client = WorkerHelper.get_fake_amqp_client(self.worker_helper.broker)
channel = yield client.get_channel()
manager.start(channel)
self.add_cleanup(manager.stop)
def start_publisher(self, publisher):
client = WorkerHelper.get_fake_amqp_client(self.worker_helper.broker)
channel = yield client.get_channel()
publisher.start(channel)
def start_manager_as_publisher(self, manager):
client = WorkerHelper.get_fake_amqp_client(self.worker_helper.broker)
channel = yield client.get_channel()
manager.start(channel)
self.add_cleanup(manager.stop)
def start_publisher(self, publisher):
client = WorkerHelper.get_fake_amqp_client(self.worker_helper.broker)
channel = yield client.get_channel()
publisher.start(channel)
