fixing JWT tests

Author: chernser

.github/workflows/build.yml

@@ -278,6 +278,7 @@ jobs:
           CLICKHOUSE_CLOUD_HOST: ${{ secrets.INTEGRATIONS_TEAM_TESTS_CLOUD_HOST_SMT_PROD }}
           CLICKHOUSE_CLOUD_PASSWORD: ${{ secrets.INTEGRATIONS_TEAM_TESTS_CLOUD_PASSWORD_SMT_PROD }}
           CLIENT_JWT: ${{ secrets.INTEGRATIONS_TEAM_TESTS_CLOUD_JWT_DESERT_VM_43 }}
+          JWT_TEST_HOST: ${{ secrets.INTEGRATIONS_TEAM_TESTS_CLOUD_HOST_SMT }}
         run: |
           mvn --batch-mode --no-transfer-progress --projects ${{ matrix.project }} -DclickhouseVersion=${{ matrix.clickhouse }} -Dprotocol=http -Dmaven.javadoc.skip=true verify
       - name: Upload test results

client-v2/src/test/java/com/clickhouse/client/HttpTransportTests.java

@@ -1218,10 +1218,17 @@ public void testJWTWithCloud() throws Exception {
         if (!isCloud()) {
             return; // only for cloud
         }
-        String jwt = System.getenv("CLIENT_JWT");
+        final String jwt = System.getenv("CLIENT_JWT");
+        final String host = System.getenv("JWT_TEST_HOST");
         Assert.assertTrue(jwt != null && !jwt.trim().isEmpty(), "JWT is missing");
         Assert.assertFalse(jwt.contains("\n") || jwt.contains("-----"), "JWT should be single string ready for HTTP header");
-        try (Client client = newClient().useBearerTokenAuth(jwt).build()) {
+        try (Client client = new Client.Builder()
+                .addEndpoint(Protocol.HTTP, host, 8443, true)
+                .setUsername("default")
+                .compressClientRequest(false)
+                .setDefaultDatabase("default")
+                .serverSetting(ServerSettings.WAIT_END_OF_QUERY, "1")
+                .useBearerTokenAuth(jwt).build()) {
             try {
                 List<GenericRecord> response = client.queryAll("SELECT user(), now()");
                 System.out.println("response: " + response.get(0).getString(1) + " time: " + response.get(0).getString(2));