Function cli_test

src/google/adk/cli/cli_tools_click.py:824–888

Runs pytest on agent test JSON files under the specified folder. FOLDER: The path to the folder containing agents and tests. Defaults to the current directory if not specified. Example: adk test path/to/agents

(ctx, folder: str, rebuild: bool)

)
@click.pass_context
def cli_test(ctx, folder: str, rebuild: bool):
  """Runs pytest on agent test JSON files under the specified folder.

  FOLDER: The path to the folder containing agents and tests.
  Defaults to the current directory if not specified.

  Example:
      adk test path/to/agents
  """
  import sys

  if rebuild:
    from .agent_test_runner import rebuild_tests

    click.echo(f"Rebuilding tests in {folder}...")
    rebuild_tests(folder)
    sys.exit(0)

  # Parse arguments to separate pytest args (after --) from regular args
  pytest_args = []
  if "--" in ctx.args:
    separator_index = ctx.args.index("--")
    pytest_args = ctx.args[separator_index + 1 :]
    regular_args = ctx.args[:separator_index]

    if regular_args:
      click.secho(
          "Error: Unexpected arguments after folder and before '--':"
          f" {' '.join(regular_args)}. \nOnly arguments after '--' are passed"
          " to pytest.",
          fg="red",
          err=True,
      )
      ctx.exit(2)
  else:
    # If no '--', all remaining arguments are passed to pytest
    pytest_args = ctx.args

  import subprocess

  os.environ["ADK_TEST_FOLDER"] = folder

  current_dir = Path(__file__).parent
  test_runner_path = current_dir / "agent_test_runner.py"

  if not test_runner_path.exists():
    click.secho(
        f"Error: Test runner not found at {test_runner_path}",
        fg="red",
        err=True,
    )
    sys.exit(1)

  click.echo(f"Running tests in {folder} using runner {test_runner_path}...")

  result = subprocess.run([
      sys.executable,
      "-m",