From 9e4974022eefa390a315f7ad41fee33f4a1627b6 Mon Sep 17 00:00:00 2001
From: rootkiller6788 <c8688rickowens@outlook.com>
Date: Fri, 21 Aug 2026 23:32:08 +0800
Subject: [PATCH] test(cli): cover adk conformance test command options

Add focused unit tests for the argument handling of `adk conformance test`
(issue #6730): multiple PATHS are all forwarded, --mode live is propagated,
--streaming-mode bidi maps to the StreamingMode enum, and invalid --mode /
--streaming-mode values are rejected by Click before any dispatch.

The runner is mocked so the tests stay fast and deterministic, matching the
existing conformance CLI tests in this module.
---
 .../cli/utils/test_cli_tools_click.py         | 105 ++++++++++++++++++
 1 file changed, 105 insertions(+)

diff --git a/tests/unittests/cli/utils/test_cli_tools_click.py b/tests/unittests/cli/utils/test_cli_tools_click.py
index dddda4471f..36090b9cf4 100644
--- a/tests/unittests/cli/utils/test_cli_tools_click.py
+++ b/tests/unittests/cli/utils/test_cli_tools_click.py
@@ -2857,6 +2857,111 @@ def test_cli_conformance_test_forwards_mode_and_report_options(
   }]
 
 
+def test_cli_conformance_test_accepts_multiple_directories(
+    tmp_path: Path, fake_conformance_test
+) -> None:
+  """Every PATHS argument is forwarded to the conformance runner."""
+  first_dir = tmp_path / "cases_one"
+  second_dir = tmp_path / "cases_two"
+  first_dir.mkdir()
+  second_dir.mkdir()
+
+  result = CliRunner().invoke(
+      cli_tools_click.main,
+      ["conformance", "test", str(first_dir), str(second_dir)],
+  )
+
+  assert result.exit_code == 0, (result.output, repr(result.exception))
+  assert fake_conformance_test == [{
+      "test_paths": [
+          Path(os.path.realpath(first_dir)),
+          Path(os.path.realpath(second_dir)),
+      ],
+      "mode": "replay",
+      "generate_report": False,
+      "report_dir": None,
+      "streaming_mode": None,
+  }]
+
+
+def test_cli_conformance_test_forwards_live_mode(
+    tmp_path: Path, fake_conformance_test
+) -> None:
+  """--mode live is propagated to the conformance runner."""
+  case_dir = tmp_path / "cases"
+  case_dir.mkdir()
+
+  result = CliRunner().invoke(
+      cli_tools_click.main,
+      ["conformance", "test", str(case_dir), "--mode", "live"],
+  )
+
+  assert result.exit_code == 0, (result.output, repr(result.exception))
+  assert fake_conformance_test == [{
+      "test_paths": [Path(os.path.realpath(case_dir))],
+      "mode": "live",
+      "generate_report": False,
+      "report_dir": None,
+      "streaming_mode": None,
+  }]
+
+
+def test_cli_conformance_test_forwards_bidi_streaming_mode(
+    tmp_path: Path, fake_conformance_test
+) -> None:
+  """--streaming-mode bidi is parsed to the StreamingMode enum."""
+  case_dir = tmp_path / "cases"
+  case_dir.mkdir()
+
+  result = CliRunner().invoke(
+      cli_tools_click.main,
+      ["conformance", "test", str(case_dir), "--streaming-mode", "bidi"],
+  )
+
+  assert result.exit_code == 0, (result.output, repr(result.exception))
+  assert fake_conformance_test == [{
+      "test_paths": [Path(os.path.realpath(case_dir))],
+      "mode": "replay",
+      "generate_report": False,
+      "report_dir": None,
+      "streaming_mode": StreamingMode.BIDI,
+  }]
+
+
+def test_cli_conformance_test_rejects_invalid_mode(
+    tmp_path: Path,
+) -> None:
+  """An unknown --mode value is rejected by Click before any dispatch."""
+  case_dir = tmp_path / "cases"
+  case_dir.mkdir()
+
+  result = CliRunner().invoke(
+      cli_tools_click.main,
+      ["conformance", "test", str(case_dir), "--mode", "fast"],
+  )
+
+  assert result.exit_code == 2
+  assert "Invalid value for '--mode'" in result.output
+  assert "'fast' is not one of 'replay', 'live'" in result.output
+
+
+def test_cli_conformance_test_rejects_invalid_streaming_mode(
+    tmp_path: Path,
+) -> None:
+  """An unknown --streaming-mode value is rejected by Click before dispatch."""
+  case_dir = tmp_path / "cases"
+  case_dir.mkdir()
+
+  result = CliRunner().invoke(
+      cli_tools_click.main,
+      ["conformance", "test", str(case_dir), "--streaming-mode", "eventstream"],
+  )
+
+  assert result.exit_code == 2
+  assert "Invalid value for '--streaming-mode'" in result.output
+  assert "'eventstream' is not one of 'none', 'sse', 'bidi'" in result.output
+
+
 # adk eval_set create
 def test_cli_create_eval_set_surfaces_duplicate_id_as_click_exception(
     tmp_path: Path,