Update to use PolarisRealm annotation after rebase

apache · Sep 18, 2024 · 7ce9662 · 7ce9662
1 parent 514a2ab
commit 7ce9662
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/polaris-service/src/test/java/org/apache/polaris/service/ratelimiter/AsyncFallbackTest.java b/polaris-service/src/test/java/org/apache/polaris/service/ratelimiter/AsyncFallbackTest.java
@@ -25,9 +25,9 @@
 import jakarta.ws.rs.core.Response;
 import java.util.function.Consumer;
 import org.apache.polaris.service.PolarisApplication;
-import org.apache.polaris.service.PolarisApplicationIntegrationTest;
 import org.apache.polaris.service.config.PolarisApplicationConfig;
 import org.apache.polaris.service.test.PolarisConnectionExtension;
+import org.apache.polaris.service.test.PolarisRealm;
 import org.apache.polaris.service.test.SnowmanCredentialsExtension;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
@@ -61,8 +61,9 @@ public class AsyncFallbackTest {
   private static String realm;
 
   @BeforeAll
-  public static void setup(PolarisConnectionExtension.PolarisToken userToken) {
-    realm = PolarisConnectionExtension.getTestRealm(PolarisApplicationIntegrationTest.class);
+  public static void setup(
+      PolarisConnectionExtension.PolarisToken userToken, @PolarisRealm String polarisRealm) {
+    realm = polarisRealm;
     AsyncFallbackTest.userToken = userToken.token();
   }
 

diff --git a/...s-service/src/test/java/org/apache/polaris/service/ratelimiter/RateLimiterFilterTest.java b/...s-service/src/test/java/org/apache/polaris/service/ratelimiter/RateLimiterFilterTest.java
@@ -25,9 +25,9 @@
 import jakarta.ws.rs.core.Response;
 import java.util.function.Consumer;
 import org.apache.polaris.service.PolarisApplication;
-import org.apache.polaris.service.PolarisApplicationIntegrationTest;
 import org.apache.polaris.service.config.PolarisApplicationConfig;
 import org.apache.polaris.service.test.PolarisConnectionExtension;
+import org.apache.polaris.service.test.PolarisRealm;
 import org.apache.polaris.service.test.SnowmanCredentialsExtension;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
@@ -54,8 +54,9 @@ public class RateLimiterFilterTest {
   private static String realm;
 
   @BeforeAll
-  public static void setup(PolarisConnectionExtension.PolarisToken userToken) {
-    realm = PolarisConnectionExtension.getTestRealm(PolarisApplicationIntegrationTest.class);
+  public static void setup(
+      PolarisConnectionExtension.PolarisToken userToken, @PolarisRealm String polarisRealm) {
+    realm = polarisRealm;
     RateLimiterFilterTest.userToken = userToken.token();
   }