82 lines
3.2 KiB
Python
82 lines
3.2 KiB
Python
|
"""Unit tests for nightly-report.py (pure helpers only).
|
||
|
|
|
||
|
|
Run: pytest stacks/k8s-version-upgrade/scripts/test_nightly_report.py
|
||
|
|
Loaded via importlib because the filename has a hyphen.
|
||
|
|
"""
|
||
|
|
import importlib.util
|
||
|
|
import pathlib
|
||
|
|
|
||
|
|
HERE = pathlib.Path(__file__).parent
|
||
|
|
_spec = importlib.util.spec_from_file_location("nightly_report", HERE / "nightly-report.py")
|
||
|
|
nr = importlib.util.module_from_spec(_spec)
|
||
|
|
_spec.loader.exec_module(nr)
|
||
|
|
|
||
|
|
LAST_RUN = 1781996424.0 # 2026-06-20T23:00:24Z — matches last night's gauge
|
||
|
|
METRICS_BLOCKED = f"""# TYPE k8s_upgrade_available gauge
|
||
|
|
k8s_upgrade_available{{instance="",job="k8s-version-check",kind="minor",running="1.34.9",target="1.35.6"}} 1
|
||
|
|
k8s_upgrade_blocked{{instance="",job="k8s-version-upgrade"}} 1
|
||
|
|
k8s_version_check_last_run_timestamp{{instance="",job="k8s-version-check"}} {LAST_RUN}
|
||
|
|
"""
|
||
|
|
NODES_UNIFORM = [(f"k8s-node{i}", "v1.34.9") for i in range(7)]
|
||
|
|
|
||
|
|
|
||
|
|
def test_parse_metrics_basic():
|
||
|
|
m = nr.parse_metrics(METRICS_BLOCKED)
|
||
|
|
names = {n for n, _, _ in m}
|
||
|
|
assert names == {"k8s_upgrade_available", "k8s_upgrade_blocked", "k8s_version_check_last_run_timestamp"}
|
||
|
|
avail = nr.select(m, "k8s_upgrade_available")
|
||
|
|
assert avail[0][0]["target"] == "1.35.6"
|
||
|
|
assert avail[0][0]["kind"] == "minor"
|
||
|
|
assert avail[0][1] == 1.0
|
||
|
|
|
||
|
|
|
||
|
|
def test_parse_metrics_ignores_comments_and_junk():
|
||
|
|
assert nr.parse_metrics("# HELP foo\n\ngarbage line\n") == []
|
||
|
|
|
||
|
|
|
||
|
|
def test_fmt_age():
|
||
|
|
assert nr.fmt_age(120) == "2m ago"
|
||
|
|
assert nr.fmt_age(7200) == "2.0h ago"
|
||
|
|
assert nr.fmt_age(172800) == "2.0d ago"
|
||
|
|
|
||
|
|
|
||
|
|
def test_compose_blocked_lists_reasons():
|
||
|
|
m = nr.parse_metrics(METRICS_BLOCKED)
|
||
|
|
reasons = ("addon external-secrets v0.12 supports k8s <= 1.31; target 1.35 exceeds it\n"
|
||
|
|
"addon kyverno v1.16 supports k8s <= 1.34; target 1.35 exceeds it")
|
||
|
|
out = nr.compose_report(LAST_RUN + 30000, NODES_UNIFORM, m, reasons, [])
|
||
|
|
assert "🔴 BLOCKED" in out and "1.35.6" in out
|
||
|
|
assert "external-secrets" in out and "kyverno" in out
|
||
|
|
assert "all 7 nodes uniform" in out
|
||
|
|
assert "fresh ✓" in out
|
||
|
|
|
||
|
|
|
||
|
|
def test_compose_noop_when_no_target():
|
||
|
|
m = nr.parse_metrics(f'k8s_version_check_last_run_timestamp{{}} {LAST_RUN}\n')
|
||
|
|
out = nr.compose_report(LAST_RUN + 30000, NODES_UNIFORM, m, None, [])
|
||
|
|
assert "⚪ No upgrade needed" in out
|
||
|
|
|
||
|
|
|
||
|
|
def test_compose_upgraded_when_nodes_match_target():
|
||
|
|
m = nr.parse_metrics(f"""k8s_upgrade_available{{kind="minor",target="1.35.6"}} 1
|
||
|
|
k8s_upgrade_blocked{{}} 0
|
||
|
|
k8s_version_check_last_run_timestamp{{}} {LAST_RUN}
|
||
|
|
""")
|
||
|
|
nodes = [(f"k8s-node{i}", "v1.35.6") for i in range(7)]
|
||
|
|
out = nr.compose_report(LAST_RUN + 30000, nodes, m, None, [])
|
||
|
|
assert "🟢 UPGRADED" in out and "1.35.6" in out
|
||
|
|
|
||
|
|
|
||
|
|
def test_compose_stale_detector_flagged():
|
||
|
|
m = nr.parse_metrics(METRICS_BLOCKED)
|
||
|
|
out = nr.compose_report(LAST_RUN + 200000, NODES_UNIFORM, m, "x", []) # ~55h later
|
||
|
|
assert "Detector did not run last night" in out
|
||
|
|
assert "STALE" in out
|
||
|
|
|
||
|
|
|
||
|
|
def test_compose_includes_recent_jobs():
|
||
|
|
m = nr.parse_metrics(METRICS_BLOCKED)
|
||
|
|
jobs = [{"name": "k8s-upgrade-preflight-1-35-6", "status": "Failed", "age_s": 3600}]
|
||
|
|
out = nr.compose_report(LAST_RUN + 30000, NODES_UNIFORM, m, "x", jobs)
|
||
|
|
assert "k8s-upgrade-preflight-1-35-6: Failed" in out
|