def test_pod_expiration(monkeypatch, aggregator, tagger):
check = KubeletCheck('kubelet', None, {}, [{}])
check.pod_list_url = "dummyurl"
# Fixtures contains four pods:
# - dd-agent-ntepl old but running
# - hello1-1550504220-ljnzx succeeded and old enough to expire
# - hello5-1550509440-rlgvf succeeded but not old enough
# - hello8-1550505780-kdnjx has one old container and a recent container, don't expire
monkeypatch.setattr(check, 'perform_kubelet_query',
mock.Mock(return_value=MockStreamResponse('pods_expired.json')))
monkeypatch.setattr(check, '_compute_pod_expiration_datetime', mock.Mock(
return_value=parse_rfc3339("2019-02-18T16:00:06Z")
))
attrs = {'is_excluded.return_value': False}
check.pod_list_utils = mock.Mock(**attrs)
pod_list = check.retrieve_pod_list()
assert pod_list['expired_count'] == 1
expected_names = ['dd-agent-ntepl', 'hello5-1550509440-rlgvf', 'hello8-1550505780-kdnjx']
collected_names = [p['metadata']['name'] for p in pod_list['items']]
assert collected_names == expected_names
# Test .pods.expired gauge is submitted
check._report_container_state_metrics(pod_list, ["custom:tag"])
aggregator.assert_metric("kubernetes.pods.expired", value=1, tags=["custom:tag"])
def test_retrieve_pod_list_success(monkeypatch):
check = KubeletCheck('kubelet', {}, [{}])
check.pod_list_url = "dummyurl"
monkeypatch.setattr(check, 'perform_kubelet_query', mock.Mock(return_value=MockStreamResponse('pod_list_raw.dat')))
monkeypatch.setattr(check, '_compute_pod_expiration_datetime', mock.Mock(return_value=None))
retrieved = check.retrieve_pod_list()
expected = json.loads(mock_from_file("pod_list_raw.json"))
assert json.dumps(retrieved, sort_keys=True) == json.dumps(expected, sort_keys=True)
def test_retrieved_pod_list_failure(monkeypatch):
def mock_perform_kubelet_query(s):
raise Exception("network error")
check = KubeletCheck('kubelet', None, {}, [{}])
check.pod_list_url = "dummyurl"
monkeypatch.setattr(check, 'perform_kubelet_query', mock_perform_kubelet_query)
retrieved = check.retrieve_pod_list()
assert retrieved is None
def test_report_container_state_metrics(monkeypatch):
check = KubeletCheck('kubelet', None, {}, [{}])
check.pod_list_url = "dummyurl"
monkeypatch.setattr(
check, 'perform_kubelet_query',
mock.Mock(return_value=MockStreamResponse('pods_crashed.json')))
monkeypatch.setattr(check, '_compute_pod_expiration_datetime',
mock.Mock(return_value=None))
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"]
with mock.patch("datadog_checks.kubelet.kubelet.get_tags",
side_effect=mocked_get_tags):
check._report_container_state_metrics(pod_list, instance_tags)
calls = [
mock.call('kubernetes.containers.last_state.terminated', 1, [
'kube_container_name:fluentd-gcp',
'kube_deployment:fluentd-gcp-v2.0.10'
] + instance_tags + ['reason:OOMKilled']),
mock.call('kubernetes.containers.state.waiting', 1, [
'kube_container_name:prometheus-to-sd-exporter',
'kube_deployment:fluentd-gcp-v2.0.10'
] + instance_tags + ['reason:CrashLoopBackOff']),
mock.call('kubernetes.containers.restarts', 1, [
'kube_container_name:fluentd-gcp',
'kube_deployment:fluentd-gcp-v2.0.10'
] + instance_tags),
mock.call('kubernetes.containers.restarts', 0, [
'kube_container_name:prometheus-to-sd-exporter',
'kube_deployment:fluentd-gcp-v2.0.10'
] + instance_tags),
]
check.gauge.assert_has_calls(calls, any_order=True)
container_state_gauges = [
x[0][2] for x in check.gauge.call_args_list
if x[0][0].startswith('kubernetes.containers.state')
]
if any(map(lambda e: 'reason:TransientReason' in e,
container_state_gauges)):
raise AssertionError(
'kubernetes.containers.state.* was submitted with a transient reason'
)
if any(
map(lambda e: not any(x for x in e if x.startswith('reason:')),
container_state_gauges)):
raise AssertionError(
'kubernetes.containers.state.* was submitted without a reason')
def test_retrieve_pod_list_success(monkeypatch):
class MockResponse:
def __init__(self, json_data):
self.json_data = json_data
def json(self):
return (json.loads(self.json_data))
check = KubeletCheck('kubelet', None, {}, [{}])
check.pod_list_url = "dummyurl"
monkeypatch.setattr(check, 'perform_kubelet_query',
mock.Mock(return_value=MockResponse(mock_from_file('pod_list_raw.dat'))))
retrieved = check.retrieve_pod_list()
expected = json.loads(mock_from_file("pod_list_raw.json"))
assert (json.dumps(retrieved, sort_keys=True) == json.dumps(expected, sort_keys=True))
