test: add more accessor tests

Author: carsheng

jdbc-core/src/test/java/com/salesforce/datacloud/jdbc/core/DataCloudDatabaseMetadataTest.java

@@ -22,6 +22,7 @@
 import java.sql.SQLFeatureNotSupportedException;
 import java.sql.Statement;
 import java.util.Arrays;
+import java.util.HashMap;
 import java.util.UUID;
 import java.util.stream.Collectors;
 import lombok.SneakyThrows;
@@ -1078,6 +1079,7 @@ public void testMetadataColumnAccessors() throws SQLException {
 
             assertThat(columnResultSet.getObject("TYPE_NAME")).isEqualTo("VARCHAR");
             assertThat(columnResultSet.getObject("DATA_TYPE")).isEqualTo(12);
+            assertThat(columnResultSet.getObject("DATA_TYPE", new HashMap<>())).isEqualTo(12);
             assertThat(columnResultSet.getObject("TYPE_NAME", String.class)).isEqualTo("VARCHAR");
 
             assertThrows(SQLException.class, () -> columnResultSet.getObject("ORDINAL_POSITION", Boolean.class));

jdbc-core/src/test/java/com/salesforce/datacloud/jdbc/core/metadata/SimpleResultSetMetaDataTest.java

@@ -8,6 +8,8 @@
 
 import com.salesforce.datacloud.jdbc.core.QueryDBMetadata;
 import com.salesforce.datacloud.jdbc.core.SimpleMetadataResultSet;
+
+import java.math.BigDecimal;
 import java.sql.JDBCType;
 import java.sql.ResultSetMetaData;
 import java.sql.SQLException;
@@ -151,11 +153,8 @@ public void isDefinitelyWritable() throws SQLException {
     @Test
     public void getColumnClassName() throws SQLException {
         assertThat(simpleResultSetMetaData.getColumnClassName(1)).isEqualTo("java.lang.String");
-    }
+        assertThat(simpleResultSetMetaData.getColumnClassName(17)).isEqualTo("java.lang.Integer");
 
-    @Test
-    public void getWarnings() throws SQLException {
-        assertThat(simpleResultSetMetaData.getColumnClassName(1)).isEqualTo("java.lang.String");
     }
 
     @Test

jdbc-core/src/test/java/com/salesforce/datacloud/jdbc/core/resultset/ColumnAccessorTest.java

@@ -0,0 +1,26 @@
+package com.salesforce.datacloud.jdbc.core.resultset;
+
+import org.junit.jupiter.api.Test;
+import org.mockito.Mockito;
+
+import static org.junit.jupiter.api.Assertions.*;
+class ColumnAccessorTest {
+
+    @Test
+    public void shouldThrowUnsupportedError() {
+        ColumnAccessor<SimpleResultSet> columnAccessor = Mockito.mock(ColumnAccessor.class, Mockito.CALLS_REAL_METHODS);
+        SimpleResultSet resultSet = Mockito.mock(SimpleResultSet.class, Mockito.CALLS_REAL_METHODS);
+
+        // Test methods from SimpleResultSet that throw SQLFeatureNotSupportedException
+        assertThrows(UnsupportedOperationException.class, () -> columnAccessor.getBoolean(resultSet));
+        assertThrows(UnsupportedOperationException.class, () -> columnAccessor.getAnyInteger(resultSet));
+        assertThrows(UnsupportedOperationException.class, () -> columnAccessor.getBigDecimal(resultSet));
+        assertThrows(UnsupportedOperationException.class, () -> columnAccessor.getAnyFloatingPoint(resultSet));
+        assertThrows(UnsupportedOperationException.class, () -> columnAccessor.getString(resultSet));
+        assertThrows(UnsupportedOperationException.class, () -> columnAccessor.getBytes(resultSet));
+        assertThrows(UnsupportedOperationException.class, () -> columnAccessor.getDate(resultSet));
+        assertThrows(UnsupportedOperationException.class, () -> columnAccessor.getTime(resultSet));
+        assertThrows(UnsupportedOperationException.class, () -> columnAccessor.getTimestamp(resultSet));
+        assertThrows(UnsupportedOperationException.class, () -> columnAccessor.getArray(resultSet));
+    }
+}

jdbc-core/src/test/java/com/salesforce/datacloud/jdbc/core/resultset/SimpleResultSetTest.java

@@ -50,6 +50,9 @@ public void shouldThrowUnsupportedError() {
         assertThrows(UnsupportedOperationException.class, () -> resultSet.getDate(1, null));
         assertThrows(UnsupportedOperationException.class, () -> resultSet.getTimestamp(1, null));
         assertThrows(UnsupportedOperationException.class, () -> resultSet.getTime(1, null));
+        assertThrows(UnsupportedOperationException.class, () -> resultSet.getBigDecimal(1, 1));
+        assertThrows(UnsupportedOperationException.class, () -> resultSet.getBytes(1));
+        assertThrows(UnsupportedOperationException.class, () -> resultSet.getArray(1));
 
         // Test methods from ForwardOnlyResultSet interface
         assertThrows(SQLFeatureNotSupportedException.class, () -> resultSet.setFetchDirection(ResultSet.FETCH_REVERSE));