Tests | Added more datatype tests to increase code coverage

src/test/java/com/microsoft/sqlserver/jdbc/TestResource.java

@@ -153,6 +153,6 @@ protected Object[][] getContents() {
             {"R_expectedValue", "Expected value: "}, {"R_expectedValueAtIndex", "Expected value at index: "},
             {"R_switchFailed", "Switch case is not matched with data"},
             {"R_resultsetNotInstance", "Result set is not instance of SQLServerResultSet"},
-
+            {"R_noJRESupport", "No JRE support for {0}"},
     };
 }

src/test/java/com/microsoft/sqlserver/jdbc/TestUtils.java

@@ -66,6 +66,8 @@ public class TestUtils {
     public static final String SERVER_TYPE_SQL_AZURE = "SQLAzure";
     // private static SqlType types = null;
     private static ArrayList<SqlType> types = null;
+    private final static int ENGINE_EDITION_FOR_SQL_AZURE = 5;
+    private final static int ENGINE_EDITION_FOR_SQL_AZURE_DW = 6;
 
     /**
      * Returns serverType
@@ -623,6 +625,8 @@ public static Object roundSmallDateTimeValue(Object value) {
      * @return boolean
      */
     public static boolean serverSupportsDataClassification(Statement stmt) {
+
+
         try {
             stmt.execute("SELECT * FROM SYS.SENSITIVITY_CLASSIFICATIONS");
         } catch (SQLException e) {
@@ -666,4 +670,34 @@ public static boolean supportJDBC43(Connection con) throws SQLException {
     public static String escapeSingleQuotes(String name) {
         return name.replace("'", "''");
     }
+
+    /**
+     * Returns if connected to SQL Azure
+     * @param con
+     *        connection to server
+     * @return boolean
+     * @throws SQLException
+     */
+    public static boolean isSqlAzure(Connection con) throws SQLException {
+        try (ResultSet rs = con.createStatement().executeQuery("SELECT CAST(SERVERPROPERTY('EngineEdition') as INT)")) {
+            rs.next();
+            int engineEdition = rs.getInt(1);
+            return (engineEdition == ENGINE_EDITION_FOR_SQL_AZURE || engineEdition == ENGINE_EDITION_FOR_SQL_AZURE_DW);
+        }
+    }
+
+    /**
+     * Returns if connected to SQL Azure DW
+     * @param con
+     *        connection to server
+     * @return boolean
+     * @throws SQLException
+     */
+    public static boolean isSqlAzureDW(Connection con) throws SQLException {
+        try (ResultSet rs = con.createStatement().executeQuery("SELECT CAST(SERVERPROPERTY('EngineEdition') as INT)")) {
+            rs.next();
+            int engineEdition = rs.getInt(1);
+            return (engineEdition == ENGINE_EDITION_FOR_SQL_AZURE_DW);
+        }
+    }
 }

src/test/java/com/microsoft/sqlserver/jdbc/callablestatement/CallableMixedTest.java

@@ -0,0 +1,103 @@
+package com.microsoft.sqlserver.jdbc.callablestatement;
+
+import java.sql.*;
+
+import com.microsoft.sqlserver.jdbc.RandomUtil;
+import com.microsoft.sqlserver.jdbc.SQLServerDriver;
+import com.microsoft.sqlserver.jdbc.TestUtils;
+import com.microsoft.sqlserver.testframework.AbstractSQLGenerator;
+
+import org.junit.jupiter.api.Test;
+import org.junit.platform.runner.JUnitPlatform;
+import org.junit.runner.RunWith;
+import static org.junit.jupiter.api.Assertions.fail;
+
+
+@RunWith(JUnitPlatform.class)
+public class CallableMixedTest {
+
+    @Test
+    public void datatypestest() throws Exception {
+        Class.forName("com.microsoft.sqlserver.jdbc.SQLServerDriver");
+        String connectionString = TestUtils.getConfiguredProperty("mssql_jdbc_test_connection_properties");
+        String tableName = RandomUtil.getIdentifier("TFOO3");
+        String procName = RandomUtil.getIdentifier("SPFOO3");
+
+        try (Connection conn = DriverManager.getConnection(connectionString)) {
+            try (Statement stmt = conn.createStatement()) {
+                TestUtils.dropTableIfExists(AbstractSQLGenerator.escapeIdentifier(tableName), stmt);
+
+                String createSQL = "create table " + AbstractSQLGenerator.escapeIdentifier(tableName)
+                        + "(c1_int int primary key, col2 int)";
+                stmt.executeUpdate(createSQL);
+
+                stmt.executeUpdate("Insert into " + AbstractSQLGenerator.escapeIdentifier(tableName) + " values(0, 1)");
+
+                stmt.executeUpdate("CREATE PROCEDURE " + AbstractSQLGenerator.escapeIdentifier(procName)
+                        + " (@p2_int int, @p2_int_out int OUTPUT, @p4_smallint smallint,  @p4_smallint_out smallint OUTPUT) AS begin transaction SELECT * FROM "
+                        + AbstractSQLGenerator.escapeIdentifier(tableName)
+                        + "  ; SELECT @p2_int_out=@p2_int, @p4_smallint_out=@p4_smallint commit transaction RETURN -2147483648");
+            }
+
+            try (CallableStatement cstmt = conn.prepareCall(
+                    "{  ? = CALL " + AbstractSQLGenerator.escapeIdentifier(procName) + " (?, ?, ?, ?) }")) {
+                cstmt.registerOutParameter((int) 1, (int) 4);
+                cstmt.setObject((int) 2, Integer.valueOf("31"), (int) 4);
+                cstmt.registerOutParameter((int) 3, (int) 4);
+                cstmt.registerOutParameter((int) 5, java.sql.Types.BINARY); // Test OUT param
+                                                                            // re-registration
+                                                                            // (Defect 60921)
+                cstmt.registerOutParameter((int) 5, (int) 5);
+                cstmt.setObject((int) 4, Short.valueOf("-5372"), (int) 5);
+
+                // get results and a value
+                ResultSet rs = cstmt.executeQuery();
+                rs.next();
+
+                if (rs.getInt(1) != 0) {
+                    fail("Received data not equal to setdata");
+                }
+
+                if (cstmt.getInt((int) 5) != -5372) {
+                    fail("Received data not equal to setdata");
+                }
+
+                // do nothing and reexecute
+                rs = cstmt.executeQuery();
+                // get the param without getting the resultset
+                rs = cstmt.executeQuery();
+                if (cstmt.getInt((int) 1) != -2147483648) {
+                    fail("Received data not equal to setdata");
+                }
+
+                if (cstmt.getInt((int) 1) != -2147483648) {
+                    fail("Received data not equal to setdata");
+                }
+
+                rs = cstmt.executeQuery();
+                rs.next();
+
+                if (rs.getInt(1) != 0) {
+                    fail("Received data not equal to setdata");
+                }
+
+                if (cstmt.getInt((int) 1) != -2147483648) {
+                    fail("Received data not equal to setdata");
+                }
+
+                if (cstmt.getInt((int) 5) != -5372) {
+                    fail("Received data not equal to setdata");
+                }
+
+                rs = cstmt.executeQuery();
+            }
+        } finally {
+            try (Connection conn = DriverManager.getConnection(connectionString);
+                    Statement stmt = conn.createStatement()) {
+                TestUtils.dropTableIfExists(AbstractSQLGenerator.escapeIdentifier(tableName), stmt);
+            } catch (SQLException e) {
+                fail(e.toString());
+            }
+        }
+    }
+}