Skip to content

Commit da965ba

Browse files
committed
test: replace /graphql call with gRPC call
1 parent 9bb0897 commit da965ba

1 file changed

Lines changed: 23 additions & 28 deletions

File tree

tests/templates/kuttl/logging/test_log_aggregation.py

Lines changed: 23 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -1,45 +1,40 @@
1-
#!/usr/bin/env python3
2-
import requests
1+
import json
2+
import subprocess
33

44

55
def check_sent_events():
6-
response = requests.post(
7-
"http://hbase-vector-aggregator:8686/graphql",
8-
json={
9-
"query": """
10-
{
11-
transforms(first:100) {
12-
nodes {
13-
componentId
14-
metrics {
15-
sentEventsTotal {
16-
sentEventsTotal
17-
}
18-
}
19-
}
20-
}
21-
}
22-
"""
23-
},
6+
response = subprocess.run(
7+
[
8+
"grpcurl",
9+
"-plaintext",
10+
"-d",
11+
'{"limit": 100}',
12+
"hbase-vector-aggregator:8686",
13+
"vector.observability.v1.ObservabilityService/GetComponents",
14+
],
15+
capture_output=True,
16+
text=True,
17+
check=True, # Raise a CalledProcessError if non-zero return
18+
timeout=20, # seconds
2419
)
20+
result = json.loads(response.stdout)
21+
components = result.get("components", [])
22+
transforms = [
23+
c for c in components if c.get("componentType") == "COMPONENT_TYPE_TRANSFORM"
24+
]
2525

26-
assert response.status_code == 200, (
27-
"Cannot access the API of the vector aggregator."
28-
)
29-
30-
result = response.json()
26+
assert len(transforms) > 0, "No transform components found"
3127

32-
transforms = result["data"]["transforms"]["nodes"]
3328
for transform in transforms:
3429
sentEvents = transform["metrics"]["sentEventsTotal"]
3530
componentId = transform["componentId"]
3631

3732
if componentId == "filteredInvalidEvents":
38-
assert sentEvents is None or sentEvents["sentEventsTotal"] == 0, (
33+
assert sentEvents is None or int(sentEvents) == 0, (
3934
"Invalid log events were sent."
4035
)
4136
else:
42-
assert sentEvents is not None and sentEvents["sentEventsTotal"] > 0, (
37+
assert sentEvents is not None and int(sentEvents) > 0, (
4338
f'No events were sent in "{componentId}".'
4439
)
4540

0 commit comments

Comments
 (0)