From c7cdf67b1436a13e5c11f182cbbd786928c73434 Mon Sep 17 00:00:00 2001 From: Karl Burtram Date: Thu, 9 Sep 2021 09:45:03 -0700 Subject: [PATCH] Properly set device code auth type (#17027) --- .../azurecore/src/account-provider/auths/azureDeviceCode.ts | 2 +- .../azurecore/src/account-provider/azureAccountProvider.ts | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/extensions/azurecore/src/account-provider/auths/azureDeviceCode.ts b/extensions/azurecore/src/account-provider/auths/azureDeviceCode.ts index 01da492b88..df9bf180fd 100644 --- a/extensions/azurecore/src/account-provider/auths/azureDeviceCode.ts +++ b/extensions/azurecore/src/account-provider/auths/azureDeviceCode.ts @@ -51,7 +51,7 @@ export class AzureDeviceCode extends AzureAuth { context: vscode.ExtensionContext, uriEventEmitter: vscode.EventEmitter, ) { - super(metadata, tokenCache, context, uriEventEmitter, AzureAuthType.AuthCodeGrant, AzureDeviceCode.USER_FRIENDLY_NAME); + super(metadata, tokenCache, context, uriEventEmitter, AzureAuthType.DeviceCode, AzureDeviceCode.USER_FRIENDLY_NAME); this.pageTitle = localize('addAccount', "Add {0} account", this.metadata.displayName); } diff --git a/extensions/azurecore/src/account-provider/azureAccountProvider.ts b/extensions/azurecore/src/account-provider/azureAccountProvider.ts index 2915c24ed9..09d3d80e78 100644 --- a/extensions/azurecore/src/account-provider/azureAccountProvider.ts +++ b/extensions/azurecore/src/account-provider/azureAccountProvider.ts @@ -113,7 +113,7 @@ export class AzureAccountProvider implements azdata.AccountProvider, vscode.Disp private async _getAccountSecurityToken(account: AzureAccount, tenantId: string, resource: azdata.AzureResource): Promise { await this.initCompletePromise; - const azureAuth = this.getAuthMethod(undefined); + const azureAuth = this.getAuthMethod(account); Logger.pii(`Getting account security token for ${JSON.stringify(account.key)} (tenant ${tenantId}). Auth Method = ${azureAuth.userFriendlyName}`, [], []); return azureAuth?.getAccountSecurityToken(account, tenantId, resource); }