Added test cases to support Fedauth in connection resiliency case

Author: Divang Sharma

src/test/java/com/microsoft/sqlserver/jdbc/fedauth/FedauthCommon.java

@@ -145,7 +145,7 @@ public class FedauthCommon extends AbstractTest {
     static final String ERR_MSG_REQUEST_THROTTLED = "Request was throttled";
     static final String ERR_FAILED_FEDAUTH = TestResource.getResource("R_failedFedauth");
 
-    enum SqlAuthentication {
+    public enum SqlAuthentication {
         NotSpecified,
         SqlPassword,
         ActiveDirectoryPassword,

src/test/java/com/microsoft/sqlserver/jdbc/resiliency/BasicConnectionTest.java

@@ -491,6 +491,20 @@ public void testUnprocessedResponseCountSuccessfulIdleConnectionRecovery() throw
         }
     }
 
+    @Test
+    public void testAccessTokenInReconnect() throws SQLException {       
+        // Should use adIntegratedConnectionStr
+        try (Connection c = DriverManager.getConnection(adIntegratedConnectionStr);
+            Statement s = c.createStatement()) {
+                s.executeQuery("SELECT 1");
+                ResiliencyUtils.killConnection(c, adIntegratedConnectionStr, 0);
+                s.executeQuery("SELECT 1");
+            } catch (SQLException e) {
+                 fail(TestResource.getResource("R_expectedFailPassed"));
+            }
+    }
+
+
     private void basicReconnect(String connectionString) throws SQLException {
         // Ensure reconnects can happen multiple times over the same connection and subsequent connections
         for (int i1 = 0; i1 < 2; i1++) {

src/test/java/com/microsoft/sqlserver/testframework/AbstractTest.java

@@ -50,6 +50,7 @@
 import com.microsoft.sqlserver.jdbc.SQLServerXADataSource;
 import com.microsoft.sqlserver.jdbc.TestResource;
 import com.microsoft.sqlserver.jdbc.TestUtils;
+import com.microsoft.sqlserver.jdbc.fedauth.FedauthCommon.SqlAuthentication;
 
 
 /**
@@ -118,6 +119,10 @@ public abstract class AbstractTest {
     protected static String connectionString = null;
     protected static String connectionStringNTLM;
     protected static String connectionStringKerberos;
+    
+    protected static String azureServer = null;
+    protected static String azureDatabase = null;
+    protected static String adIntegratedConnectionStr;
 
     protected static ConfidentialClientApplication fedauthClientApp = null;
 
@@ -184,6 +189,11 @@ public static void setup() throws Exception {
         connectionString = TestUtils.addOrOverrideProperty(connectionString, "trustServerCertificate",
                 trustServerCertificate);
 
+        azureServer = getConfiguredProperty("azureServer");
+        azureDatabase = getConfiguredProperty("azureDatabase");
+        adIntegratedConnectionStr = "jdbc:sqlserver://" + azureServer + ";database=" + azureDatabase
+                + ";Authentication=" + SqlAuthentication.ActiveDirectoryIntegrated.toString();
+                
         javaKeyPath = TestUtils.getCurrentClassPath() + Constants.JKS_NAME;
 
         keyIDs = getConfiguredProperty("keyID", "").split(Constants.SEMI_COLON);