def test_validation(self):
cases = {
make_record():
"Invalid record: no current leader recorded",
make_record(holder="me"):
"Invalid record: no lease duration set",
make_record(holder="me", duration=30):
"Invalid record: no renew time set",
make_record(holder="me", duration=30, renew="2018-12-18T12:32:22Z"):
"Invalid record: no acquire time recorded",
make_record(holder="me",
duration=30,
renew=datetime.now(),
acquire="2018-12-18T12:32:22Z"):
None,
make_record(holder="me",
duration=30,
renew="invalid",
acquire="2018-12-18T12:32:22Z"):
"Invalid record: bad format for renewTime field",
make_record(holder="me",
duration=30,
renew="2018-12-18T12:32:22Z",
acquire="0000-12-18T12:32:22Z"):
"Invalid record: bad format for acquireTime field",
}
for raw, expected_reason in iteritems(cases):
valid, reason = ElectionRecord(raw).validate()
assert reason == expected_reason
if expected_reason is None:
assert valid is True
else:
assert valid is False
def test_seconds_until_renew(self):
raw = make_record(
holder="me", duration=30, acquire="2018-12-18T12:32:22Z", renew=datetime.utcnow() + timedelta(seconds=20)
)
record = ElectionRecord(raw)
assert record.seconds_until_renew > 19
assert record.seconds_until_renew < 21
raw = make_record(
holder="me", duration=30, acquire="2018-12-18T12:32:22Z", renew=datetime.utcnow() - timedelta(seconds=5)
)
record = ElectionRecord(raw)
assert record.seconds_until_renew > -6
assert record.seconds_until_renew < -4
def mock_leader():
# Inject a fake object in the leader-election monitoring logic
# don't forget to update the [testenv] in tox.ini with the 'kube' dependency
with mock.patch(
'datadog_checks.kube_scheduler.KubeSchedulerCheck._get_record',
return_value=ElectionRecord(
'{"holderIdentity":"pod1","leaseDurationSeconds":15,"leaderTransitions":3,'
+ '"acquireTime":"2018-12-19T18:23:24Z","renewTime":"2019-01-02T16:30:07Z"}'
),
):
yield
def mock_leader():
# Inject a fake object in the leader-election monitoring logic
with mock.patch(
'datadog_checks.kube_controller_manager.KubeControllerManagerCheck._get_record',
return_value=ElectionRecord(
'{"holderIdentity":"pod1","leaseDurationSeconds":15,"leaderTransitions":3,'
+
'"acquireTime":"2018-12-19T18:23:24Z","renewTime":"2019-01-02T16:30:07Z"}'
),
):
yield
def test_parse_raw(self):
record = ElectionRecord(RAW_VALID_RECORD)
valid, reason = record.validate()
assert valid is True
assert reason is None
assert record.leader_name == "dd-cluster-agent-568f458dd6-kj6vt"
assert record.lease_duration == 60
assert record.transitions == 7
assert record.renew_time > record.acquire_time
assert record.seconds_until_renew < 0
assert record.summary == ("Leader: dd-cluster-agent-568f458dd6-kj6vt "
"since 2018-12-17 11:53:07+00:00, "
"next renew 2018-12-18 12:32:22+00:00")