fixed in statement with multiple arguments

Author: chernser

jdbc-v2/src/test/java/com/clickhouse/jdbc/PreparedStatementTest.java

@@ -1375,9 +1375,9 @@ public void testCTEWithUnboundCol() throws Exception {
     public void testWithInClause() throws Exception {
 
         try (Connection conn = getJdbcConnection()) {
-            String cte = "select number from system.numbers where number in (?) limit 10";
-            Long[] filter =  new Long[]{2L, 4L, 6L};
-            try (PreparedStatement stmt = conn.prepareStatement(cte)) {
+            final String q1 = "select number from system.numbers where number in (?) limit 10";
+            try (PreparedStatement stmt = conn.prepareStatement(q1)) {
+                Long[] filter =  new Long[]{2L, 4L, 6L};
                 stmt.setArray(1, conn.createArrayOf("Int64", filter));
                 ResultSet rs = stmt.executeQuery();
 
@@ -1387,6 +1387,21 @@ public void testWithInClause() throws Exception {
                 }
                 Assert.assertFalse(rs.next());
             }
+
+            final String q2 = "with t as (select arrayJoin([1, 2, 3]) as a )  select * from t where a in(?, ?)";
+            try (PreparedStatement stmt = conn.prepareStatement(q2)) {
+                Long[] filter =  new Long[]{2L, 3L};
+
+                stmt.setInt(1, 2);
+                stmt.setInt(2, 3);
+                ResultSet rs = stmt.executeQuery();
+
+                for (Long filterValue : filter) {
+                    assertTrue(rs.next());
+                    assertEquals(rs.getLong(1), filterValue);
+                }
+                Assert.assertFalse(rs.next());
+            }
         }
     }
 

jdbc-v2/src/test/java/com/clickhouse/jdbc/internal/BaseSqlParserFacadeTest.java

@@ -283,6 +283,7 @@ public Object[][] testMiscStmtDp() {
             {"alter table user delete where reg_time = ?", 1},
             {"SELECT * FROM a,b WHERE id > ?", 1},
             {"select ip from myusers where tenant=?", 1},
+            {"SELECT myColumn FROM myTable WHERE myColumn in (?, ?, ?)", 3},
             {"DROP USER IF EXISTS default_impersonation_user", 0},
             {"DROP ROLE IF EXISTS `vkonfwxapllzkkgkqdvt`", 0},
             {"CREATE ROLE `kjxrsscptauligukwgmf` ON CLUSTER '{cluster}'", 0},