fix

databendlabs · Sep 5, 2024 · e153ada · e153ada
1 parent ae8b885
commit e153ada
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/databend-jdbc/src/test/java/com/databend/jdbc/TestMultiHost.java b/databend-jdbc/src/test/java/com/databend/jdbc/TestMultiHost.java
@@ -90,6 +90,10 @@ public void testTransactionInRobinLoadBalancing()
             DatabendStatement statement = (DatabendStatement) connection.createStatement();
             statement.execute("BEGIN");
             statement.execute("create table if not exists test (id int)");
+        } catch (SQLException e) {
+            e.printStackTrace();
+            // there should have exception because change query node
+            Assert.fail();
         }
         try (Connection connection = createConnection(ROUND_ROBIN_JDBC_URL)) {
             DatabendStatement statement = (DatabendStatement) connection.createStatement();

diff --git a/databend-jdbc/src/test/java/com/databend/jdbc/TestPrepareStatement.java b/databend-jdbc/src/test/java/com/databend/jdbc/TestPrepareStatement.java
@@ -548,7 +548,7 @@ public void testSelectWithClusterKey() throws SQLException {
         try (PreparedStatement statement = conn.prepareStatement(selectSQL)) {
             ResultSet rs = statement.executeQuery();
             while (rs.next()) {
-                Assertions.assertEquals("0.0", rs.getString(5));
+                Assertions.assertEquals("0", rs.getString(5));
             }
         }
     }