test: cover ariadne triage evidence helpers

This commit is contained in:
codex 2026-05-20 00:54:31 -03:00
parent 69a5baa955
commit 38a5e924fe

View File

@ -1,5 +1,7 @@
from __future__ import annotations from __future__ import annotations
import json
from ariadne.services import testing_triage from ariadne.services import testing_triage
@ -36,6 +38,18 @@ class DummyStorage:
return matching[-limit:][::-1] return matching[-limit:][::-1]
class SettingsStub:
def __init__(self, **overrides) -> None: # type: ignore[no-untyped-def]
self.vm_url = ""
self.cluster_state_vm_timeout_sec = 1.0
self.jenkins_base_url = ""
self.jenkins_api_user = ""
self.jenkins_api_token = ""
self.jenkins_api_timeout_sec = 1.0
for key, value in overrides.items():
setattr(self, key, value)
def test_collect_testing_triage_builds_bundle(monkeypatch) -> None: def test_collect_testing_triage_builds_bundle(monkeypatch) -> None:
storage = DummyStorage() storage = DummyStorage()
monkeypatch.setattr( monkeypatch.setattr(
@ -72,3 +86,381 @@ def test_run_testing_triage_stores_latest(monkeypatch) -> None:
assert summary.status == "ok" assert summary.status == "ok"
assert storage.events[0][0] == testing_triage.TRIAGE_EVENT_TYPE assert storage.events[0][0] == testing_triage.TRIAGE_EVENT_TYPE
assert latest["summary"]["status"] == "ok" assert latest["summary"]["status"] == "ok"
def test_latest_testing_triage_bundle_handles_json_strings() -> None:
class JsonStorage:
def list_events(self, limit: int = 1, event_type: str | None = None): # type: ignore[no-untyped-def]
assert limit == 1
assert event_type == testing_triage.TRIAGE_EVENT_TYPE
return [{"detail": json.dumps({"summary": {"status": "ok"}})}]
latest = testing_triage.latest_testing_triage_bundle(JsonStorage()) # type: ignore[arg-type]
assert latest == {"summary": {"status": "ok"}}
def test_latest_testing_triage_bundle_ignores_bad_payloads() -> None:
class BadStorage:
def __init__(self, detail) -> None: # type: ignore[no-untyped-def]
self.detail = detail
def list_events(self, limit: int = 1, event_type: str | None = None): # type: ignore[no-untyped-def]
return [{"detail": self.detail}]
assert testing_triage.latest_testing_triage_bundle(BadStorage("{")) is None # type: ignore[arg-type]
assert testing_triage.latest_testing_triage_bundle(BadStorage(["nope"])) is None # type: ignore[arg-type]
assert testing_triage.latest_testing_triage_bundle(BadStorage(None)) is None # type: ignore[arg-type]
class EmptyStorage:
def list_events(self, limit: int = 1, event_type: str | None = None): # type: ignore[no-untyped-def]
return []
assert testing_triage.latest_testing_triage_bundle(EmptyStorage()) is None # type: ignore[arg-type]
def test_latest_cluster_snapshot_falls_back_to_live_collect(monkeypatch) -> None:
class BrokenStorage:
def latest_cluster_state(self): # type: ignore[no-untyped-def]
raise RuntimeError("storage down")
monkeypatch.setattr(
testing_triage,
"collect_cluster_state",
lambda: ({"collected_at": "live"}, {"status": "ok"}),
)
errors: list[str] = []
snapshot = testing_triage._latest_cluster_snapshot(BrokenStorage(), errors) # noqa: SLF001
assert snapshot == {"collected_at": "live"}
assert errors == ["cluster_state_latest: storage down"]
def test_latest_cluster_snapshot_records_collect_error(monkeypatch) -> None:
monkeypatch.setattr(
testing_triage,
"collect_cluster_state",
lambda: (_ for _ in ()).throw(RuntimeError("api down")),
)
errors: list[str] = []
snapshot = testing_triage._latest_cluster_snapshot(None, errors) # noqa: SLF001
assert snapshot == {}
assert errors == ["cluster_state_collect: api down"]
def test_cluster_evidence_limits_and_defaults() -> None:
snapshot = {
"summary": {
"health_bullets": list(range(20)),
"attention_ranked": [{"item": i} for i in range(20)],
},
"nodes_summary": {"total": 2, "ready": 1, "not_ready": 1, "not_ready_names": ["titan-06"]},
"flux": {"items": [{"name": str(i)} for i in range(20)]},
"pod_issues": {
"items": [{"pod": str(i)} for i in range(20)],
"pending_oldest": [{"pod": "p"}],
},
"jobs": {"failing": [{"job": "j"}], "active_oldest": [{"job": "old"}]},
"events": {"warnings_recent": [{"message": "warn"}]},
}
evidence = testing_triage._cluster_evidence(snapshot) # noqa: SLF001
assert len(evidence["health_bullets"]) == testing_triage._MAX_EVIDENCE_ITEMS # noqa: SLF001
assert evidence["nodes"]["not_ready_names"] == ["titan-06"]
assert evidence["jobs_failing"] == [{"job": "j"}]
def test_vm_items_handles_success_failure_and_bad_values(monkeypatch) -> None:
class FakeResponse:
def __init__(self, payload) -> None: # type: ignore[no-untyped-def]
self.payload = payload
def raise_for_status(self) -> None:
return None
def json(self): # type: ignore[no-untyped-def]
return self.payload
class FakeClient:
def __init__(self, *args, **kwargs) -> None: # type: ignore[no-untyped-def]
self.payload = kwargs.pop("payload", None)
def __enter__(self):
return self
def __exit__(self, *args) -> None: # type: ignore[no-untyped-def]
return None
def get(self, url, params=None): # type: ignore[no-untyped-def]
assert url.endswith("/api/v1/query")
assert params == {"query": "up"}
return FakeResponse(
{
"status": "success",
"data": {
"result": [
{"metric": {"suite": "ariadne", "__name__": "ignored"}, "value": [1, "2.5"]},
{"metric": {"suite": "metis"}, "value": [1, "bad"]},
]
},
}
)
monkeypatch.setattr(testing_triage, "settings", SettingsStub(vm_url="http://vm"))
monkeypatch.setattr(testing_triage.httpx, "Client", FakeClient)
errors: list[str] = []
items = testing_triage._vm_items("up", errors) # noqa: SLF001
assert items == [
{"labels": {"suite": "ariadne"}, "value": 2.5},
{"labels": {"suite": "metis"}, "value": 0.0},
]
assert errors == []
def test_vm_items_records_errors(monkeypatch) -> None:
class BrokenClient:
def __init__(self, *args, **kwargs) -> None: # type: ignore[no-untyped-def]
return None
def __enter__(self):
raise RuntimeError("network down")
def __exit__(self, *args) -> None: # type: ignore[no-untyped-def]
return None
monkeypatch.setattr(testing_triage, "settings", SettingsStub(vm_url="http://vm"))
monkeypatch.setattr(testing_triage.httpx, "Client", BrokenClient)
errors: list[str] = []
assert testing_triage._vm_items("up", errors) == [] # noqa: SLF001
assert errors == ["victoria_metrics: network down"]
def test_vm_items_handles_disabled_and_query_failure(monkeypatch) -> None:
class FailedResponse:
def raise_for_status(self) -> None:
return None
def json(self): # type: ignore[no-untyped-def]
return {"status": "error"}
class FailedClient:
def __init__(self, *args, **kwargs) -> None: # type: ignore[no-untyped-def]
return None
def __enter__(self):
return self
def __exit__(self, *args) -> None: # type: ignore[no-untyped-def]
return None
def get(self, url, params=None): # type: ignore[no-untyped-def]
return FailedResponse()
errors: list[str] = []
monkeypatch.setattr(testing_triage, "settings", SettingsStub(vm_url=""))
assert testing_triage._vm_items("up", errors) == [] # noqa: SLF001
assert errors == []
monkeypatch.setattr(testing_triage, "settings", SettingsStub(vm_url="http://vm"))
monkeypatch.setattr(testing_triage.httpx, "Client", FailedClient)
assert testing_triage._vm_items("up", errors) == [] # noqa: SLF001
assert errors == ["victoria_metrics: query failed"]
def test_jenkins_flatten_status_and_log_tail(monkeypatch) -> None:
rows = [
"skip",
{"name": "", "lastBuild": {}},
{
"name": "folder",
"jobs": [
{
"name": "child",
"url": "http://jenkins/job/folder/job/child/",
"color": "red",
"lastBuild": {
"number": 7,
"result": "FAILURE",
"timestamp": 2000,
"duration": 3000,
"url": "http://jenkins/build/7/",
},
}
],
},
{"name": "empty", "jobs": []},
]
flattened = testing_triage._flatten_jobs(rows) # noqa: SLF001
job = testing_triage._jenkins_job(flattened[0]) # noqa: SLF001
assert flattened[0]["name"] == "folder/child"
assert job is not None
assert job["status"] == "failure"
assert job["last_run_ts"] == 2.0
assert job["last_duration_seconds"] == 3.0
assert job["console_url"] == "http://jenkins/build/7/consoleText"
assert testing_triage._jenkins_status({"color": "blue_anime"}, "") == "running" # noqa: SLF001
assert testing_triage._jenkins_status({}, "SUCCESS") == "success" # noqa: SLF001
assert testing_triage._jenkins_status({"color": "green"}, "") == "success" # noqa: SLF001
assert testing_triage._jenkins_status({"color": "yellow"}, "") == "failure" # noqa: SLF001
assert testing_triage._jenkins_status({"color": "grey"}, "") == "unknown" # noqa: SLF001
assert testing_triage._jenkins_job({"name": 1, "url": "u"}) is None # noqa: SLF001
long_tail = testing_triage._tail_text("x" * (testing_triage._MAX_JENKINS_LOG_CHARS + 10)) # noqa: SLF001
assert len(long_tail) == testing_triage._MAX_JENKINS_LOG_CHARS # noqa: SLF001
assert testing_triage._tail_text("\n".join(str(i) for i in range(100))).startswith("20\n") # noqa: SLF001
def test_attach_jenkins_log_tail_ignores_missing_url_and_records_errors(monkeypatch) -> None:
class BrokenClient:
def __init__(self, **kwargs) -> None: # type: ignore[no-untyped-def]
return None
def __enter__(self):
return self
def __exit__(self, *args) -> None: # type: ignore[no-untyped-def]
return None
def get(self, url): # type: ignore[no-untyped-def]
raise RuntimeError("log gone")
errors: list[str] = []
testing_triage._attach_jenkins_log_tail({"job": "missing"}, errors) # noqa: SLF001
assert errors == []
monkeypatch.setattr(testing_triage, "settings", SettingsStub(jenkins_api_timeout_sec=1))
monkeypatch.setattr(testing_triage.httpx, "Client", BrokenClient)
testing_triage._attach_jenkins_log_tail({"job": "ariadne", "console_url": "http://jenkins/log"}, errors) # noqa: SLF001
assert errors == ["jenkins_log:ariadne: log gone"]
def test_fetch_jenkins_jobs_and_log_tail(monkeypatch) -> None:
class FakeResponse:
def __init__(self, payload=None, text="") -> None: # type: ignore[no-untyped-def]
self.payload = payload
self.text = text
def raise_for_status(self) -> None:
return None
def json(self): # type: ignore[no-untyped-def]
return self.payload
class FakeClient:
def __init__(self, **kwargs) -> None: # type: ignore[no-untyped-def]
assert kwargs["auth"] == ("jenkins", "token")
def __enter__(self):
return self
def __exit__(self, *args) -> None: # type: ignore[no-untyped-def]
return None
def get(self, url, params=None): # type: ignore[no-untyped-def]
if url.endswith("/api/json"):
assert "tree" in params
return FakeResponse(
{
"jobs": [
{
"name": "ariadne",
"url": "http://jenkins/job/ariadne/",
"color": "red",
"lastBuild": {"number": 8, "result": "FAILURE", "timestamp": 1000, "duration": 1000},
}
]
}
)
return FakeResponse(text="line1\nline2")
monkeypatch.setattr(
testing_triage,
"settings",
SettingsStub(jenkins_api_user="jenkins", jenkins_api_token="token", jenkins_api_timeout_sec=3),
)
monkeypatch.setattr(testing_triage.httpx, "Client", FakeClient)
jobs = testing_triage._fetch_jenkins_jobs("http://jenkins") # noqa: SLF001
errors: list[str] = []
testing_triage._attach_jenkins_log_tail(jobs[0], errors) # noqa: SLF001
assert jobs[0]["job"] == "ariadne"
assert jobs[0]["status"] == "failure"
assert jobs[0]["log_tail"] == "line1\nline2"
assert errors == []
def test_jenkins_signals_handles_disabled_and_failures(monkeypatch) -> None:
monkeypatch.setattr(testing_triage, "settings", SettingsStub(jenkins_base_url=""))
assert testing_triage._jenkins_signals([]) == {"failed_builds": []} # noqa: SLF001
monkeypatch.setattr(testing_triage, "settings", SettingsStub(jenkins_base_url="http://jenkins"))
monkeypatch.setattr(
testing_triage,
"_fetch_jenkins_jobs",
lambda base_url: (_ for _ in ()).throw(RuntimeError("boom")),
)
errors: list[str] = []
assert testing_triage._jenkins_signals(errors) == {"failed_builds": []} # noqa: SLF001
assert errors == ["jenkins: boom"]
def test_jenkins_signals_attaches_recent_failed_builds(monkeypatch) -> None:
jobs = [
{"job": "old", "status": "failure", "last_run_ts": 1},
{"job": "ok", "status": "success", "last_run_ts": 5},
{"job": "running", "status": "running", "last_run_ts": 10},
{"job": "unknown", "status": "unknown", "last_run_ts": 3},
]
attached: list[str] = []
monkeypatch.setattr(testing_triage, "settings", SettingsStub(jenkins_base_url="http://jenkins"))
monkeypatch.setattr(testing_triage, "_fetch_jenkins_jobs", lambda base_url: jobs)
monkeypatch.setattr(testing_triage, "_attach_jenkins_log_tail", lambda job, errors: attached.append(job["job"]))
signals = testing_triage._jenkins_signals([]) # noqa: SLF001
assert [item["job"] for item in signals["failed_builds"]] == ["running", "unknown", "old"]
assert attached == ["running", "unknown", "old"]
def test_summary_and_markdown_helpers() -> None:
quality = {
"failed_runs_24h": {"items": [{"labels": {"suite": "ariadne"}, "value": 2}]},
"empty": {"items": []},
"bad": "skip",
}
cluster = {
"flux_not_ready": [{"name": "monitoring"}],
"pod_issues": [],
"jobs_failing": [{"job": "job"}],
"collected_at": "now",
}
jenkins = {"failed_builds": [{"job": "ariadne"}]}
summary = testing_triage._summary(cluster, quality, jenkins, []) # noqa: SLF001
markdown = testing_triage._render_markdown( # noqa: SLF001
{
"generated_at": "now",
"summary": summary,
"evidence": {"cluster": cluster, "quality": quality, "jenkins": jenkins},
"unknowns": ["missing vm"],
}
)
assert summary["status"] == "needs_attention"
assert summary["failed_suites"] == ["ariadne"]
assert "- Flux: monitoring" in markdown
assert "- failed_runs_24h: {'suite': 'ariadne'} value=2" in markdown
assert "- missing vm" in markdown
assert testing_triage._markdown_named_items("Pods", ["bad"], "pod") == ["- Pods: none"] # noqa: SLF001
assert testing_triage._jenkins_auth() is None # noqa: SLF001