From bafbca492bf32e83fa70d60f477bdc3a4e8db150 Mon Sep 17 00:00:00 2001 From: Christopher Suh Date: Thu, 17 Feb 2022 17:25:05 -0800 Subject: [PATCH] Clear account tokens before refresh (#18452) * clear account tokens on refresh * update refresh button after refresh --- .../azurecore/src/account-provider/azureAccountProvider.ts | 5 +++++ .../services/connection/browser/connectionWidget.ts | 1 + 2 files changed, 6 insertions(+) diff --git a/extensions/azurecore/src/account-provider/azureAccountProvider.ts b/extensions/azurecore/src/account-provider/azureAccountProvider.ts index 8a572ee297..a0fe4040f0 100644 --- a/extensions/azurecore/src/account-provider/azureAccountProvider.ts +++ b/extensions/azurecore/src/account-provider/azureAccountProvider.ts @@ -172,6 +172,11 @@ export class AzureAccountProvider implements azdata.AccountProvider, vscode.Disp } refresh(account: AzureAccount): Thenable { + return this._refresh(account); + } + + private async _refresh(account: AzureAccount): Promise { + await this._clear(account.key); return this.prompt(); } diff --git a/src/sql/workbench/services/connection/browser/connectionWidget.ts b/src/sql/workbench/services/connection/browser/connectionWidget.ts index 74f8d2642f..9f50c94b06 100644 --- a/src/sql/workbench/services/connection/browser/connectionWidget.ts +++ b/src/sql/workbench/services/connection/browser/connectionWidget.ts @@ -418,6 +418,7 @@ export class ConnectionWidget extends lifecycle.Disposable { if (account) { await this._accountManagementService.refreshAccount(account); await this.fillInAzureAccountOptions(); + this.updateRefreshCredentialsLink(); } })); }