diff --git a/java/lance-namespace-hive2/src/main/java/org/lance/namespace/hive2/Hive2Namespace.java b/java/lance-namespace-hive2/src/main/java/org/lance/namespace/hive2/Hive2Namespace.java index 19dea63..07047f3 100644 --- a/java/lance-namespace-hive2/src/main/java/org/lance/namespace/hive2/Hive2Namespace.java +++ b/java/lance-namespace-hive2/src/main/java/org/lance/namespace/hive2/Hive2Namespace.java @@ -91,6 +91,10 @@ public void setHadoopConf(Configuration conf) { this.hadoopConf = conf; } + public Hive2ClientPool getClientPool() { + return clientPool; + } + @Override public void initialize(Map configProperties, BufferAllocator allocator) { this.allocator = allocator; diff --git a/java/lance-namespace-hive2/src/test/java/org/lance/namespace/hive2/TestHive2Namespace.java b/java/lance-namespace-hive2/src/test/java/org/lance/namespace/hive2/TestHive2Namespace.java index 991d947..91477d5 100644 --- a/java/lance-namespace-hive2/src/test/java/org/lance/namespace/hive2/TestHive2Namespace.java +++ b/java/lance-namespace-hive2/src/test/java/org/lance/namespace/hive2/TestHive2Namespace.java @@ -55,6 +55,7 @@ import static java.nio.file.attribute.PosixFilePermissions.fromString; import static org.junit.jupiter.api.Assertions.assertEquals; import static org.junit.jupiter.api.Assertions.assertFalse; +import static org.junit.jupiter.api.Assertions.assertNotNull; import static org.junit.jupiter.api.Assertions.assertThrows; import static org.junit.jupiter.api.Assertions.assertTrue; @@ -103,6 +104,11 @@ public void cleanup() throws Exception { metastore.reset(); } + @Test + public void testGetClientPool() { + assertNotNull(((Hive2Namespace) namespace).getClientPool()); + } + @Test public void testDescribeNonExistentTable() { // Setup: Create database diff --git a/java/lance-namespace-hive3/src/main/java/org/lance/namespace/hive3/Hive3Namespace.java b/java/lance-namespace-hive3/src/main/java/org/lance/namespace/hive3/Hive3Namespace.java index a22d9bf..ed12ab2 100644 --- a/java/lance-namespace-hive3/src/main/java/org/lance/namespace/hive3/Hive3Namespace.java +++ b/java/lance-namespace-hive3/src/main/java/org/lance/namespace/hive3/Hive3Namespace.java @@ -82,6 +82,10 @@ public void setHadoopConf(Configuration conf) { this.hadoopConf = conf; } + public Hive3ClientPool getClientPool() { + return clientPool; + } + @Override public void initialize(Map configProperties, BufferAllocator allocator) { this.allocator = allocator; diff --git a/java/lance-namespace-hive3/src/test/java/org/lance/namespace/hive3/TestHive3Namespace.java b/java/lance-namespace-hive3/src/test/java/org/lance/namespace/hive3/TestHive3Namespace.java index 47a36e0..e2fb051 100644 --- a/java/lance-namespace-hive3/src/test/java/org/lance/namespace/hive3/TestHive3Namespace.java +++ b/java/lance-namespace-hive3/src/test/java/org/lance/namespace/hive3/TestHive3Namespace.java @@ -56,6 +56,7 @@ import static java.nio.file.attribute.PosixFilePermissions.fromString; import static org.junit.jupiter.api.Assertions.assertEquals; import static org.junit.jupiter.api.Assertions.assertFalse; +import static org.junit.jupiter.api.Assertions.assertNotNull; import static org.junit.jupiter.api.Assertions.assertThrows; import static org.junit.jupiter.api.Assertions.assertTrue; @@ -128,6 +129,11 @@ public void cleanup() throws Exception { namespace.createNamespace(nsRequest); } + @Test + public void testGetClientPool() { + assertNotNull(((Hive3Namespace) namespace).getClientPool()); + } + @Test public void testDescribeNonExistentTable() { // Test: Describe non-existent table