Support retiring test states

microsoft · Jul 25, 2021 · 319e4d7 · 319e4d7
1 parent 4972baa
commit 319e4d7
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/testConverter.ts b/src/testConverter.ts
@@ -66,6 +66,23 @@ export class TestConverter implements vscode.Disposable {
       })
     );
 
+    if (adapter.retire) {
+      this.disposables.push(
+        adapter.retire(evt => {
+          if (!this.controller) {
+            return;
+          }
+          if (evt.tests) {
+            for (const id of evt.tests) {
+              this.itemsById.get(id)?.invalidateResults();
+            }
+          } else {
+            this.controller.items.forEach(item => item.invalidateResults());
+          }
+        })
+      );
+    }
+
     setTimeout(() => this.adapter.load(), 1);
   }