global: snapshot

packages/brk_client/tests/tree.py

@@ -0,0 +1,89 @@
+"""Comprehensive test that fetches all endpoints in the tree."""
+
+from brk_client import BrkClient
+
+
+def get_all_metrics(obj, path=""):
+    """Recursively collect all MetricPattern instances from the tree."""
+    metrics = []
+
+    for attr_name in dir(obj):
+        if attr_name.startswith("_"):
+            continue
+
+        try:
+            attr = getattr(obj, attr_name)
+        except Exception:
+            continue
+
+        current_path = f"{path}.{attr_name}" if path else attr_name
+
+        # Check if this is a metric pattern (has 'by' attribute with index methods)
+        if hasattr(attr, "by"):
+            by = attr.by
+            indexes = []
+            for idx_name in dir(by):
+                if not idx_name.startswith("_") and callable(
+                    getattr(by, idx_name, None)
+                ):
+                    indexes.append(idx_name)
+            if indexes:
+                metrics.append((current_path, attr, indexes))
+
+        # Recurse into nested tree nodes
+        if hasattr(attr, "__dict__") and not callable(attr):
+            metrics.extend(get_all_metrics(attr, current_path))
+
+    return metrics
+
+
+def test_all_endpoints():
+    """Test fetching last 3 values from all metric endpoints."""
+    client = BrkClient("http://localhost:3110")
+
+    metrics = get_all_metrics(client.tree)
+    print(f"\nFound {len(metrics)} metrics")
+
+    success = 0
+    failed = 0
+    errors = []
+
+    for path, metric, indexes in metrics:
+        for idx_name in indexes:
+            try:
+                by = metric.by
+                endpoint = getattr(by, idx_name)()
+                res = endpoint.range(-3)
+                count = len(res["data"])
+                if count != 3:
+                    failed += 1
+                    error_msg = (
+                        f"FAIL: {path}.by.{idx_name}() -> expected 3, got {count}"
+                    )
+                    errors.append(error_msg)
+                    print(error_msg)
+                else:
+                    success += 1
+                    print(f"OK: {path}.by.{idx_name}() -> {count} items")
+            except Exception as e:
+                failed += 1
+                error_msg = f"FAIL: {path}.by.{idx_name}() -> {e}"
+                errors.append(error_msg)
+                print(error_msg)
+
+    print("\n=== Results ===")
+    print(f"Success: {success}")
+    print(f"Failed: {failed}")
+
+    if errors:
+        print("\nErrors:")
+        for err in errors[:10]:  # Show first 10 errors
+            print(f"  {err}")
+        if len(errors) > 10:
+            print(f"  ... and {len(errors) - 10} more")
+
+    assert failed == 0, f"{failed} endpoints failed"
+
+
+if __name__ == "__main__":
+    test_all_endpoints()