Fix access token refresh design (ADS only) (#21206)

This commit is contained in:
Cheena Malhotra
2022-11-21 19:35:00 -08:00
committed by GitHub
parent 8db40ab55f
commit 3b6ce47acc
5 changed files with 67 additions and 39 deletions

View File

@@ -403,6 +403,7 @@ suite('SQL Connection Tree Action tests', () => {
resolve(connection);
}));
connectionManagementService.setup(x => x.isConnected(undefined, TypeMoq.It.isAny())).returns(() => isConnectedReturnValue);
connectionManagementService.setup(x => x.refreshAzureAccountTokenIfNecessary(TypeMoq.It.isAny())).returns(async () => true);
let objectExplorerSession = {
success: true,
@@ -449,6 +450,7 @@ suite('SQL Connection Tree Action tests', () => {
return refreshAction.run().then((value) => {
connectionManagementService.verify(x => x.isConnected(undefined, TypeMoq.It.isAny()), TypeMoq.Times.atLeastOnce());
connectionManagementService.verify(x => x.refreshAzureAccountTokenIfNecessary(TypeMoq.It.isAny()), TypeMoq.Times.atLeastOnce());
objectExplorerService.verify(x => x.getObjectExplorerNode(TypeMoq.It.isAny()), TypeMoq.Times.atLeastOnce());
objectExplorerService.verify(x => x.refreshTreeNode(TypeMoq.It.isAny(), TypeMoq.It.isAny()), TypeMoq.Times.exactly(0));
tree.verify(x => x.refresh(TypeMoq.It.isAny()), TypeMoq.Times.atLeastOnce());