Addressing feedback

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

@@ -506,7 +506,10 @@ namespace Microsoft.SqlTools.ServiceLayer.Connection
             {
                 try
                 {
-                    info.SqlConnection.ChangeDatabase(newDatabaseName);
+                    if (info.SqlConnection.State == ConnectionState.Open)
+                    {
+                        info.SqlConnection.ChangeDatabase(newDatabaseName);
+                    }
                     info.ConnectionDetails.DatabaseName = newDatabaseName;
 
                     // Fire a connection changed event

src/Microsoft.SqlTools.ServiceLayer/QueryExecution/Query.cs

@@ -21,6 +21,15 @@ namespace Microsoft.SqlTools.ServiceLayer.QueryExecution
     /// </summary>
     public class Query : IDisposable
     {
+        #region Constants
+        
+        /// <summary>
+        /// "Error" code produced by SQL Server when the database context (name) for a connection changes.
+        /// </summary>
+        private const int DatabaseContextChangeErrorNumber = 5701;
+
+        #endregion
+
         #region Properties
 
         /// <summary>
@@ -146,10 +155,10 @@ namespace Microsoft.SqlTools.ServiceLayer.QueryExecution
             {
                 await conn.OpenAsync();
 
-                if (conn.GetType() == typeof(SqlConnection))
+                SqlConnection sqlConn = conn as SqlConnection;
+                if (sqlConn != null)
                 {
                     // Subscribe to database informational messages
-                    SqlConnection sqlConn = conn as SqlConnection;
                     sqlConn.InfoMessage += OnInfoMessage;
                 }
 
@@ -161,11 +170,6 @@ namespace Microsoft.SqlTools.ServiceLayer.QueryExecution
             }
         }
 
-        /// <summary>
-        /// "Error" code produced by SQL Server when the database context (name) for a connection changes.
-        /// </summary>
-        private const int DatabaseContextChangeErrorNumber = 5701;
-
         /// <summary>
         /// Handler for database messages during query execution
         /// </summary>