Add refresh support to the test explorer

Author: vinistock

vscode/src/test/suite/testController.test.ts

@@ -898,4 +898,22 @@ suite("TestController", () => {
     assert.ok(runStub.passed.calledWith(testItem));
     assert.ok(runStub.end.calledWithExactly());
   }).timeout(10000);
+
+  test("refresh handler clears all items and starts from scratch", async () => {
+    await controller.testController.resolveHandler!(undefined);
+    assert.ok(controller.testController.items.size > 0);
+
+    const replaceSpy = sandbox.spy(controller.testController.items, "replace");
+    const resolveHandlerStub = sandbox.stub().resolves(true);
+    controller.testController.resolveHandler = resolveHandlerStub;
+
+    const source = new vscode.CancellationTokenSource();
+    await controller.testController.refreshHandler!(source.token);
+
+    assert.ok(replaceSpy.calledOnce);
+    assert.ok(replaceSpy.calledWith([]));
+    assert.ok(resolveHandlerStub.calledOnce);
+
+    source.dispose();
+  });
 });

vscode/src/testController.ts

@@ -86,6 +86,7 @@ export class TestController {
 
     if (this.fullDiscovery) {
       this.testController.resolveHandler = this.resolveHandler.bind(this);
+      this.testController.refreshHandler = this.refreshHandler.bind(this);
     }
 
     this.testCommands = new WeakMap<vscode.TestItem, string>();
@@ -510,6 +511,12 @@ export class TestController {
     return this.runner.tcpPort;
   }
 
+  private async refreshHandler(_token: vscode.CancellationToken) {
+    this.testController.items.replace([]);
+    this.testController.invalidateTestResults();
+    await this.testController.resolveHandler!(undefined);
+  }
+
   private async handleTests(
     request: vscode.TestRunRequest,
     token: vscode.CancellationToken,