From a584aca96997a18ecedee73a575536332e542718 Mon Sep 17 00:00:00 2001
From: Chris LaFreniere <40371649+chlafreniere@users.noreply.github.com>
Date: Wed, 18 Sep 2019 17:21:32 -0700
Subject: [PATCH] Stop clearing out connecting and connected state when
 cancelling out of connection dialog (#7254)

* Stop clearing out connecting and connected state

* tweaks

* Handling cancel while connecting

* fix typo

* PR comments
---
 src/sql/workbench/parts/query/common/queryInput.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/sql/workbench/parts/query/common/queryInput.ts b/src/sql/workbench/parts/query/common/queryInput.ts
index fb4523a164..19e537b44d 100644
--- a/src/sql/workbench/parts/query/common/queryInput.ts
+++ b/src/sql/workbench/parts/query/common/queryInput.ts
@@ -302,8 +302,12 @@ export class QueryInput extends EditorInput implements IEncodingSupport, IConnec
 	}
 
 	public onConnectCanceled(): void {
+		// If we're currently connecting and then cancel, set connected state to false
+		// Otherwise, keep connected state as it was
+		if (this.state.connecting) {
+			this.state.connected = false;
+		}
 		this.state.connecting = false;
-		this.state.connected = false;
 	}
 
 	public onConnectSuccess(params?: INewConnectionParams): void {