arga-cli 0.1.1__tar.gz → 0.1.3__tar.gz

{arga_cli-0.1.1 → arga_cli-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arga-cli
-Version: 0.1.1
+Version: 0.1.3
 Summary: Command-line interface for Arga authentication, MCP installation, and browser validation
 Author: Arga Labs
 Project-URL: Homepage, https://github.com/ArgaLabs/arga-cli
@@ -115,6 +115,7 @@ List and inspect recent validation runs:
 ```bash
 arga runs list --repo arga-labs/validation-server --limit 20
 arga runs status <run_id>
+arga runs logs <run_id>
 arga runs cancel <run_id>
 ```
 
@@ -186,6 +187,7 @@ arga scan report <run_id>
 ```bash
 arga runs list --repo arga-labs/validation-server --status running --limit 20
 arga runs status <run_id>
+arga runs logs <run_id>
 arga runs cancel <run_id>
 ```
 
@@ -193,6 +195,10 @@ arga runs cancel <run_id>
 - `--repo` narrows the list to a single repository.
 - `--status` accepts `completed`, `failed`, or `running`. The `running` filter includes non-terminal states such as `queued`.
 - `arga runs status <run_id>` prints a detailed summary for a specific run.
+- `arga runs logs <run_id>` prints worker logs plus recent runtime logs for a run you own.
+- When you omit `<run_id>`, `arga runs logs` falls back to `./.arga-session.json` when present, which makes wizard-created twin sessions easy to inspect from the same directory.
+- Add `--json` to `arga runs logs` for a machine-readable response.
+- Add `--errors-only` to keep only failed worker logs plus warning/error runtime entries.
 - `arga runs cancel <run_id>` cancels the run through the validation API.
 
 ### Git Wrappers

{arga_cli-0.1.1 → arga_cli-0.1.3}/README.md

@@ -95,6 +95,7 @@ List and inspect recent validation runs:
 ```bash
 arga runs list --repo arga-labs/validation-server --limit 20
 arga runs status <run_id>
+arga runs logs <run_id>
 arga runs cancel <run_id>
 ```
 
@@ -166,6 +167,7 @@ arga scan report <run_id>
 ```bash
 arga runs list --repo arga-labs/validation-server --status running --limit 20
 arga runs status <run_id>
+arga runs logs <run_id>
 arga runs cancel <run_id>
 ```
 
@@ -173,6 +175,10 @@ arga runs cancel <run_id>
 - `--repo` narrows the list to a single repository.
 - `--status` accepts `completed`, `failed`, or `running`. The `running` filter includes non-terminal states such as `queued`.
 - `arga runs status <run_id>` prints a detailed summary for a specific run.
+- `arga runs logs <run_id>` prints worker logs plus recent runtime logs for a run you own.
+- When you omit `<run_id>`, `arga runs logs` falls back to `./.arga-session.json` when present, which makes wizard-created twin sessions easy to inspect from the same directory.
+- Add `--json` to `arga runs logs` for a machine-readable response.
+- Add `--errors-only` to keep only failed worker logs plus warning/error runtime entries.
 - `arga runs cancel <run_id>` cancels the run through the validation API.
 
 ### Git Wrappers

{arga_cli-0.1.1 → arga_cli-0.1.3}/arga_cli/main.py

@@ -20,6 +20,8 @@ from arga_cli.mcp import install_mcp_configuration
 
 DEFAULT_API_URL = os.environ.get("ARGA_API_URL", "https://api.argalabs.com")
 CONFIG_PATH = Path.home() / ".config" / "arga" / "config.json"
+WIZARD_SESSION_FILE = ".arga-session.json"
+WIZARD_SESSION_PATH = Path(WIZARD_SESSION_FILE)
 POLL_INTERVAL_SECONDS = 2.0
 POLL_TIMEOUT_SECONDS = 600.0
 SKIP_TRAILER = "[skip arga]"
@@ -128,6 +130,13 @@ class ApiClient:
         )
         return self._parse_json(response, "Failed to load run details")
 
+    def get_run_logs(self, run_id: str) -> dict[str, Any]:
+        response = self._client.get(
+            f"{self._api_url}/runs/{run_id}/logs",
+            headers=self._auth_headers(),
+        )
+        return self._parse_json(response, "Failed to load run logs")
+
     def get_redteam_report(self, run_id: str) -> dict[str, Any]:
         response = self._client.get(
             f"{self._api_url}/redteam/{run_id}/report",
@@ -251,6 +260,31 @@ def delete_api_key() -> bool:
     return True
 
 
+def load_wizard_session(path: Path = WIZARD_SESSION_PATH) -> dict[str, Any] | None:
+    try:
+        data = json.loads(path.read_text())
+    except FileNotFoundError:
+        return None
+    except json.JSONDecodeError as exc:
+        raise CliError(f"Invalid wizard session file: {path}") from exc
+
+    if not isinstance(data, dict):
+        raise CliError(f"Invalid wizard session file: {path}")
+    return data
+
+
+def resolve_logs_run_id(run_id: str | None, *, session_path: Path = WIZARD_SESSION_PATH) -> str:
+    if run_id:
+        return run_id
+    session = load_wizard_session(session_path)
+    session_run_id = str((session or {}).get("run_id") or "").strip()
+    if session_run_id:
+        return session_run_id
+    raise CliError(
+        f"Run ID is required. Pass one explicitly or run this command from a directory containing {WIZARD_SESSION_FILE}."
+    )
+
+
 def build_verification_url(start_payload: dict[str, str]) -> str:
     verification_url = start_payload["verification_url"]
     device_code = start_payload["device_code"]
@@ -375,6 +409,10 @@ def run_test_url(args: argparse.Namespace) -> int:
     finally:
         client.close()
 
+    if getattr(args, "json", False):
+        print(json.dumps({"run_id": payload.get("run_id"), "status": payload.get("status")}))
+        return 0
+
     print("Starting validation...\n")
     print(f"URL: {args.url}")
     print(f"Prompt: {args.prompt}\n")
@@ -391,6 +429,10 @@ def run_validate_pr(args: argparse.Namespace) -> int:
     finally:
         client.close()
 
+    if args.json:
+        print(json.dumps({"run_id": payload.get("run_id"), "status": payload.get("status")}))
+        return 0
+
     print("Starting PR validation...\n")
     print(f"Repository: {args.repo}")
     print(f"PR: #{args.pr}\n")
@@ -796,6 +838,139 @@ def _print_runs_table(runs: list[dict[str, Any]]) -> None:
         print(format_row(row))
 
 
+def _print_worker_logs(worker_logs: list[dict[str, Any]]) -> None:
+    print("Worker Logs:")
+    if not worker_logs:
+        print("None")
+        return
+
+    for index, worker_log in enumerate(worker_logs):
+        job_id = str(worker_log.get("job_id") or "-")
+        metadata = [
+            str(worker_log.get("job_type") or "").strip(),
+            str(worker_log.get("target_role") or "").strip(),
+            str(worker_log.get("status") or "").strip(),
+        ]
+        metadata_label = " / ".join(value for value in metadata if value)
+        print(f"{job_id}: {metadata_label or 'worker log'}")
+
+        error = str(worker_log.get("error") or "").strip()
+        content = str(worker_log.get("content") or "").rstrip()
+        if error:
+            print(f"Error: {error}")
+        elif content:
+            print(content)
+        else:
+            print("No log content available.")
+
+        if index < len(worker_logs) - 1:
+            print()
+
+
+def _print_runtime_logs(runtime_logs: list[dict[str, Any]]) -> None:
+    print("Runtime Logs:")
+    if not runtime_logs:
+        print("None")
+        return
+
+    for index, runtime_log in enumerate(runtime_logs):
+        header_parts = [
+            _format_timestamp(runtime_log.get("timestamp")),
+            str(runtime_log.get("severity") or "").strip(),
+            str(runtime_log.get("service_name") or "").strip(),
+            str(runtime_log.get("event") or "").strip(),
+            str(runtime_log.get("code") or "").strip(),
+        ]
+        header = " | ".join(part for part in header_parts if part and part != "-")
+        print(header or "Log entry")
+
+        message = str(runtime_log.get("message") or "").strip()
+        if message:
+            print(message)
+
+        metadata: list[str] = []
+        request_id = str(runtime_log.get("request_id") or "").strip()
+        if request_id:
+            metadata.append(f"request_id={request_id}")
+        job_id = str(runtime_log.get("job_id") or "").strip()
+        if job_id:
+            metadata.append(f"job_id={job_id}")
+        surface_name = str(runtime_log.get("surface_name") or "").strip()
+        if surface_name:
+            metadata.append(f"surface={surface_name}")
+        if metadata:
+            print(" ".join(metadata))
+
+        if index < len(runtime_logs) - 1:
+            print()
+
+
+def _is_error_runtime_log(runtime_log: dict[str, Any]) -> bool:
+    severity = str(runtime_log.get("severity") or "").strip().upper()
+    return severity in {"WARNING", "ERROR", "CRITICAL", "ALERT", "EMERGENCY"}
+
+
+def _is_error_worker_log(worker_log: dict[str, Any]) -> bool:
+    status = str(worker_log.get("status") or "").strip().lower()
+    error = str(worker_log.get("error") or "").strip()
+    return status in {"failed", "error", "cancelled"} or bool(error)
+
+
+def _filter_run_logs_payload(payload: dict[str, Any], *, errors_only: bool) -> dict[str, Any]:
+    if not errors_only:
+        return payload
+
+    filtered_payload = dict(payload)
+    worker_logs = payload.get("worker_logs")
+    runtime_logs = payload.get("runtime_logs")
+    warnings = payload.get("warnings")
+
+    filtered_payload["worker_logs"] = (
+        [item for item in worker_logs if isinstance(item, dict) and _is_error_worker_log(item)]
+        if isinstance(worker_logs, list)
+        else []
+    )
+    filtered_payload["runtime_logs"] = (
+        [item for item in runtime_logs if isinstance(item, dict) and _is_error_runtime_log(item)]
+        if isinstance(runtime_logs, list)
+        else []
+    )
+    filtered_payload["warnings"] = warnings if isinstance(warnings, list) else []
+    return filtered_payload
+
+
+def _print_run_logs(payload: dict[str, Any], fallback_run_id: str) -> None:
+    run = payload.get("run")
+    run_data = run if isinstance(run, dict) else {}
+    worker_logs = payload.get("worker_logs")
+    runtime_logs = payload.get("runtime_logs")
+    warnings = payload.get("warnings")
+    timeline_events = run_data.get("event_log_json")
+
+    print(f"Run ID: {run_data.get('id', fallback_run_id)}")
+    print(f"Status: {run_data.get('status', 'unknown')}")
+    print(f"Type: {run_data.get('run_type', 'unknown')}")
+    print(f"Mode: {run_data.get('mode', 'unknown')}")
+    print(f"Repository: {run_data.get('repo_full_name') or '-'}")
+    print(f"PR/Branch: {_run_ref_label(run_data)}")
+    print(f"Commit: {run_data.get('commit_sha') or '-'}")
+    print(f"Created: {_format_timestamp(run_data.get('created_at'))}")
+    print(f"Environment URL: {run_data.get('environment_url') or '-'}")
+    if isinstance(timeline_events, list):
+        print(f"Timeline Events: {len(timeline_events)}")
+
+    print()
+    _print_worker_logs(worker_logs if isinstance(worker_logs, list) else [])
+    print()
+    _print_runtime_logs(runtime_logs if isinstance(runtime_logs, list) else [])
+
+    if isinstance(warnings, list) and warnings:
+        print()
+        print("Warnings:")
+        for warning in warnings:
+            print(f"- {warning}")
+
+
 def run_runs_list(args: argparse.Namespace) -> int:
     api_key = load_api_key()
     client = ApiClient(args.api_url, api_key=api_key)
@@ -809,6 +984,10 @@ def run_runs_list(args: argparse.Namespace) -> int:
     finally:
         client.close()
 
+    if args.json:
+        print(json.dumps(runs))
+        return 0
+
     if not runs:
         print("No matching validation runs found.")
         return 0
@@ -825,6 +1004,10 @@ def run_runs_status(args: argparse.Namespace) -> int:
     finally:
         client.close()
 
+    if args.json:
+        print(json.dumps(run))
+        return 0
+
     print(f"Run ID: {run.get('id', args.run_id)}")
     print(f"Status: {run.get('status', 'unknown')}")
     print(f"Type: {run.get('run_type', 'unknown')}")
@@ -838,6 +1021,25 @@ def run_runs_status(args: argparse.Namespace) -> int:
     return 0
 
 
+def run_runs_logs(args: argparse.Namespace) -> int:
+    run_id = resolve_logs_run_id(args.run_id)
+    api_key = load_api_key()
+    client = ApiClient(args.api_url, api_key=api_key)
+    try:
+        payload = client.get_run_logs(run_id)
+    finally:
+        client.close()
+
+    payload = _filter_run_logs_payload(payload, errors_only=args.errors_only)
+
+    if args.json:
+        print(json.dumps(payload, indent=2))
+        return 0
+
+    _print_run_logs(payload, run_id)
+    return 0
+
+
 def run_runs_cancel(args: argparse.Namespace) -> int:
     api_key = load_api_key()
     client = ApiClient(args.api_url, api_key=api_key)
@@ -1061,6 +1263,7 @@ def build_parser() -> argparse.ArgumentParser:
     test_url_parser.add_argument("--prompt", required=True, help="Natural language instructions for the agent")
     test_url_parser.add_argument("--email", help="Optional login email")
     test_url_parser.add_argument("--password", help="Optional login password")
+    test_url_parser.add_argument("--json", action="store_true", default=False, help="Output result as JSON")
     test_url_parser.set_defaults(func=run_test_url)
 
     validate_parser = subparsers.add_parser("validate", help="Start PR or URL validation runs")
@@ -1070,6 +1273,7 @@ def build_parser() -> argparse.ArgumentParser:
     validate_pr_parser.add_argument("--api-url", default=DEFAULT_API_URL, help="Arga API base URL")
     validate_pr_parser.add_argument("--repo", required=True, help="Repository in owner/repo format")
     validate_pr_parser.add_argument("--pr", required=True, type=int, help="Pull request number")
+    validate_pr_parser.add_argument("--json", action="store_true", default=False, help="Output result as JSON")
     validate_pr_parser.set_defaults(func=run_validate_pr)
 
     validate_url_parser = validate_subparsers.add_parser(
@@ -1081,6 +1285,7 @@ def build_parser() -> argparse.ArgumentParser:
     validate_url_parser.add_argument("--prompt", required=True, help="Natural language instructions for the agent")
     validate_url_parser.add_argument("--email", help="Optional login email")
     validate_url_parser.add_argument("--password", help="Optional login password")
+    validate_url_parser.add_argument("--json", action="store_true", default=False, help="Output result as JSON")
     validate_url_parser.set_defaults(func=run_test_url)
 
     mcp_parser = subparsers.add_parser("mcp", help="Manage MCP integrations")
@@ -1093,7 +1298,7 @@ def build_parser() -> argparse.ArgumentParser:
     mcp_install_parser.add_argument("--api-url", default=DEFAULT_API_URL, help="Arga API base URL")
     mcp_install_parser.set_defaults(func=run_mcp_install)
 
-    runs_parser = subparsers.add_parser("runs", help="List, inspect, or cancel validation runs")
+    runs_parser = subparsers.add_parser("runs", help="List, inspect, cancel, or read validation run logs")
     runs_subparsers = runs_parser.add_subparsers(dest="runs_command", required=True)
 
     runs_list_parser = runs_subparsers.add_parser("list", help="List recent validation runs")
@@ -1105,13 +1310,30 @@ def build_parser() -> argparse.ArgumentParser:
         help="Filter by validation status",
     )
     runs_list_parser.add_argument("--limit", type=int, default=20, help="Maximum number of runs to show")
+    runs_list_parser.add_argument("--json", action="store_true", default=False, help="Output result as JSON")
     runs_list_parser.set_defaults(func=run_runs_list)
 
     runs_status_parser = runs_subparsers.add_parser("status", help="Show detailed status for a validation run")
     runs_status_parser.add_argument("--api-url", default=DEFAULT_API_URL, help="Arga API base URL")
     runs_status_parser.add_argument("run_id", help="Validation run ID")
+    runs_status_parser.add_argument("--json", action="store_true", default=False, help="Output result as JSON")
     runs_status_parser.set_defaults(func=run_runs_status)
 
+    runs_logs_parser = runs_subparsers.add_parser("logs", help="Show worker and runtime logs for a validation run")
+    runs_logs_parser.add_argument("--api-url", default=DEFAULT_API_URL, help="Arga API base URL")
+    runs_logs_parser.add_argument(
+        "run_id",
+        nargs="?",
+        help=f"Validation run ID. Defaults to {WIZARD_SESSION_FILE} in the current directory when available.",
+    )
+    runs_logs_parser.add_argument("--json", action="store_true", help="Print the raw JSON response")
+    runs_logs_parser.add_argument(
+        "--errors-only",
+        action="store_true",
+        help="Show only failed worker logs and warning/error runtime logs",
+    )
+    runs_logs_parser.set_defaults(func=run_runs_logs)
+
     runs_cancel_parser = runs_subparsers.add_parser("cancel", help="Cancel a validation run")
     runs_cancel_parser.add_argument("--api-url", default=DEFAULT_API_URL, help="Arga API base URL")
     runs_cancel_parser.add_argument("run_id", help="Validation run ID")

{arga_cli-0.1.1 → arga_cli-0.1.3}/arga_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arga-cli
-Version: 0.1.1
+Version: 0.1.3
 Summary: Command-line interface for Arga authentication, MCP installation, and browser validation
 Author: Arga Labs
 Project-URL: Homepage, https://github.com/ArgaLabs/arga-cli
@@ -115,6 +115,7 @@ List and inspect recent validation runs:
 ```bash
 arga runs list --repo arga-labs/validation-server --limit 20
 arga runs status <run_id>
+arga runs logs <run_id>
 arga runs cancel <run_id>
 ```
 
@@ -186,6 +187,7 @@ arga scan report <run_id>
 ```bash
 arga runs list --repo arga-labs/validation-server --status running --limit 20
 arga runs status <run_id>
+arga runs logs <run_id>
 arga runs cancel <run_id>
 ```
 
@@ -193,6 +195,10 @@ arga runs cancel <run_id>
 - `--repo` narrows the list to a single repository.
 - `--status` accepts `completed`, `failed`, or `running`. The `running` filter includes non-terminal states such as `queued`.
 - `arga runs status <run_id>` prints a detailed summary for a specific run.
+- `arga runs logs <run_id>` prints worker logs plus recent runtime logs for a run you own.
+- When you omit `<run_id>`, `arga runs logs` falls back to `./.arga-session.json` when present, which makes wizard-created twin sessions easy to inspect from the same directory.
+- Add `--json` to `arga runs logs` for a machine-readable response.
+- Add `--errors-only` to keep only failed worker logs plus warning/error runtime entries.
 - `arga runs cancel <run_id>` cancels the run through the validation API.
 
 ### Git Wrappers

{arga_cli-0.1.1 → arga_cli-0.1.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "arga-cli"
-version = "0.1.1"
+version = "0.1.3"
 description = "Command-line interface for Arga authentication, MCP installation, and browser validation"
 readme = "README.md"
 requires-python = ">=3.12"

arga_cli-0.1.3/tests/test_cli_runs.py

@@ -0,0 +1,448 @@
+from __future__ import annotations
+
+import json
+
+from arga_cli import main
+
+
+def test_runs_list_prints_filtered_table(monkeypatch, capsys) -> None:
+    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
+    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
+
+    def fake_list(self, *, repo: str | None = None, limit: int = 20, offset: int = 0):
+        assert repo == "arga-labs/validation-server"
+        if offset == 0:
+            return {
+                "items": [
+                    {
+                        "run_id": "run_completed",
+                        "status": "completed",
+                        "repo": "arga-labs/validation-server",
+                        "pr_number": 182,
+                        "created_at": "2026-03-25T12:30:00Z",
+                    },
+                    {
+                        "run_id": "run_failed",
+                        "status": "failed",
+                        "repo": "arga-labs/validation-server",
+                        "branch": "main",
+                        "created_at": "2026-03-25T12:00:00Z",
+                    },
+                ],
+                "limit": limit,
+                "offset": offset,
+                "has_more": False,
+                "total": 2,
+            }
+        raise AssertionError("Unexpected extra page request")
+
+    monkeypatch.setattr(main.ApiClient, "list_pr_validation_runs", fake_list)
+
+    args = main.build_parser().parse_args(
+        ["runs", "list", "--repo", "arga-labs/validation-server", "--status", "failed", "--limit", "20"]
+    )
+    exit_code = args.func(args)
+    output = capsys.readouterr().out
+
+    assert exit_code == 0
+    assert "RUN_ID" in output
+    assert "run_failed" in output
+    assert "main" in output
+    assert "run_completed" not in output
+
+
+def test_runs_list_running_filter_includes_non_terminal_statuses(monkeypatch, capsys) -> None:
+    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
+    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
+
+    def fake_list(self, *, repo: str | None = None, limit: int = 20, offset: int = 0):
+        return {
+            "items": [
+                {
+                    "run_id": "run_queued",
+                    "status": "queued",
+                    "repo": "arga-labs/validation-server",
+                    "branch": "feature/arga",
+                    "created_at": "2026-03-25T13:00:00Z",
+                },
+                {
+                    "run_id": "run_completed",
+                    "status": "completed",
+                    "repo": "arga-labs/validation-server",
+                    "branch": "main",
+                    "created_at": "2026-03-25T12:00:00Z",
+                },
+            ],
+            "limit": limit,
+            "offset": offset,
+            "has_more": False,
+            "total": 2,
+        }
+
+    monkeypatch.setattr(main.ApiClient, "list_pr_validation_runs", fake_list)
+
+    args = main.build_parser().parse_args(["runs", "list", "--status", "running"])
+    exit_code = args.func(args)
+    output = capsys.readouterr().out
+
+    assert exit_code == 0
+    assert "run_queued" in output
+    assert "run_completed" not in output
+
+
+def test_runs_status_prints_detail_summary(monkeypatch, capsys) -> None:
+    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
+    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
+
+    def fake_get_run(self, run_id: str):
+        assert run_id == "run_123"
+        return {
+            "id": run_id,
+            "status": "running",
+            "run_type": "pr",
+            "mode": "auto_visual",
+            "repo_full_name": "arga-labs/validation-server",
+            "github_pr_number": 182,
+            "commit_sha": "abc123",
+            "created_at": "2026-03-25T12:30:00Z",
+            "environment_url": "https://preview.example.com",
+            "session_id": "session_1",
+        }
+
+    monkeypatch.setattr(main.ApiClient, "get_run", fake_get_run)
+
+    args = main.build_parser().parse_args(["runs", "status", "run_123"])
+    exit_code = args.func(args)
+    output = capsys.readouterr().out
+
+    assert exit_code == 0
+    assert "Run ID: run_123" in output
+    assert "Status: running" in output
+    assert "Repository: arga-labs/validation-server" in output
+    assert "PR/Branch: PR #182" in output
+
+
+def test_runs_list_json_flag(monkeypatch, capsys) -> None:
+    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
+    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
+
+    def fake_list(self, *, repo: str | None = None, limit: int = 20, offset: int = 0):
+        return {
+            "items": [
+                {
+                    "run_id": "run_1",
+                    "status": "completed",
+                    "repo": "arga-labs/validation-server",
+                    "pr_number": 182,
+                    "created_at": "2026-03-25T12:30:00Z",
+                },
+                {
+                    "run_id": "run_2",
+                    "status": "failed",
+                    "repo": "arga-labs/validation-server",
+                    "branch": "main",
+                    "created_at": "2026-03-25T12:00:00Z",
+                },
+            ],
+            "limit": limit,
+            "offset": offset,
+            "has_more": False,
+            "total": 2,
+        }
+
+    monkeypatch.setattr(main.ApiClient, "list_pr_validation_runs", fake_list)
+
+    args = main.build_parser().parse_args(["runs", "list", "--json"])
+    exit_code = args.func(args)
+    output = capsys.readouterr().out
+
+    assert exit_code == 0
+    parsed = json.loads(output)
+    assert isinstance(parsed, list)
+    assert len(parsed) == 2
+    assert parsed[0]["run_id"] == "run_1"
+    assert parsed[1]["run_id"] == "run_2"
+
+
+def test_runs_list_json_flag_empty(monkeypatch, capsys) -> None:
+    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
+    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
+
+    def fake_list(self, *, repo: str | None = None, limit: int = 20, offset: int = 0):
+        return {"items": [], "limit": limit, "offset": offset, "has_more": False, "total": 0}
+
+    monkeypatch.setattr(main.ApiClient, "list_pr_validation_runs", fake_list)
+
+    args = main.build_parser().parse_args(["runs", "list", "--json"])
+    exit_code = args.func(args)
+    output = capsys.readouterr().out
+
+    assert exit_code == 0
+    assert json.loads(output) == []
+
+
+def test_runs_status_json_flag(monkeypatch, capsys) -> None:
+    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
+    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
+
+    run_data = {
+        "id": "run_123",
+        "status": "running",
+        "run_type": "pr",
+        "mode": "auto_visual",
+        "repo_full_name": "arga-labs/validation-server",
+        "github_pr_number": 182,
+        "commit_sha": "abc123",
+        "created_at": "2026-03-25T12:30:00Z",
+        "environment_url": "https://preview.example.com",
+        "session_id": "session_1",
+    }
+
+    def fake_get_run(self, run_id: str):
+        return run_data
+
+    monkeypatch.setattr(main.ApiClient, "get_run", fake_get_run)
+
+    args = main.build_parser().parse_args(["runs", "status", "run_123", "--json"])
+    exit_code = args.func(args)
+    output = capsys.readouterr().out
+
+    assert exit_code == 0
+    parsed = json.loads(output)
+    assert parsed == run_data
+
+
+def test_runs_logs_prints_worker_and_runtime_logs(monkeypatch, capsys) -> None:
+    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
+    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
+
+    def fake_get_logs(self, run_id: str):
+        assert run_id == "run_123"
+        return {
+            "run": {
+                "id": run_id,
+                "status": "ready",
+                "run_type": "twin_quickstart",
+                "mode": "staging",
+                "repo_full_name": None,
+                "commit_sha": None,
+                "created_at": "2026-03-25T12:30:00Z",
+                "environment_url": "https://preview.example.com",
+                "event_log_json": [{"type": "environment_ready"}],
+            },
+            "worker_logs": [
+                {
+                    "job_id": "job_1",
+                    "job_type": "deploy",
+                    "target_role": "warm-vm",
+                    "status": "succeeded",
+                    "content": "deploy output",
+                    "truncated": False,
+                    "error": None,
+                }
+            ],
+            "runtime_logs": [
+                {
+                    "timestamp": "2026-03-25T12:31:00Z",
+                    "service_name": "arga-api",
+                    "severity": "INFO",
+                    "event": "environment_ready",
+                    "code": "ok",
+                    "request_id": "req_1",
+                    "job_id": "job_1",
+                    "surface_name": "app",
+                    "message": "Environment ready.",
+                }
+            ],
+            "warnings": ["Cloud Logging query was partially truncated."],
+        }
+
+    monkeypatch.setattr(main.ApiClient, "get_run_logs", fake_get_logs)
+
+    args = main.build_parser().parse_args(["runs", "logs", "run_123"])
+    exit_code = args.func(args)
+    output = capsys.readouterr().out
+
+    assert exit_code == 0
+    assert "Run ID: run_123" in output
+    assert "Worker Logs:" in output
+    assert "deploy output" in output
+    assert "Runtime Logs:" in output
+    assert "Environment ready." in output
+    assert "Warnings:" in output
+
+
+def test_runs_logs_prints_json(monkeypatch, capsys) -> None:
+    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
+    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
+
+    payload = {
+        "run": {"id": "run_123", "status": "ready"},
+        "worker_logs": [],
+        "runtime_logs": [],
+        "warnings": [],
+    }
+
+    monkeypatch.setattr(main.ApiClient, "get_run_logs", lambda self, run_id: payload)
+
+    args = main.build_parser().parse_args(["runs", "logs", "run_123", "--json"])
+    exit_code = args.func(args)
+    output = capsys.readouterr().out
+
+    assert exit_code == 0
+    assert json.loads(output) == payload
+
+
+def test_runs_logs_errors_only_filters_plain_output(monkeypatch, capsys) -> None:
+    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
+    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
+
+    def fake_get_logs(self, run_id: str):
+        assert run_id == "run_123"
+        return {
+            "run": {
+                "id": run_id,
+                "status": "ready",
+                "run_type": "twin_quickstart",
+                "mode": "staging",
+                "repo_full_name": None,
+                "commit_sha": None,
+                "created_at": "2026-03-25T12:30:00Z",
+                "environment_url": "https://preview.example.com",
+                "event_log_json": [],
+            },
+            "worker_logs": [
+                {
+                    "job_id": "job_ok",
+                    "job_type": "build",
+                    "target_role": "builder",
+                    "status": "succeeded",
+                    "content": "all good",
+                    "truncated": False,
+                    "error": None,
+                },
+                {
+                    "job_id": "job_bad",
+                    "job_type": "deploy",
+                    "target_role": "warm-vm",
+                    "status": "failed",
+                    "content": "deploy failed output",
+                    "truncated": False,
+                    "error": None,
+                },
+            ],
+            "runtime_logs": [
+                {
+                    "timestamp": "2026-03-25T12:31:00Z",
+                    "service_name": "arga-api",
+                    "severity": "INFO",
+                    "event": "environment_ready",
+                    "code": None,
+                    "request_id": "req_ok",
+                    "job_id": None,
+                    "surface_name": None,
+                    "message": "Environment ready.",
+                },
+                {
+                    "timestamp": "2026-03-25T12:32:00Z",
+                    "service_name": "preview-proxy",
+                    "severity": "WARNING",
+                    "event": "preview.request.finish",
+                    "code": None,
+                    "request_id": "req_warn",
+                    "job_id": None,
+                    "surface_name": "slack",
+                    "message": "Preview request completed.",
+                },
+            ],
+            "warnings": ["Cloud Logging query was partially truncated."],
+        }
+
+    monkeypatch.setattr(main.ApiClient, "get_run_logs", fake_get_logs)
+
+    args = main.build_parser().parse_args(["runs", "logs", "run_123", "--errors-only"])
+    exit_code = args.func(args)
+    output = capsys.readouterr().out
+
+    assert exit_code == 0
+    assert "job_bad" in output
+    assert "deploy failed output" in output
+    assert "job_ok" not in output
+    assert "all good" not in output
+    assert "Preview request completed." in output
+    assert "Environment ready." not in output
+    assert "Warnings:" in output
+
+
+def test_runs_logs_errors_only_filters_json(monkeypatch, capsys) -> None:
+    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
+    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
+
+    payload = {
+        "run": {"id": "run_123", "status": "ready"},
+        "worker_logs": [
+            {"job_id": "job_ok", "status": "succeeded", "error": None},
+            {"job_id": "job_bad", "status": "failed", "error": None},
+        ],
+        "runtime_logs": [
+            {"severity": "INFO", "message": "Environment ready."},
+            {"severity": "WARNING", "message": "Preview request completed."},
+        ],
+        "warnings": ["Cloud Logging query was partially truncated."],
+    }
+
+    monkeypatch.setattr(main.ApiClient, "get_run_logs", lambda self, run_id: payload)
+
+    args = main.build_parser().parse_args(["runs", "logs", "run_123", "--json", "--errors-only"])
+    exit_code = args.func(args)
+    output = capsys.readouterr().out
+
+    assert exit_code == 0
+    parsed = json.loads(output)
+    assert parsed["worker_logs"] == [{"job_id": "job_bad", "status": "failed", "error": None}]
+    assert parsed["runtime_logs"] == [{"severity": "WARNING", "message": "Preview request completed."}]
+    assert parsed["warnings"] == ["Cloud Logging query was partially truncated."]
+
+
+def test_runs_logs_uses_wizard_session_file_when_run_id_missing(monkeypatch, capsys, tmp_path) -> None:
+    monkeypatch.chdir(tmp_path)
+    (tmp_path / main.WIZARD_SESSION_FILE).write_text(json.dumps({"run_id": "run_from_session"}) + "\n")
+    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
+    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
+
+    def fake_get_logs(self, run_id: str):
+        assert run_id == "run_from_session"
+        return {
+            "run": {"id": run_id, "status": "ready"},
+            "worker_logs": [],
+            "runtime_logs": [],
+            "warnings": [],
+        }
+
+    monkeypatch.setattr(main.ApiClient, "get_run_logs", fake_get_logs)
+
+    args = main.build_parser().parse_args(["runs", "logs"])
+    exit_code = args.func(args)
+    output = capsys.readouterr().out
+
+    assert exit_code == 0
+    assert "Run ID: run_from_session" in output
+
+
+def test_runs_cancel_prints_cancelled_status(monkeypatch, capsys) -> None:
+    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
+    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
+
+    def fake_cancel(self, run_id: str):
+        assert run_id == "run_123"
+        return {"status": "cancelled"}
+
+    monkeypatch.setattr(main.ApiClient, "cancel_validation_run", fake_cancel)
+
+    args = main.build_parser().parse_args(["runs", "cancel", "run_123"])
+    exit_code = args.func(args)
+    output = capsys.readouterr().out
+
+    assert exit_code == 0
+    assert "Run ID: run_123" in output
+    assert "Status: cancelled" in output

{arga_cli-0.1.1 → arga_cli-0.1.3}/tests/test_cli_test_url.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+import json
+
 from arga_cli import main
 
 
@@ -26,3 +28,23 @@ def test_test_url_command_prints_run_id(monkeypatch, capsys) -> None:
     assert "Starting validation..." in output
     assert "Run ID: run_3421" in output
     assert "Status: queued" in output
+
+
+def test_test_url_json_flag(monkeypatch, capsys) -> None:
+    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
+
+    def fake_start(self, *, url: str, prompt: str, email: str | None = None, password: str | None = None):
+        return {"run_id": "run_3421", "status": "queued", "session_id": "session_1"}
+
+    monkeypatch.setattr(main.ApiClient, "start_url_validation", fake_start)
+    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
+
+    args = main.build_parser().parse_args(
+        ["test", "url", "--url", "https://demo-app.com", "--prompt", "test login flow", "--json"]
+    )
+    exit_code = args.func(args)
+    output = capsys.readouterr().out
+
+    assert exit_code == 0
+    parsed = json.loads(output)
+    assert parsed == {"run_id": "run_3421", "status": "queued"}

{arga_cli-0.1.1 → arga_cli-0.1.3}/tests/test_cli_validate_pr.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+import json
+
 from arga_cli import main
 
 
@@ -51,3 +53,23 @@ def test_validate_url_alias_reuses_url_validation(monkeypatch, capsys) -> None:
     assert exit_code == 0
     assert "Starting validation..." in output
     assert "Run ID: run_123" in output
+
+
+def test_validate_pr_json_flag(monkeypatch, capsys) -> None:
+    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
+
+    def fake_start(self, *, repo: str, pr_number: int):
+        return {"run_id": "run_83921", "status": "queued"}
+
+    monkeypatch.setattr(main.ApiClient, "start_pr_validation", fake_start)
+    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
+
+    args = main.build_parser().parse_args(
+        ["validate", "pr", "--repo", "arga-labs/validation-server", "--pr", "182", "--json"]
+    )
+    exit_code = args.func(args)
+    output = capsys.readouterr().out
+
+    assert exit_code == 0
+    parsed = json.loads(output)
+    assert parsed == {"run_id": "run_83921", "status": "queued"}

arga_cli-0.1.1/tests/test_cli_runs.py

@@ -1,139 +0,0 @@
-from __future__ import annotations
-
-from arga_cli import main
-
-
-def test_runs_list_prints_filtered_table(monkeypatch, capsys) -> None:
-    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
-    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
-
-    def fake_list(self, *, repo: str | None = None, limit: int = 20, offset: int = 0):
-        assert repo == "arga-labs/validation-server"
-        if offset == 0:
-            return {
-                "items": [
-                    {
-                        "run_id": "run_completed",
-                        "status": "completed",
-                        "repo": "arga-labs/validation-server",
-                        "pr_number": 182,
-                        "created_at": "2026-03-25T12:30:00Z",
-                    },
-                    {
-                        "run_id": "run_failed",
-                        "status": "failed",
-                        "repo": "arga-labs/validation-server",
-                        "branch": "main",
-                        "created_at": "2026-03-25T12:00:00Z",
-                    },
-                ],
-                "limit": limit,
-                "offset": offset,
-                "has_more": False,
-                "total": 2,
-            }
-        raise AssertionError("Unexpected extra page request")
-
-    monkeypatch.setattr(main.ApiClient, "list_pr_validation_runs", fake_list)
-
-    args = main.build_parser().parse_args(
-        ["runs", "list", "--repo", "arga-labs/validation-server", "--status", "failed", "--limit", "20"]
-    )
-    exit_code = args.func(args)
-    output = capsys.readouterr().out
-
-    assert exit_code == 0
-    assert "RUN_ID" in output
-    assert "run_failed" in output
-    assert "main" in output
-    assert "run_completed" not in output
-
-
-def test_runs_list_running_filter_includes_non_terminal_statuses(monkeypatch, capsys) -> None:
-    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
-    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
-
-    def fake_list(self, *, repo: str | None = None, limit: int = 20, offset: int = 0):
-        return {
-            "items": [
-                {
-                    "run_id": "run_queued",
-                    "status": "queued",
-                    "repo": "arga-labs/validation-server",
-                    "branch": "feature/arga",
-                    "created_at": "2026-03-25T13:00:00Z",
-                },
-                {
-                    "run_id": "run_completed",
-                    "status": "completed",
-                    "repo": "arga-labs/validation-server",
-                    "branch": "main",
-                    "created_at": "2026-03-25T12:00:00Z",
-                },
-            ],
-            "limit": limit,
-            "offset": offset,
-            "has_more": False,
-            "total": 2,
-        }
-
-    monkeypatch.setattr(main.ApiClient, "list_pr_validation_runs", fake_list)
-
-    args = main.build_parser().parse_args(["runs", "list", "--status", "running"])
-    exit_code = args.func(args)
-    output = capsys.readouterr().out
-
-    assert exit_code == 0
-    assert "run_queued" in output
-    assert "run_completed" not in output
-
-
-def test_runs_status_prints_detail_summary(monkeypatch, capsys) -> None:
-    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
-    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
-
-    def fake_get_run(self, run_id: str):
-        assert run_id == "run_123"
-        return {
-            "id": run_id,
-            "status": "running",
-            "run_type": "pr",
-            "mode": "auto_visual",
-            "repo_full_name": "arga-labs/validation-server",
-            "github_pr_number": 182,
-            "commit_sha": "abc123",
-            "created_at": "2026-03-25T12:30:00Z",
-            "environment_url": "https://preview.example.com",
-            "session_id": "session_1",
-        }
-
-    monkeypatch.setattr(main.ApiClient, "get_run", fake_get_run)
-
-    args = main.build_parser().parse_args(["runs", "status", "run_123"])
-    exit_code = args.func(args)
-    output = capsys.readouterr().out
-
-    assert exit_code == 0
-    assert "Run ID: run_123" in output
-    assert "Status: running" in output
-    assert "Repository: arga-labs/validation-server" in output
-    assert "PR/Branch: PR #182" in output
-
-
-def test_runs_cancel_prints_cancelled_status(monkeypatch, capsys) -> None:
-    monkeypatch.setattr(main, "load_api_key", lambda: "arga_api_key")
-    monkeypatch.setattr(main.ApiClient, "close", lambda self: None)
-
-    def fake_cancel(self, run_id: str):
-        assert run_id == "run_123"
-        return {"status": "cancelled"}
-
-    monkeypatch.setattr(main.ApiClient, "cancel_validation_run", fake_cancel)
-
-    args = main.build_parser().parse_args(["runs", "cancel", "run_123"])
-    exit_code = args.func(args)
-    output = capsys.readouterr().out
-
-    assert exit_code == 0
-    assert "Run ID: run_123" in output
-    assert "Status: cancelled" in output