microsoft · jdneo · Nov 29, 2019 · Nov 29, 2019 · Nov 29, 2019
diff --git a/resources/media/dark/error.svg b/resources/media/dark/error.svg
diff --git a/resources/media/dark/pass.svg b/resources/media/dark/pass.svg
diff --git a/resources/media/light/error.svg b/resources/media/light/error.svg
diff --git a/resources/media/light/pass.svg b/resources/media/light/pass.svg
diff --git a/src/explorer/testExplorer.ts b/src/explorer/testExplorer.ts
@@ -5,7 +5,9 @@ import * as path from 'path';
 import { Command, Disposable, Event, EventEmitter, ExtensionContext, Range, TreeDataProvider, TreeItem, TreeItemCollapsibleState, Uri, workspace, WorkspaceFolder } from 'vscode';
 import { JavaTestRunnerCommands } from '../constants/commands';
 import { ITestItem, TestKind, TestLevel } from '../protocols';
+import { ITestResult, TestStatus } from '../runners/models';
 import { testItemModel } from '../testItemModel';
+import { testResultManager } from '../testResultManager';
 
 export class TestExplorer implements TreeDataProvider<ITestItem>, Disposable {
     public readonly testExplorerViewId: string = 'testExplorer';
@@ -99,6 +101,21 @@ export class TestExplorer implements TreeDataProvider<ITestItem>, Disposable {
     private resolveIconPath(element: ITestItem): undefined | { dark: string | Uri, light: string | Uri } {
         switch (element.level) {
             case TestLevel.Method:
+                const result: ITestResult | undefined = testResultManager.getResultById(element.id);
+                if (result) {
+                    if (result.status === TestStatus.Pass) {
+                        return {
+                            dark: this._context.asAbsolutePath(path.join('resources', 'media', 'dark', 'pass.svg')),
+                            light: this._context.asAbsolutePath(path.join('resources', 'media', 'light', 'pass.svg')),
+                        };
+                    } else if (result.status === TestStatus.Fail) {
+                        return {
+                            dark: this._context.asAbsolutePath(path.join('resources', 'media', 'dark', 'error.svg')),
+                            light: this._context.asAbsolutePath(path.join('resources', 'media', 'light', 'error.svg')),
+                        };
+                    }
+                }
+
                 return {
                     dark: this._context.asAbsolutePath(path.join('resources', 'media', 'dark', 'method.svg')),
                     light: this._context.asAbsolutePath(path.join('resources', 'media', 'light', 'method.svg')),

diff --git a/src/testResultManager.ts b/src/testResultManager.ts
@@ -1,15 +1,19 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT license.
 
-import { Disposable } from 'vscode';
+import { Disposable, Uri } from 'vscode';
+import { testExplorer } from './explorer/testExplorer';
+import { ITestItem } from './protocols';
 import { ITestResult } from './runners/models';
+import { testItemModel } from './testItemModel';
 
 class TestResultManager implements Disposable {
     private testResultMap: Map<string, ITestResult> = new Map<string, ITestResult>();
 
     public async storeResult(...results: ITestResult[]): Promise<void> {
         for (const result of results) {
             this.testResultMap.set(result.id, result);
+            this.notifyExplorer(result);
         }
     }
 
@@ -24,6 +28,16 @@ class TestResultManager implements Disposable {
     public dispose(): void {
         this.testResultMap.clear();
     }
+
+    private notifyExplorer(result: ITestResult): void {
+        const item: ITestItem | undefined = testItemModel.getItemById(result.id);
+        if (item) {
+            const node: ITestItem | undefined = testExplorer.getNodeByFsPath(Uri.parse(item.location.uri).fsPath);
+            if (node) {
+                testExplorer.refresh(node);
+            }
+        }
+    }
 }
 
 export const testResultManager: TestResultManager = new TestResultManager();