Merge branch 'fix-broken-tests' into 'main'

Update reporter tests according to the code changes

See merge request postgres-ai/postgres_ai!81

Author: Dementii Priadko

tests/reporter/test_formatters.py

@@ -70,5 +70,6 @@ def test_format_report_data_structure(generator: PostgresReportGenerator) -> Non
     payload = generator.format_report_data("A002", {"foo": "bar"}, host)
 
     assert payload["checkId"] == "A002"
-    assert payload["hosts"]["master"] == host
+    # Newer reporter returns a 'nodes' structure instead of legacy 'hosts'.
+    assert payload["nodes"]["primary"] == host
     assert payload["results"][host]["data"] == {"foo": "bar"}

tests/reporter/test_generators_unit.py

@@ -784,6 +784,11 @@ class DummyGenerator:
         def __init__(self, *args, **kwargs):
             self.closed = False
 
+        def get_all_clusters(self):
+            # Match current reporter.main() behavior which always calls
+            # get_all_clusters() when cluster is not explicitly provided.
+            return ["local"]
+
         def test_connection(self) -> bool:
             return True
 
@@ -799,7 +804,19 @@ def close_postgres_sink(self):
     postgres_reports_module.main()
 
     captured = capsys.readouterr().out
-    output = json.loads(captured)
+
+    # main() prints progress banners along with the JSON payload.
+    # Extract the JSON object from the captured stdout by finding the
+    # first line that looks like JSON and joining from there.
+    lines = captured.splitlines()
+    start_idx = 0
+    for i, line in enumerate(lines):
+        if line.strip().startswith("{"):
+            start_idx = i
+            break
+    json_str = "\n".join(lines[start_idx:])
+
+    output = json.loads(json_str)
     assert output["checkId"] == "A002"
     assert "results" in output