Pass environment and endpoint (fixes #103)

microsoft · Nov 7, 2018 · a954e04 · a954e04
1 parent d8de6da
commit a954e04
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/azure-account.ts b/src/azure-account.ts
@@ -715,8 +715,8 @@ export async function tokenFromRefreshToken(environment: AzureEnvironment, refre
 async function tokensFromToken(environment: AzureEnvironment, firstTokenResponse: TokenResponse) {
 	const tokenCache = new MemoryCache();
 	await addTokenToCache(environment, tokenCache, firstTokenResponse);
-	const credentials = new DeviceTokenCredentials({ username: firstTokenResponse.userId, clientId, tokenCache });
-	const client = new SubscriptionClient.SubscriptionClient(credentials);
+	const credentials = new DeviceTokenCredentials({ username: firstTokenResponse.userId, clientId, tokenCache, environment });
+	const client = new SubscriptionClient.SubscriptionClient(credentials, environment.resourceManagerEndpointUrl);
 	const tenants = await listAll(client.tenants, client.tenants.list());
 	const responses = <TokenResponse[]>(await Promise.all<TokenResponse | null>(tenants.map((tenant, i) => {
 		if (tenant.tenantId === firstTokenResponse.tenantId) {