add more tests

Author: sfc-gh-ext-simba-jf

src/test/java/net/snowflake/client/jdbc/BindingDataLatestIT.java

@@ -177,6 +177,58 @@ public void testTimestampBindingWithNTZTypeExceedingBindingThreshold() throws SQ
         statement.execute("ALTER SESSION SET CLIENT_STAGE_ARRAY_BINDING_THRESHOLD = 1");
         executePrepStmtForNumRows(connection, "stageinsert", currT, 3);
 
+        // Compare the results
+        try (ResultSet rs1 = statement.executeQuery("select * from stageinsert");
+            ResultSet rs2 = statement.executeQuery("select * from regularinsert")) {
+          for (int i = 0; i < 3; i++) {
+            assertTrue(rs1.next());
+            assertTrue(rs2.next());
+
+            assertEquals(rs1.getInt(1), rs2.getInt(1));
+
+            // Check tz type and ltz type columns have the same value.
+            assertEquals(rs1.getTimestamp(2), rs1.getTimestamp(3));
+
+            assertEquals(rs2.getTimestamp(2), rs2.getTimestamp(2));
+            assertEquals(rs1.getTimestamp(3), rs1.getTimestamp(3));
+            assertEquals(rs2.getTimestamp(4), rs2.getTimestamp(4));
+          }
+        }
+      } finally {
+        statement.execute("drop table if exists stageinsert");
+        statement.execute("drop table if exists regularinsert");
+        TimeZone.setDefault(origTz);
+      }
+    }
+  }
+
+  /**
+   * Test that stage binding and regular binding insert and return the expected value for
+   * timestamp_ntz, timestamp_ltz, and timestamp_tz when not exceeding binding threshold.
+   *
+   * @throws SQLException if there is an error during query execution.
+   */
+  @Test
+  public void testTimestampBindingWithNTZTypeNotExceedingBindingThreshold() throws SQLException {
+    TimeZone.setDefault(tokyoTz);
+    try (Connection connection = getConnection();
+        Statement statement = connection.createStatement()) {
+      try {
+        statement.execute(
+            "create or replace table stageinsert(ind int, ltz0 timestamp_ltz, tz0 timestamp_tz, ntz0 timestamp_ntz)");
+        statement.execute(
+            "create or replace table regularinsert(ind int, ltz0 timestamp_ltz, tz0 timestamp_tz, ntz0 timestamp_ntz)");
+        statement.execute("alter session set CLIENT_TIMESTAMP_TYPE_MAPPING=TIMESTAMP_NTZ");
+        statement.execute("alter session set TIMEZONE='Asia/Tokyo'");
+        Timestamp currT = new Timestamp(System.currentTimeMillis());
+
+        // insert using regular binging
+        executePrepStmtForNumRows(connection, "regularinsert", currT, 3);
+
+        // insert using stage binding
+        statement.execute("ALTER SESSION SET CLIENT_STAGE_ARRAY_BINDING_THRESHOLD = 10");
+        executePrepStmtForNumRows(connection, "stageinsert", currT, 3);
+
         // Compare the results
         try (ResultSet rs1 = statement.executeQuery("select * from stageinsert");
             ResultSet rs2 = statement.executeQuery("select * from regularinsert")) {
@@ -208,7 +260,7 @@ private void executePrepStmtForNumRows(
     try (PreparedStatement prepStatement =
         connection.prepareStatement("insert into " + tableName + " values (?,?,?,?)")) {
       for (int i = 0; i < numRows; i++) {
-        prepStatement.setInt(1, 1);
+        prepStatement.setInt(1, i);
         prepStatement.setTimestamp(2, timestamp);
         prepStatement.setTimestamp(3, timestamp);
         prepStatement.setTimestamp(4, timestamp);