def test_check(self): """Check status of a znode""" z = ZkFarmer(self.client) self.client.ensure_path("/something") for i in range(10): self.client.ensure_path("/something/mysql%d" % i) self.client.set("/something/mysql%d" % i, json.dumps({"stuff": "ok"})) self.client.set("/something", json.dumps({"size": 12})) self.assertEqual(z.check("/something", "3", "4")[0], z.STATUS_OK) self.assertEqual(z.check("/something", "3", "1")[0], z.STATUS_WARNING) self.assertEqual(z.check("/something", "2", "4")[0], z.STATUS_CRITICAL) self.assertEqual(z.check("/something", "2", "1")[0], z.STATUS_CRITICAL) self.assertEqual(z.check("/something", "20%", "4")[0], z.STATUS_OK) self.assertEqual(z.check("/something", "10%", "4")[0], z.STATUS_CRITICAL) self.assertEqual(z.check("/something", "20%")[0], z.STATUS_OK)
def test_check(self): """Check status of a znode""" z = ZkFarmer(self.client) self.client.ensure_path("/something") for i in range(10): self.client.ensure_path("/something/mysql%d" % i) self.client.set("/something/mysql%d" % i, json.dumps({"stuff": "ok"})) self.client.set("/something", json.dumps({"size": 12})) self.assertEqual(z.check("/something", "3", "4")[0], z.STATUS_OK) self.assertEqual(z.check("/something", "3", "1")[0], z.STATUS_WARNING) self.assertEqual(z.check("/something", "2", "4")[0], z.STATUS_CRITICAL) self.assertEqual(z.check("/something", "2", "1")[0], z.STATUS_CRITICAL) self.assertEqual(z.check("/something", "20%", "4")[0], z.STATUS_OK) self.assertEqual( z.check("/something", "10%", "4")[0], z.STATUS_CRITICAL) self.assertEqual(z.check("/something", "20%")[0], z.STATUS_OK)
def test_check_with_filter(self): """Check status of a znode with filters""" z = ZkFarmer(self.client) self.client.ensure_path("/something") for i in range(10): self.client.ensure_path("/something/mysql%d" % i) self.client.set("/something/mysql%d" % i, json.dumps({"enabled": ((i == 0) and "0" or "1"), "weight": "%d" % (i+10)})) self.client.set("/something", json.dumps({"size": 12, "running_filter": "enabled=1,weight>11"})) self.assertEqual(z.check("/something", "5")[0], z.STATUS_OK) self.assertEqual(z.check("/something", "4")[0], z.STATUS_CRITICAL)
def test_check_with_filter(self): """Check status of a znode with filters""" z = ZkFarmer(self.client) self.client.ensure_path("/something") for i in range(10): self.client.ensure_path("/something/mysql%d" % i) self.client.set( "/something/mysql%d" % i, json.dumps({ "enabled": ((i == 0) and "0" or "1"), "weight": "%d" % (i + 10) })) self.client.set( "/something", json.dumps({ "size": 12, "running_filter": "enabled=1,weight>11" })) self.assertEqual(z.check("/something", "5")[0], z.STATUS_OK) self.assertEqual(z.check("/something", "4")[0], z.STATUS_CRITICAL)