def rmq(args): host = config.get('rabbit_host') port = config.get('rabbit_port') try: cons = RMQConsumer(args.topic, host, port) for method, _, msg in cons.consume(): print '{}: {}'.format(method.routing_key, msg) except RMQConnError: print 'No RabbitMQ host found.' except KeyboardInterrupt: cons.close_connection() print 'Connection closed'
def rmq(args): host = config.get("rabbit_host") port = config.get("rabbit_port") try: cons = RMQConsumer(args.topic, host, port) for method, _, msg in cons.consume(): print "{}: {}".format(method.routing_key, msg) except RMQConnError: print "No RabbitMQ host found." except KeyboardInterrupt: cons.close_connection() print "Connection closed"
class TestCPUMemUsage(RMQTestCase): def onRabbitUp(self): self.rmq = RMQConsumer() def test_cpu_mem_are_percentages(self): cid = self._start_test_container(time=100) s = CPUMemoryPercSensor({"cid": cid}) s.start() _, _, top = next(self.rmq.consume()) vals = top["tot"].values() del top["tot"] for p in top.values(): usages = p["usage"].values() vals.extend(usages) for v in vals: self.assertTrue(v >= 0 and v <= 100) s.kill() def test_tot_is_reasonable(self): cid = self._start_test_container(time=100) s = CPUMemoryPercSensor({"cid": cid}) s.start() _, _, top = next(self.rmq.consume()) real_tot = top["tot"] del top["tot"] expected_tot = {k: 0 for k in real_tot.keys()} for p in top.values(): for k, v in p["usage"].items(): expected_tot[k] += v self.assertEqual(real_tot, expected_tot) s.kill()
class TestCPUMemUsage(RMQTestCase): def onRabbitUp(self): self.rmq = RMQConsumer() def test_cpu_mem_are_percentages(self): cid = self._start_test_container(time=100) s = CPUMemoryPercSensor({'cid': cid}) s.start() _, _, top = next(self.rmq.consume()) vals = top['tot'].values() del top['tot'] for p in top.values(): usages = p['usage'].values() vals.extend(usages) for v in vals: self.assertTrue(v >= 0 and v <= 100) s.kill() def test_tot_is_reasonable(self): cid = self._start_test_container(time=100) s = CPUMemoryPercSensor({'cid': cid}) s.start() _, _, top = next(self.rmq.consume()) real_tot = top['tot'] del top['tot'] expected_tot = {k: 0 for k in real_tot.keys()} for p in top.values(): for k, v in p['usage'].items(): expected_tot[k] += v self.assertEqual(real_tot, expected_tot) s.kill()
class BaseRMQTestCase(RMQTestCase): def onRabbitUp(self): self.rmq = RMQConsumer() self.s = DummyRMQSensor({'cid': 'non-existent-container'}) self.s.start() def onRabbitDown(self): self.s.kill() def test_values(self): n_msg = 10 # push results to rabbit expected = [] for i, d in enumerate(self.s.get_data()): expected.append(d) if i == n_msg - 1: break # get messages from rabbit real = [body for _, _, body in self.rmq.consume(n_msg)] self.assertEqual(expected, real)