Updating test, PR feedback

Author: Benjin

src/Microsoft.SqlTools.ServiceLayer/Connection/ConnectionService.cs

@@ -1749,22 +1749,22 @@ await Task.Factory.StartNew(() => Task.Run(async () => {
         public ConnectionDetails ParseConnectionString(string connectionString)
         {
             SqlConnectionStringBuilder builder = new SqlConnectionStringBuilder(connectionString);
-            bool isAzureMFA = builder.Authentication == SqlAuthenticationMethod.ActiveDirectoryInteractive; // Azure Multi-Factor Authentication
-            bool isSqlLogin = builder.Authentication == SqlAuthenticationMethod.SqlPassword;// SQL username/password
-            bool isIntegrated = builder.IntegratedSecurity; // Windows Integrated Security
-            string Authenticationtype = null;
 
-            if (isAzureMFA)
+            string mappedAuthenticationType;
+            if (builder.IntegratedSecurity)
             {
-                Authenticationtype = "AzureMFA";
+                mappedAuthenticationType = "Integrated";
+            } else if (builder.Authentication == SqlAuthenticationMethod.ActiveDirectoryInteractive)
+            {
+                mappedAuthenticationType = "AzureMFA";
             }
-            else if(isSqlLogin)
+            else if (builder.Authentication == SqlAuthenticationMethod.SqlPassword || builder.Authentication == SqlAuthenticationMethod.NotSpecified)
             {
-                Authenticationtype = "SqlLogin";
+                mappedAuthenticationType = "SqlLogin";
             }
-            else if (isIntegrated)
+            else
             {
-                Authenticationtype = "Integrated";
+                mappedAuthenticationType = builder.Authentication.ToString();
             }
 
             // Set defaults as per MSSQL connection property defaults, not SqlClient's Connection string buider defaults
@@ -1773,7 +1773,7 @@ public ConnectionDetails ParseConnectionString(string connectionString)
                 ApplicationIntent = defaultBuilder.ApplicationIntent != builder.ApplicationIntent ? builder.ApplicationIntent.ToString() : null,
                 ApplicationName = defaultBuilder.ApplicationName != builder.ApplicationName ? builder.ApplicationName : ApplicationName,
                 AttachDbFilename = defaultBuilder.AttachDBFilename != builder.AttachDBFilename ? builder.AttachDBFilename.ToString() : null,
-                AuthenticationType = Authenticationtype,
+                AuthenticationType = mappedAuthenticationType,
                 ConnectRetryCount = defaultBuilder.ConnectRetryCount != builder.ConnectRetryCount ? builder.ConnectRetryCount : 1,
                 ConnectRetryInterval = defaultBuilder.ConnectRetryInterval != builder.ConnectRetryInterval ? builder.ConnectRetryInterval : 10,
                 ConnectTimeout = defaultBuilder.ConnectTimeout != builder.ConnectTimeout ? builder.ConnectTimeout : 30,

test/Microsoft.SqlTools.ServiceLayer.UnitTests/Connection/ConnectionServiceTests.cs

@@ -1863,18 +1863,24 @@ public void ParseConnectionStringTest_AuthTypes()
             details = service.ParseConnectionString(connectionString);
             Assert.That(details.AuthenticationType, Is.EqualTo("Integrated"));
 
-            // AAD auth types boil down to AzureMFA
-            connectionString = "Server=tcp:{servername},1433;Initial Catalog={databasename};Authentication=ActiveDirectoryIntegrated;";
+            // AAD auth types
+            // ActiveDirectoryInteractive is mapped to to AzureMFA
+            connectionString = "Server=tcp:{servername},1433;Initial Catalog={databasename};Authentication=ActiveDirectoryInteractive;";
             details = service.ParseConnectionString(connectionString);
             Assert.That(details.AuthenticationType, Is.EqualTo("AzureMFA"));
 
-            connectionString = "Server=tcp:{servername},1433;Initial Catalog={databasename};Authentication=ActiveDirectoryInteractive;";
+            // Other AAD types are unchanged
+            connectionString = "Server=tcp:{servername},1433;Initial Catalog={databasename};Authentication=ActiveDirectoryIntegrated;";
             details = service.ParseConnectionString(connectionString);
-            Assert.That(details.AuthenticationType, Is.EqualTo("AzureMFA"));
+            Assert.That(details.AuthenticationType, Is.EqualTo("ActiveDirectoryIntegrated"));
 
             connectionString = "Server=tcp:{servername},1433;Initial Catalog={databasename};Authentication=ActiveDirectoryDefault;";
             details = service.ParseConnectionString(connectionString);
-            Assert.That(details.AuthenticationType, Is.EqualTo("AzureMFA"));
+            Assert.That(details.AuthenticationType, Is.EqualTo("ActiveDirectoryDefault"));
+
+            // Invalid throws
+            connectionString = "Server=tcp:{servername},1433;Initial Catalog={databasename};Authentication=InvalidAuthType;";
+            Assert.Throws<ArgumentException>(() => service.ParseConnectionString(connectionString), "Invalid value for key 'authentication'.");
         }
 
         [Test]