Skip to content

Commit a8cda01

Browse files
committed
Fix tests
1 parent 1ade402 commit a8cda01

File tree

4 files changed

+14
-21
lines changed

4 files changed

+14
-21
lines changed

runtime/service/src/test/java/org/apache/polaris/service/metrics/MetricsTestBase.java

Lines changed: 9 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,10 @@
3939
import org.junit.jupiter.api.Assertions;
4040
import org.junit.jupiter.api.BeforeAll;
4141
import org.junit.jupiter.api.BeforeEach;
42+
import org.junit.jupiter.api.Test;
4243
import org.junit.jupiter.api.TestInfo;
4344
import org.junit.jupiter.api.TestInstance;
4445
import org.junit.jupiter.api.extension.ExtendWith;
45-
import org.junit.jupiter.params.ParameterizedTest;
46-
import org.junit.jupiter.params.provider.ValueSource;
4746

4847
@TestInstance(TestInstance.Lifecycle.PER_CLASS)
4948
@ExtendWith(TestEnvironmentExtension.class)
@@ -72,26 +71,23 @@ public void clearMetrics() {
7271
registry.clear();
7372
}
7473

75-
private Map<String, MetricFamily> fetchMetrics(String endpoint) {
74+
private Map<String, MetricFamily> fetchMetrics() {
7675
AtomicReference<Map<String, MetricFamily>> value = new AtomicReference<>();
7776
Awaitility.await()
7877
.atMost(Duration.ofMinutes(2))
7978
.untilAsserted(
8079
() -> {
81-
value.set(
82-
TestMetricsUtil.fetchMetrics(
83-
fixture.client, testEnv.baseManagementUri(), endpoint));
80+
value.set(TestMetricsUtil.fetchMetrics(fixture.client, testEnv.baseManagementUri()));
8481
assertThat(value.get()).containsKey(API_METRIC_NAME);
8582
assertThat(value.get()).containsKey(HTTP_METRIC_NAME);
8683
});
8784
return value.get();
8885
}
8986

90-
@ParameterizedTest
91-
@ValueSource(strings = {"%s/metrics", "%s/q/metrics"})
92-
public void testMetricsEmittedOnSuccessfulRequest(String endpoint) {
87+
@Test
88+
public void testMetricsEmittedOnSuccessfulRequest() {
9389
sendSuccessfulRequest();
94-
Map<String, MetricFamily> allMetrics = fetchMetrics(endpoint);
90+
Map<String, MetricFamily> allMetrics = fetchMetrics();
9591
assertThat(allMetrics).containsKey(API_METRIC_NAME);
9692
assertThat(allMetrics.get(API_METRIC_NAME).getMetrics())
9793
.satisfiesOnlyOnce(
@@ -138,11 +134,10 @@ public void testMetricsEmittedOnSuccessfulRequest(String endpoint) {
138134
});
139135
}
140136

141-
@ParameterizedTest
142-
@ValueSource(strings = {"%s/metrics", "%s/q/metrics"})
143-
public void testMetricsEmittedOnFailedRequest(String endpoint) {
137+
@Test
138+
public void testMetricsEmittedOnFailedRequest() {
144139
sendFailingRequest();
145-
Map<String, MetricFamily> allMetrics = fetchMetrics(endpoint);
140+
Map<String, MetricFamily> allMetrics = fetchMetrics();
146141
assertThat(allMetrics).containsKey(API_METRIC_NAME);
147142
assertThat(allMetrics.get(API_METRIC_NAME).getMetrics())
148143
.satisfiesOnlyOnce(

runtime/service/src/test/java/org/apache/polaris/service/ratelimiter/RateLimiterFilterTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ public void testMetricsAreEmittedWhenRateLimiting() {
158158
// polaris_principal_roles_listPrincipalRoles_seconds_count{application="Polaris",class="org.apache.polaris.service.admin.api.PolarisPrincipalRolesApi",environment="prod",exception="none",method="listPrincipalRoles"} 50.0
159159

160160
Map<String, MetricFamily> metrics =
161-
TestMetricsUtil.fetchMetrics(fixture.client, testEnv.baseManagementUri(), "%s/q/metrics");
161+
TestMetricsUtil.fetchMetrics(fixture.client, testEnv.baseManagementUri());
162162

163163
assertThat(metrics)
164164
.isNotEmpty()

runtime/service/src/test/java/org/apache/polaris/service/test/DefaultTestEnvironmentResolver.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ public class DefaultTestEnvironmentResolver implements TestEnvironmentResolver {
2929
@Override
3030
public TestEnvironment resolveTestEnvironment(ExtensionContext extensionContext) {
3131
return new TestEnvironment(
32-
String.format("http://localhost:%d", localPort),
33-
String.format("http://localhost:%d", localManagementPort));
32+
String.format("http://localhost:%d/", localPort),
33+
String.format("http://localhost:%d/", localManagementPort));
3434
}
3535
}

runtime/service/src/test/java/org/apache/polaris/service/test/TestMetricsUtil.java

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,10 +36,8 @@
3636
/** Utils for working with metrics in tests */
3737
public class TestMetricsUtil {
3838

39-
public static Map<String, MetricFamily> fetchMetrics(
40-
Client client, URI baseManagementUri, String endpointPath) {
41-
Response response =
42-
client.target(String.format(endpointPath, baseManagementUri)).request().get();
39+
public static Map<String, MetricFamily> fetchMetrics(Client client, URI baseManagementUri) {
40+
Response response = client.target(baseManagementUri.resolve("q/metrics")).request().get();
4341
if (response.getStatus() == Status.MOVED_PERMANENTLY.getStatusCode()) {
4442
response = client.target(response.getLocation()).request().get();
4543
}

0 commit comments

Comments
 (0)