From dcba79cda0053864d1a269c165043b919585b328 Mon Sep 17 00:00:00 2001
From: Willem van den Ende <willem@qwan.eu>
Date: Sun, 28 Jun 2026 12:21:05 +0200
Subject: [PATCH] Refactor both if branches to be the same

Parsing no longer depends on ui widget
---
 pi-test-status/extensions/index.ts | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/pi-test-status/extensions/index.ts b/pi-test-status/extensions/index.ts
index 2fcc04e..7ed8a77 100644
--- a/pi-test-status/extensions/index.ts
+++ b/pi-test-status/extensions/index.ts
@@ -1,4 +1,4 @@
-import {type ExtensionAPI, isBashToolResult} from "@earendil-works/pi-coding-agent";
+import {type ExtensionAPI, isBashToolResult, ThemeColor} from "@earendil-works/pi-coding-agent";
 import {isTestCommand, parseTestOutput} from "./parsing/typescript-command-output";
 
 // noinspection JSUnusedGlobalSymbols
@@ -15,13 +15,14 @@ export default function testStatusExtension(pi: ExtensionAPI) {
 
         const testResult = parseTestOutput(output);
 
+        const feedbackTheme : ThemeColor = (testResult.status === "Success") ? "success" : "error";
         if (testResult.status === "Success") {
             ctx.ui.setWidget("test-status", [
-                `${theme.fg("success", "●")} ${theme.fg("text", testResult.summary)}`,
+                `${theme.fg(feedbackTheme,  "●")} ${theme.fg("text", testResult.summary)}`,
             ]);
         } else if (testResult.status === "Fail" || event.isError) {
             ctx.ui.setWidget("test-status", [
-                `${theme.fg("error", "●")} ${theme.fg("text", testResult.summary)}`,
+                `${theme.fg(feedbackTheme, "●")} ${theme.fg("text", testResult.summary)}`,
             ]);
         }
     });