tests: add coverage for --color option

Add tests for:
- CLI override precedence in WestCommand.color_ui
- behavior with and without configuration
- parsing of the new --color argument in parse_early_args

Signed-off-by: Tharaka Jayasena <9dmpires2k17.tuj@gmail.com>

Author: TharakaUJ

tests/test_commands.py

@@ -26,6 +26,15 @@ def do_run(self):
 
 cmd = WestCommandImpl(name="x", help="y", description="z")
 
+
+class DummyConfig:
+    def __init__(self, color_ui_value):
+        self.color_ui_value = color_ui_value
+
+    def getboolean(self, _option, default=True):
+        return self.color_ui_value if self.color_ui_value is not None else default
+
+
 TEST_STR = "This is some test string"
 COL_RED = "\x1b[91m"
 COL_YELLOW = "\x1b[93m"
@@ -120,3 +129,28 @@ def test_die(capsys, test_case):
     stderr = captured.err
     assert stderr == exp_err
     assert stdout == exp_out
+
+
+def test_color_ui_uses_color():
+    command = WestCommandImpl(name="x", help="y", description="z")
+    command.color = 'always'
+    assert command.color_ui is True
+
+    command.color = 'never'
+    assert command.color_ui is False
+
+
+def test_color_ui_default_is_enabled_when_mode_is_auto():
+    command = WestCommandImpl(name="x", help="y", description="z")
+    command.color = 'auto'
+    command.config = DummyConfig(False)
+
+    assert command.color_ui is True
+
+
+def test_color_ui_default_without_config():
+    command = WestCommandImpl(name="x", help="y", description="z")
+    command.color = None
+    command.config = None
+
+    assert command.color_ui is True

tests/test_main.py

@@ -5,6 +5,7 @@
 import pytest
 from conftest import cmd, cmd_subprocess
 
+import west.app.main as west_main
 import west.version
 
 
@@ -48,3 +49,18 @@ def test_module_run(tmp_path, monkeypatch):
     # check that that the sys.path was correctly inserted
     expected_path = Path(__file__).parents[1] / 'src'
     assert actual_path == [f'{expected_path}', 'initial-path']
+
+
+@pytest.mark.parametrize(
+    "argv, expected_color, expected_command, expected_unexpected",
+    [
+        (['--color=always', 'help'], 'always', 'help', []),
+        (['--color', 'never', 'status'], 'never', 'status', []),
+        (['--color', 'invalid', 'status'], None, 'status', ['--color=invalid']),
+    ],
+)
+def test_parse_early_args_color(argv, expected_color, expected_command, expected_unexpected):
+    ea = west_main.parse_early_args(argv)
+    assert ea.color == expected_color
+    assert ea.command_name == expected_command
+    assert ea.unexpected_arguments == expected_unexpected