|
7 | 7 | import tempfile
|
8 | 8 | from time import sleep
|
9 | 9 |
|
| 10 | +import pytest |
| 11 | + |
10 | 12 |
|
11 | 13 | def parse_prometheus(input_string):
|
12 | 14 | result = {}
|
@@ -74,34 +76,20 @@ def test_memory_spooling_metrics(mini_sentry, relay):
|
74 | 76 | assert int(body["relay_spool_total_size"]) == 0
|
75 | 77 |
|
76 | 78 |
|
77 |
| -def test_service_utilization_metrics(mini_sentry, relay): |
78 |
| - relay = relay(mini_sentry) |
79 |
| - |
80 |
| - response = relay.get("/api/relay/autoscaling/") |
81 |
| - parsed = parse_prometheus(response.text) |
82 |
| - assert response.status_code == 200 |
83 |
| - |
84 |
| - assert int(parsed["relay_up"]) == 1 |
85 |
| - assert ( |
86 |
| - 0 <= int(parsed['relay_utilization{relay_service="AggregatorService"}']) <= 100 |
87 |
| - ) |
88 |
| - |
89 |
| - |
90 |
| -def test_pool_utilization(mini_sentry, relay): |
91 |
| - relay = relay(mini_sentry) |
92 |
| - |
93 |
| - response = relay.get("/api/relay/autoscaling/") |
94 |
| - parsed = parse_prometheus(response.text) |
95 |
| - assert response.status_code == 200 |
96 |
| - |
97 |
| - assert 0 <= int(parsed["relay_worker_pool_utilization"]) <= 100 |
98 |
| - |
99 |
| - |
100 |
| -def test_runtime_utilization(mini_sentry, relay): |
| 79 | +@pytest.mark.parametrize( |
| 80 | + "metric_name", |
| 81 | + ( |
| 82 | + 'relay_utilization{relay_service="AggregatorService"}', |
| 83 | + 'relay_service_utilization{relay_service="AggregatorService"}', |
| 84 | + "relay_worker_pool_utilization", |
| 85 | + "relay_runtime_utilization", |
| 86 | + ), |
| 87 | +) |
| 88 | +def test_service_utilization_metrics(mini_sentry, relay, metric_name): |
101 | 89 | relay = relay(mini_sentry)
|
102 | 90 |
|
103 | 91 | response = relay.get("/api/relay/autoscaling/")
|
104 | 92 | parsed = parse_prometheus(response.text)
|
105 | 93 | assert response.status_code == 200
|
106 | 94 |
|
107 |
| - assert 0 <= int(parsed["relay_runtime_utilization"]) <= 100 |
| 95 | + assert 0 <= int(parsed[metric_name]) <= 100 |
0 commit comments