Hook up the command again (#9587)

extensions/azurecore/src/account-provider/azureAccountProvider2.ts

@@ -44,6 +44,10 @@ export class AzureAccountProvider implements azdata.AccountProvider {
 	constructor(private metadata: AzureAccountProviderMetadata, private _tokenCache: TokenCache, private _context: vscode.ExtensionContext) {
 		this.commonAuthorityUrl = url.resolve(this.metadata.settings.host, AzureAccountProvider.AadCommonTenant);
 	}
+	// interface method
+	clearTokenCache(): Thenable<void> {
+		return this._tokenCache.clear();
+	}
 
 	// interface method
 	initialize(storedAccounts: azdata.Account[]): Thenable<azdata.Account[]> {

extensions/azurecore/src/account-provider/azureAccountProviderService.ts

@@ -70,8 +70,7 @@ export class AzureAccountProviderService implements vscode.Disposable {
 		// let self = this;
 
 		let promises: Thenable<void>[] = providerSettings.map(provider => {
-			// return self._accountProviders[provider.metadata.id].clearTokenCache();
+			return this._accountProviders[provider.metadata.id]?.clearTokenCache();
-			return Promise.resolve();
 		});
 
 		return Promise.all(promises)

src/sql/azdata.d.ts

@@ -2300,6 +2300,11 @@ declare module 'azdata' {
 		 * and call the end OAuth method.
 		 */
 		autoOAuthCancelled(): Thenable<void>;
+
+		/**
+		 * Clears token cache
+		 */
+		clearTokenCache(): Thenable<void>;
 	}
 
 	// Resource provider interfaces  -----------------------------------------------------------------------

src/sql/platform/accounts/test/common/testAccountManagementService.ts

@@ -77,6 +77,9 @@ export class TestAccountManagementService implements IAccountManagementService {
 }
 
 export class AccountProviderStub implements azdata.AccountProvider {
+	clearTokenCache(): Thenable<void> {
+		return Promise.resolve();
+	}
 	autoOAuthCancelled(): Thenable<void> {
 		return Promise.resolve();
 	}

src/sql/workbench/api/browser/mainThreadAccountManagement.ts

@@ -86,6 +86,9 @@ export class MainThreadAccountManagement extends Disposable implements MainThrea
 			},
 			refresh(account: azdata.Account): Thenable<azdata.Account | azdata.PromptFailedResult> {
 				return self._proxy.$refresh(handle, account);
+			},
+			clearTokenCache(): Thenable<void> {
+				return self._proxy.$clearTokenCache();
 			}
 		};
 		this._accountManagementService.registerProvider(providerMetadata, accountProvider);

src/sql/workbench/api/common/sqlExtHost.protocol.ts

@@ -35,6 +35,7 @@ export abstract class ExtHostAccountManagementShape {
 	$prompt(handle: number): Thenable<azdata.Account | azdata.PromptFailedResult> { throw ni(); }
 	$refresh(handle: number, account: azdata.Account): Thenable<azdata.Account | azdata.PromptFailedResult> { throw ni(); }
 	$accountsChanged(handle: number, accounts: azdata.Account[]): Thenable<void> { throw ni(); }
+	$clearTokenCache(): Thenable<void> { throw ni(); }
 }
 
 export abstract class ExtHostConnectionManagementShape {