Add tag xAzureSQLDB

Author: muskan124947

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

@@ -5,6 +5,7 @@
 package com.microsoft.sqlserver.jdbc;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
 import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 import static org.junit.jupiter.api.Assertions.fail;
@@ -1370,4 +1371,19 @@ public void testGetSqlFedAuthTokenFailureNagativeWaiting() throws SQLException {
         }        
     }
 
+    @Test
+    @Tag(Constants.xAzureSQLDB)
+    public void testManagedIdentityWithEncryptStrict() {
+        SQLServerDataSource ds = new SQLServerDataSource();
+        ds.setURL(connectionString);
+        ds.setAuthentication("ActiveDirectoryMSI");
+        ds.setEncrypt("strict");
+
+        try (Connection con = ds.getConnection()) {
+            assertTrue(con.isValid(0), "Connection should be valid");
+        } catch (SQLException e) {
+            fail("Connection failed: " + e.getMessage());
+        }
+    }
+
 }

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

@@ -505,19 +505,6 @@ public void testAccessTokenCache() {
 
     }
 
-    @Test
-    public void testManagedIdentityWithEncryptStrict() throws SQLException {
-        String url = "jdbc:sqlserver://" + azureServer + ";database=" + azureDatabase + ";authentication=ActiveDirectoryMSI;encrypt=strict;";
-        SQLServerDataSource ds = new SQLServerDataSource();
-        ds.setURL(url);
-
-        try (Connection conn = ds.getConnection()) {
-            assertNotNull(conn);
-        } catch (SQLException e) {
-            fail("Expected connection to succeed with encrypt=strict, but it failed with error: " + e.getMessage());
-        }
-    }
-
     private static void validateException(String url, String resourceKey) {
         try (Connection conn = DriverManager.getConnection(url)) {
             fail(TestResource.getResource("R_expectedFailPassed"));