Hms standalone rest server with Spring Boot

itests/hive-iceberg/pom.xml

@@ -51,6 +51,12 @@
       <version>${keycloak.version}</version>
       <scope>test</scope>
     </dependency>
+    <dependency>
+      <groupId>jakarta.annotation</groupId>
+      <artifactId>jakarta.annotation-api</artifactId>
+      <version>${jakarta.annotation.version}</version>
+      <scope>test</scope>
+    </dependency>
     <dependency>
       <groupId>org.apache.hive</groupId>
       <artifactId>hive-standalone-metastore-common</artifactId>

itests/qtest-iceberg/pom.xml

@@ -475,6 +475,27 @@
       <version>${project.version}</version>
       <scope>test</scope>
     </dependency>
+    <!-- Spring Boot test dependencies -->
+    <dependency>
+      <groupId>org.springframework.boot</groupId>
+      <artifactId>spring-boot-starter-test</artifactId>
+      <version>${spring-boot.version}</version>
+      <scope>test</scope>
+      <exclusions>
+        <exclusion>
+          <groupId>org.springframework.boot</groupId>
+          <artifactId>spring-boot-starter-logging</artifactId>
+        </exclusion>
+        <exclusion>
+          <groupId>org.junit.jupiter</groupId>
+          <artifactId>junit-jupiter</artifactId>
+        </exclusion>
+        <exclusion>
+          <groupId>org.junit.vintage</groupId>
+          <artifactId>junit-vintage-engine</artifactId>
+        </exclusion>
+      </exclusions>
+    </dependency>
     <dependency>
       <groupId>org.testcontainers</groupId>
       <artifactId>testcontainers</artifactId>

itests/qtest-iceberg/src/test/java/org/apache/hadoop/hive/cli/BaseStandaloneRESTCatalogServerTest.java

@@ -0,0 +1,209 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.hadoop.hive.cli;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.UUID;
+
+import org.apache.hadoop.conf.Configuration;
+import org.apache.hadoop.hive.common.FileUtils;
+import org.apache.http.HttpHeaders;
+import org.apache.http.client.methods.CloseableHttpResponse;
+import org.apache.http.client.methods.HttpGet;
+import org.apache.http.client.methods.HttpPost;
+import org.apache.http.entity.StringEntity;
+import org.apache.http.impl.client.CloseableHttpClient;
+import org.apache.http.impl.client.HttpClients;
+import org.apache.http.util.EntityUtils;
+import org.apache.hadoop.hive.metastore.MetaStoreTestUtils;
+import org.apache.hadoop.hive.metastore.security.HadoopThriftAuthBridge;
+import org.apache.hadoop.hive.metastore.conf.MetastoreConf;
+import org.apache.hadoop.hive.metastore.conf.MetastoreConf.ConfVars;
+import org.apache.iceberg.rest.standalone.IcebergCatalogConfiguration;
+import org.apache.iceberg.rest.standalone.StandaloneRESTCatalogServer;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.context.annotation.Import;
+import org.springframework.core.Ordered;
+import org.springframework.core.annotation.Order;
+import org.springframework.test.context.TestContext;
+import org.springframework.test.context.TestExecutionListener;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * Base class for Standalone REST Catalog Server integration tests.
+ *
+ * Provides shared setup (HMS, listeners), HTTP helpers (with optional auth), and common tests
+ * (liveness, readiness, Prometheus, server port). Subclasses provide auth-specific configuration
+ * and tests.
+ */
+public abstract class BaseStandaloneRESTCatalogServerTest {
+  protected static final Logger LOG = LoggerFactory.getLogger(BaseStandaloneRESTCatalogServerTest.class);
+  private static final String REST_CATALOG_URL_TEMPLATE = "http://localhost:%d%s";
+
+  protected static Configuration hmsConf;
+  protected static int hmsPort;
+  protected static File warehouseDir;
+  protected static File hmsTempDir;
+
+  /**
+   * Starts HMS before the Spring ApplicationContext loads.
+   * Spring loads the context before @BeforeClass, so we use a TestExecutionListener
+   * which runs before context initialization.
+   */
+  @Order(Ordered.HIGHEST_PRECEDENCE)
+  public static class HmsStartupListener implements TestExecutionListener {
+    private static final String TEMP_DIR_PREFIX = "StandaloneRESTCatalogServer";
+
+    @Override
+    public void beforeTestClass(TestContext testContext) throws Exception {
+      if (hmsPort > 0) {
+        return;
+      }
+      String uniqueTestKey = String.format("%s_%s", TEMP_DIR_PREFIX, UUID.randomUUID());
+      hmsTempDir = new File(MetaStoreTestUtils.getTestWarehouseDir(uniqueTestKey));
+      hmsTempDir.mkdirs();
+      warehouseDir = new File(hmsTempDir, "warehouse");
+      warehouseDir.mkdirs();
+
+      hmsConf = MetastoreConf.newMetastoreConf();
+      MetaStoreTestUtils.setConfForStandloneMode(hmsConf);
+
+      String jdbcUrl = String.format("jdbc:derby:memory:%s;create=true",
+          new File(hmsTempDir, "metastore_db").getAbsolutePath());
+      MetastoreConf.setVar(hmsConf, ConfVars.CONNECT_URL_KEY, jdbcUrl);
+      MetastoreConf.setVar(hmsConf, ConfVars.WAREHOUSE, warehouseDir.getAbsolutePath());
+      MetastoreConf.setVar(hmsConf, ConfVars.WAREHOUSE_EXTERNAL, warehouseDir.getAbsolutePath());
+
+      hmsPort = MetaStoreTestUtils.startMetaStoreWithRetry(
+          HadoopThriftAuthBridge.getBridge(), hmsConf, true, false, false, false);
+      LOG.info("Started embedded HMS on port: {} (before Spring context)", hmsPort);
+    }
+  }
+
+  @SpringBootApplication
+  @Import(IcebergCatalogConfiguration.class)
+  public static class TestRestCatalogApplication {}
+
+  protected String url(String path) {
+    return String.format(REST_CATALOG_URL_TEMPLATE, getPort(), path);
+  }
+
+  /**
+   * Returns the server port. Subclasses must provide this (e.g. from @LocalServerPort).
+   */
+  protected abstract int getPort();
+
+  /**
+   * Creates a GET request with optional Bearer token.
+   *
+   * @param path the request path (e.g. "/iceberg/v1/config")
+   * @param bearerToken optional Bearer token; if null, no Authorization header is set
+   */
+  protected HttpGet get(String path, String bearerToken) {
+    HttpGet request = new HttpGet(url(path));
+    request.setHeader("Content-Type", "application/json");
+    if (bearerToken != null) {
+      request.setHeader(HttpHeaders.AUTHORIZATION, "Bearer " + bearerToken);
+    }
+    return request;
+  }
+
+  /**
+   * Creates a GET request without auth.
+   */
+  protected HttpGet get(String path) {
+    return get(path, null);
+  }
+
+  /**
+   * Creates a POST request with optional Bearer token.
+   *
+   * @param path the request path
+   * @param jsonBody the JSON body
+   * @param bearerToken optional Bearer token; if null, no Authorization header is set
+   */
+  protected HttpPost post(String path, String jsonBody, String bearerToken) {
+    HttpPost request = new HttpPost(url(path));
+    request.setHeader("Content-Type", "application/json");
+    if (bearerToken != null) {
+      request.setHeader(HttpHeaders.AUTHORIZATION, "Bearer " + bearerToken);
+    }
+    if (jsonBody != null) {
+      request.setEntity(new StringEntity(jsonBody, "UTF-8"));
+    }
+    return request;
+  }
+
+  /**
+   * Creates a POST request without auth.
+   */
+  protected HttpPost post(String path, String jsonBody) {
+    return post(path, jsonBody, null);
+  }
+
+  protected static void teardownBase() throws IOException {
+    if (hmsPort > 0) {
+      MetaStoreTestUtils.close(hmsPort);
+    }
+    if (hmsTempDir != null && hmsTempDir.exists()) {
+      FileUtils.deleteDirectory(hmsTempDir);
+    }
+  }
+
+  protected void testLivenessProbe() throws Exception {
+    try (CloseableHttpClient httpClient = HttpClients.createDefault();
+        CloseableHttpResponse response = httpClient.execute(get("/actuator/health/liveness"))) {
+      assertEquals("Liveness probe should return 200", 200, response.getStatusLine().getStatusCode());
+      String body = EntityUtils.toString(response.getEntity());
+      assertTrue("Liveness should be UP", body.contains("UP"));
+      LOG.info("Liveness probe passed: {}", body);
+    }
+  }
+
+  protected void testReadinessProbe() throws Exception {
+    try (CloseableHttpClient httpClient = HttpClients.createDefault();
+        CloseableHttpResponse response = httpClient.execute(get("/actuator/health/readiness"))) {
+      assertEquals("Readiness probe should return 200", 200, response.getStatusLine().getStatusCode());
+      String body = EntityUtils.toString(response.getEntity());
+      assertTrue("Readiness should be UP", body.contains("UP"));
+      LOG.info("Readiness probe passed: {}", body);
+    }
+  }
+
+  protected void testPrometheusMetrics() throws Exception {
+    try (CloseableHttpClient httpClient = HttpClients.createDefault();
+        CloseableHttpResponse response = httpClient.execute(get("/actuator/prometheus"))) {
+      assertEquals("Metrics endpoint should return 200", 200, response.getStatusLine().getStatusCode());
+      String body = EntityUtils.toString(response.getEntity());
+      assertTrue("Should contain JVM metrics", body.contains("jvm_memory"));
+      LOG.info("Prometheus metrics available");
+    }
+  }
+
+  protected void testServerPort(StandaloneRESTCatalogServer server) {
+    assertTrue("Server port should be > 0", getPort() > 0);
+    assertNotNull("REST endpoint should not be null", server.getRestEndpoint());
+    LOG.info("Server port: {}, Endpoint: {}", getPort(), server.getRestEndpoint());
+  }
+}