Add test run durations to status updates

Author: vinistock

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

@@ -857,9 +857,9 @@ suite("TestController", () => {
       const cancellationSource = new vscode.CancellationTokenSource();
       await controller.runTest(runRequest, cancellationSource.token);
 
-      assert.ok(runStub.enqueued.calledWithExactly(testItem));
-      assert.ok(runStub.started.calledWithExactly(testItem));
-      assert.ok(runStub.passed.calledWithExactly(testItem));
+      assert.ok(runStub.enqueued.calledWith(testItem));
+      assert.ok(runStub.started.calledWith(testItem));
+      assert.ok(runStub.passed.calledWith(testItem));
       assert.ok(runStub.end.calledWithExactly());
 
       createRunStub.restore();
@@ -1008,9 +1008,9 @@ suite("TestController", () => {
       await controller.runTest(runRequest, cancellationSource.token);
       fsStub.restore();
 
-      assert.ok(runStub.enqueued.calledWithExactly(testItem));
-      assert.ok(runStub.started.calledWithExactly(testItem));
-      assert.ok(runStub.passed.calledWithExactly(testItem));
+      assert.ok(runStub.enqueued.calledWith(testItem));
+      assert.ok(runStub.started.calledWith(testItem));
+      assert.ok(runStub.passed.calledWith(testItem));
       assert.ok(runStub.end.calledWithExactly());
       assert.ok(
         runStub.appendOutput.calledWithExactly(

vscode/src/testController.ts

@@ -1275,6 +1275,15 @@ export class TestController {
   ) {
     await new Promise<void>((resolve, reject) => {
       const promises: Promise<void>[] = [];
+      const startTimestamps = new Map<string, number>();
+      const withDuration = (
+        id: string,
+        callback: (duration?: number) => void,
+      ) => {
+        const startTime = startTimestamps.get(id);
+        const duration = startTime ? Date.now() - startTime : undefined;
+        callback(duration);
+      };
 
       const abortController = new AbortController();
       token.onCancellationRequested(() => {
@@ -1326,6 +1335,7 @@ export class TestController {
               (test) => {
                 if (test) {
                   run.started(test);
+                  startTimestamps.set(test.id, Date.now());
                 }
               },
             ),
@@ -1339,7 +1349,9 @@ export class TestController {
             this.findTestItem(params.id, vscode.Uri.parse(params.uri)).then(
               (test) => {
                 if (test) {
-                  run.passed(test);
+                  withDuration(test.id, (duration) =>
+                    run.passed(test, duration),
+                  );
                 }
               },
             ),
@@ -1353,7 +1365,13 @@ export class TestController {
             this.findTestItem(params.id, vscode.Uri.parse(params.uri)).then(
               (test) => {
                 if (test) {
-                  run.failed(test, new vscode.TestMessage(params.message));
+                  withDuration(test.id, (duration) =>
+                    run.failed(
+                      test,
+                      new vscode.TestMessage(params.message),
+                      duration,
+                    ),
+                  );
                 }
               },
             ),
@@ -1367,7 +1385,13 @@ export class TestController {
             this.findTestItem(params.id, vscode.Uri.parse(params.uri)).then(
               (test) => {
                 if (test) {
-                  run.errored(test, new vscode.TestMessage(params.message));
+                  withDuration(test.id, (duration) =>
+                    run.errored(
+                      test,
+                      new vscode.TestMessage(params.message),
+                      duration,
+                    ),
+                  );
                 }
               },
             ),