From 137c3ec5e52c25d062f443d4e71186b642c79ed7 Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Sun, 11 Dec 2022 21:44:20 -0800
Subject: [PATCH] Add label for test mode in hardware diagnostics

---
 scripts/wk/hw/diags.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/scripts/wk/hw/diags.py b/scripts/wk/hw/diags.py
index 7a673c45..c5f9484a 100644
--- a/scripts/wk/hw/diags.py
+++ b/scripts/wk/hw/diags.py
@@ -79,7 +79,7 @@ PLATFORM = std.PLATFORM
 # Classes
 class State():
   """Object for tracking hardware diagnostic data."""
-  def __init__(self):
+  def __init__(self, test_mode=False):
     self.disks = []
     self.layout = cfg.hw.TMUX_LAYOUT.copy()
     self.log_dir = None
@@ -87,6 +87,8 @@ class State():
     self.system = None
     self.test_groups = []
     self.top_text = std.color_string('Hardware Diagnostics', 'GREEN')
+    if test_mode:
+      self.top_text += std.color_string(' (Test Mode)', 'YELLOW')
 
     # Init tmux and start a background process to maintain layout
     self.init_tmux()
@@ -760,7 +762,7 @@ def main() -> None:
   # Init
   atexit.register(tmux.kill_all_panes)
   menu = build_menu(cli_mode=args['--cli'], quick_mode=args['--quick'])
-  state = State()
+  state = State(test_mode=args['--test-mode'])
 
   # Quick Mode
   if args['--quick']: