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 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)