def test_parse_monitor_response_time(self):
obj = json.loads(
self.get_data_test_input(
"data/pods_with_response_time_monitor.json"))
pod_gauge = watchdog.gen_k8s_pod_gauge()
container_gauge = watchdog.gen_k8s_container_gauge()
pod_info = collections.defaultdict(lambda: [])
endpoints = []
vc_usage = watchdog.VcUsage()
watchdog.process_pods_status(obj, pod_gauge, container_gauge, pod_info,
endpoints, vc_usage)
self.assertEqual(2, len(endpoints))
endpoint0 = endpoints[0]
endpoint1 = endpoints[1]
self.assertEqual("job-exporter", endpoint0.name)
self.assertEqual("job-exporter", endpoint1.name)
self.assertEqual("10.151.40.231", endpoint0.ip)
self.assertEqual("10.151.40.227", endpoint1.ip)
self.assertEqual(9102, endpoint0.port)
self.assertEqual(9102, endpoint1.port)
self.assertEqual("/healthz", endpoint0.path)
self.assertEqual("/healthz", endpoint1.path)
self.assertEqual(10, endpoint0.timeout)
self.assertEqual(10, endpoint1.timeout)
def test_parse_pods_status(self):
obj = json.loads(self.get_data_test_input("data/pods_list.json"))
pod_gauge = watchdog.gen_pai_pod_gauge()
container_gauge = watchdog.gen_pai_container_gauge()
job_pod_gauge = watchdog.gen_pai_job_pod_gauge()
pod_info = collections.defaultdict(lambda : [])
watchdog.process_pods_status(obj, pod_gauge, container_gauge, job_pod_gauge, pod_info)
self.assertTrue(len(pod_gauge.samples) > 0)
self.assertEqual('10.151.40.4', pod_gauge.samples[0].labels['host_ip'])
self.assertEqual('running', pod_gauge.samples[0].labels['phase'])
self.assertEqual('true', pod_gauge.samples[0].labels['pod_scheduled'])
self.assertEqual('true', pod_gauge.samples[0].labels['initialized'])
self.assertTrue(len(container_gauge.samples) > 0)
self.assertEqual('10.151.40.4', container_gauge.samples[0].labels['host_ip'])
self.assertEqual('running', container_gauge.samples[0].labels['state'])
self.assertEqual('default', container_gauge.samples[0].labels['namespace'])
self.assertEqual('nvidia-drivers', container_gauge.samples[0].labels['name'])
self.assertTrue(len(job_pod_gauge.samples) > 0)
self.assertEqual('10.1.3.29', job_pod_gauge.samples[0].labels['host_ip'])
self.assertEqual('pending', job_pod_gauge.samples[0].labels['phase'])
self.assertEqual('true', job_pod_gauge.samples[0].labels['pod_bound'])
self.assertEqual('true', job_pod_gauge.samples[0].labels['initialized'])
def collect(self):
pod_gauge = watchdog.gen_pai_pod_gauge()
container_gauge = watchdog.gen_pai_container_gauge()
watchdog.process_pods_status(pod_gauge, container_gauge, obj)
yield pod_gauge
yield container_gauge
def test_process_pods_with_no_condition(self):
obj = json.loads(self.get_data_test_input("data/no_condtion_pod.json"))
pod_gauge = watchdog.gen_pai_pod_gauge()
container_gauge = watchdog.gen_pai_container_gauge()
watchdog.process_pods_status(pod_gauge, container_gauge, obj)
self.assertTrue(len(pod_gauge.samples) > 0)
self.assertEquals(0, len(container_gauge.samples))
def test_parse_pods_status(self):
obj = json.loads(self.get_data_test_input("data/pods_list.json"))
pod_gauge = watchdog.gen_pai_node_gauge()
container_gauge = watchdog.gen_pai_container_gauge()
watchdog.process_pods_status(pod_gauge, container_gauge, obj)
self.assertTrue(len(pod_gauge.samples) > 0)
self.assertTrue(len(container_gauge.samples) > 0)
def collect(self):
pod_gauge = watchdog.gen_k8s_pod_gauge()
container_gauge = watchdog.gen_k8s_container_gauge()
pod_info = collections.defaultdict(lambda: [])
watchdog.process_pods_status(obj, pod_gauge, container_gauge,
pod_info, [], watchdog.VcUsage())
yield pod_gauge
yield container_gauge
def collect(self):
pod_gauge = watchdog.gen_pai_pod_gauge()
container_gauge = watchdog.gen_pai_container_gauge()
job_pod_gauge = watchdog.gen_pai_job_pod_gauge()
pod_info = collections.defaultdict(lambda : [])
watchdog.process_pods_status(obj,
pod_gauge, container_gauge, job_pod_gauge, pod_info)
yield pod_gauge
yield container_gauge
def test_process_pods_with_no_condition(self):
obj = json.loads(self.get_data_test_input("data/no_condtion_pod.json"))
pod_gauge = watchdog.gen_pai_pod_gauge()
container_gauge = watchdog.gen_pai_container_gauge()
pod_info = collections.defaultdict(lambda: [])
watchdog.process_pods_status(obj, pod_gauge, container_gauge, pod_info)
self.assertTrue(len(pod_gauge.samples) > 0)
self.assertEqual(0, len(container_gauge.samples))
def test_parse_pods_status(self):
obj = json.loads(self.get_data_test_input("data/pods_list.json"))
pod_gauge = watchdog.gen_k8s_pod_gauge()
container_gauge = watchdog.gen_k8s_container_gauge()
pod_info = collections.defaultdict(lambda: [])
watchdog.process_pods_status(obj, pod_gauge, container_gauge, pod_info,
[], watchdog.VcUsage())
self.assertTrue(len(pod_gauge.samples) > 0)
self.assertTrue(len(container_gauge.samples) > 0)