def test_report_container_spec_metrics(monkeypatch): check = KubeletCheck('kubelet', None, {}, [{}]) monkeypatch.setattr(check, 'retrieve_pod_list', mock.Mock(return_value=json.loads(mock_from_file('pods.json')))) monkeypatch.setattr(check, 'gauge', mock.Mock()) attrs = {'is_excluded.return_value': False} check.container_filter = mock.Mock(**attrs) pod_list = check.retrieve_pod_list() instance_tags = ["one:1", "two:2"] with mock.patch("datadog_checks.kubelet.kubelet.get_tags", side_effect=mocked_get_tags): check._report_container_spec_metrics(pod_list, instance_tags) calls = [ mock.call('kubernetes.cpu.requests', 0.1, ['pod_name:fluentd-gcp-v2.0.10-9q9t4'] + instance_tags), mock.call('kubernetes.memory.requests', 209715200.0, ['pod_name:fluentd-gcp-v2.0.10-9q9t4'] + instance_tags), mock.call('kubernetes.memory.limits', 314572800.0, ['pod_name:fluentd-gcp-v2.0.10-9q9t4'] + instance_tags), mock.call('kubernetes.cpu.requests', 0.1, instance_tags), mock.call('kubernetes.cpu.requests', 0.1, instance_tags), mock.call('kubernetes.memory.requests', 134217728.0, instance_tags), mock.call('kubernetes.cpu.limits', 0.25, instance_tags), mock.call('kubernetes.memory.limits', 536870912.0, instance_tags), mock.call('kubernetes.cpu.requests', 0.1, ["pod_name=demo-app-success-c485bc67b-klj45"] + instance_tags), ] check.gauge.assert_has_calls(calls, any_order=True)
def test_report_container_requests_limits(monkeypatch, tagger): check = KubeletCheck('kubelet', None, {}, [{}]) monkeypatch.setattr( check, 'retrieve_pod_list', mock.Mock(return_value=json.loads( mock_from_file('pods_requests_limits.json')))) monkeypatch.setattr(check, 'gauge', mock.Mock()) attrs = {'is_excluded.return_value': False} check.pod_list_utils = mock.Mock(**attrs) pod_list = check.retrieve_pod_list() tags = ['kube_container_name:cassandra'] check._report_container_spec_metrics(pod_list, tags) calls = [ mock.call('kubernetes.cpu.requests', 0.5, ['pod_name:cassandra-0'] + tags), mock.call('kubernetes.memory.requests', 1073741824.0, ['pod_name:cassandra-0'] + tags), mock.call('kubernetes.ephemeral-storage.requests', 0.5, ['pod_name:cassandra-0'] + tags), mock.call('kubernetes.cpu.limits', 0.5, ['pod_name:cassandra-0'] + tags), mock.call('kubernetes.memory.limits', 1073741824.0, ['pod_name:cassandra-0'] + tags), mock.call('kubernetes.ephemeral-storage.limits', 2147483648.0, ['pod_name:cassandra-0'] + tags), ] check.gauge.assert_has_calls(calls, any_order=True)
def test_report_container_spec_metrics(monkeypatch, tagger): check = KubeletCheck('kubelet', None, {}, [{}]) monkeypatch.setattr(check, 'retrieve_pod_list', mock.Mock(return_value=json.loads(mock_from_file('pods.json')))) monkeypatch.setattr(check, 'gauge', mock.Mock()) attrs = {'is_excluded.return_value': False} check.pod_list_utils = mock.Mock(**attrs) pod_list = check.retrieve_pod_list() instance_tags = ["one:1", "two:2"] check._report_container_spec_metrics(pod_list, instance_tags) calls = [ mock.call('kubernetes.cpu.requests', 0.1, [ 'kube_container_name:fluentd-gcp', 'kube_deployment:fluentd-gcp-v2.0.10' ] + instance_tags), mock.call('kubernetes.memory.requests', 209715200.0, [ 'kube_container_name:fluentd-gcp', 'kube_deployment:fluentd-gcp-v2.0.10' ] + instance_tags), mock.call('kubernetes.memory.limits', 314572800.0, [ 'kube_container_name:fluentd-gcp', 'kube_deployment:fluentd-gcp-v2.0.10' ] + instance_tags), mock.call('kubernetes.cpu.requests', 0.1, instance_tags), mock.call('kubernetes.cpu.requests', 0.1, instance_tags), mock.call('kubernetes.memory.requests', 134217728.0, instance_tags), mock.call('kubernetes.cpu.limits', 0.25, instance_tags), mock.call('kubernetes.memory.limits', 536870912.0, instance_tags), mock.call('kubernetes.cpu.requests', 0.1, ["pod_name:demo-app-success-c485bc67b-klj45"] + instance_tags), ] if any(map(lambda e: 'pod_name:pi-kff76' in e, [x[0][2] for x in check.gauge.call_args_list])): raise AssertionError("kubernetes.cpu.requests was submitted for a non-running pod") check.gauge.assert_has_calls(calls, any_order=True)