Adding tests and a helper method

ClickHouse · Jun 20, 2024 · b061b48 · b061b48
1 parent d0853b4
commit b061b48
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/clickhouse-client/src/main/java/com/clickhouse/client/ClickHouseResponseSummary.java b/clickhouse-client/src/main/java/com/clickhouse/client/ClickHouseResponseSummary.java
@@ -345,6 +345,10 @@ public long getResultRows() {
         return progress.get().getResultRows();
     }
 
+    public String getQueryId() {
+        return progress.get().getQueryId();
+    }
+
     public boolean isEmpty() {
         return progress.get().isEmpty() && stats.get().isEmpty();
     }

diff --git a/...khouse-http-client/src/test/java/com/clickhouse/client/http/ClickHouseHttpClientTest.java b/...khouse-http-client/src/test/java/com/clickhouse/client/http/ClickHouseHttpClientTest.java
@@ -414,6 +414,21 @@ public void testPost() throws ClickHouseException {
         }
     }
 
+    @Test(groups = {"integration"})
+    public void testQueryId() throws ClickHouseException {
+        ClickHouseNode server = getServer(ClickHouseProtocol.HTTP);
+        String uuid = UUID.randomUUID().toString();
+
+        try (ClickHouseClient client = ClickHouseClient.builder().options(getClientOptions())
+                .defaultCredentials(ClickHouseCredentials.fromUserAndPassword("foo", "bar")).build()) {
+            try (ClickHouseResponse resp = newRequest(client, server).compressServerResponse(false)
+                    .format(ClickHouseFormat.RowBinaryWithNamesAndTypes)
+                    .query("select 1,2", uuid).executeAndWait()) {
+                Assert.assertEquals(resp.getSummary().getQueryId(), uuid);
+            }
+        }
+    }
+
     @Test(groups = {"integration"})
     public void testProxyConnection() throws ClickHouseException, IOException {
         ToxiproxyContainer toxiproxy = null;