mirror of
https://github.com/HKUDS/nanobot.git
synced 2026-08-31 08:13:11 +03:00
110 lines
3.1 KiB
Python
110 lines
3.1 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from typer.testing import CliRunner
|
|
|
|
from nanobot.cli.commands import app
|
|
from nanobot.cli.process_identity import named_executable, set_cli_process_identity
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("args", "expected"),
|
|
[
|
|
(["agent"], "nanobot-agent"),
|
|
(["gateway", "--background"], "nanobot-gateway"),
|
|
(["webui"], "nanobot-webui"),
|
|
(["status"], "nanobot"),
|
|
([], "nanobot"),
|
|
],
|
|
)
|
|
def test_cli_process_identity_uses_product_and_role(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
args: list[str],
|
|
expected: str,
|
|
) -> None:
|
|
titles: list[str] = []
|
|
monkeypatch.setattr("nanobot.cli.process_identity.os.name", "posix")
|
|
monkeypatch.setattr("nanobot.cli.process_identity._set_process_title", titles.append)
|
|
|
|
set_cli_process_identity(args)
|
|
|
|
assert titles == [expected]
|
|
|
|
|
|
def test_cli_process_identity_keeps_windows_launcher_name(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
titles: list[str] = []
|
|
monkeypatch.setattr("nanobot.cli.process_identity.os.name", "nt")
|
|
monkeypatch.setattr("nanobot.cli.process_identity._set_process_title", titles.append)
|
|
|
|
set_cli_process_identity(["agent"])
|
|
|
|
assert titles == []
|
|
|
|
|
|
def test_legacy_console_entrypoint_still_sets_subcommand_identity(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
commands: list[list[str]] = []
|
|
monkeypatch.setattr("nanobot.cli.commands.set_cli_process_identity", commands.append)
|
|
|
|
result = CliRunner().invoke(app, ["webui", "--help"])
|
|
|
|
assert result.exit_code == 0
|
|
assert commands == [["webui"]]
|
|
|
|
|
|
def test_legacy_console_entrypoint_routes_bare_command_to_agent(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
identities: list[list[str]] = []
|
|
launches: list[tuple[list[str], str]] = []
|
|
monkeypatch.setattr("nanobot.cli.commands.set_cli_process_identity", identities.append)
|
|
monkeypatch.setattr(
|
|
"nanobot.cli.entry._run_agent",
|
|
lambda args, *, prog_name: launches.append((args, prog_name)),
|
|
)
|
|
|
|
result = CliRunner().invoke(app, [])
|
|
|
|
assert result.exit_code == 0
|
|
assert identities == [["agent"]]
|
|
assert launches == [([], "nanobot")]
|
|
|
|
|
|
def test_named_executable_creates_stable_role_symlink(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
if os.name == "nt":
|
|
pytest.skip("POSIX symlink naming is not used on Windows")
|
|
executable = tmp_path / "bun"
|
|
executable.write_text("runtime", encoding="utf-8")
|
|
|
|
first = Path(
|
|
named_executable(executable.as_posix(), name="nanobot-tui", directory=tmp_path / "run")
|
|
)
|
|
second = Path(
|
|
named_executable(executable.as_posix(), name="nanobot-tui", directory=tmp_path / "run")
|
|
)
|
|
|
|
assert first == second
|
|
assert first.name == "nanobot-tui"
|
|
assert first.is_symlink()
|
|
assert first.resolve() == executable
|
|
|
|
|
|
def test_named_executable_uses_original_on_windows(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
tmp_path: Path,
|
|
) -> None:
|
|
monkeypatch.setattr("nanobot.cli.process_identity.os.name", "nt")
|
|
|
|
assert (
|
|
named_executable("bun.exe", name="nanobot-tui", directory=tmp_path / "run")
|
|
== "bun.exe"
|
|
)
|