tegwick
e284c9f63c
Some checks failed
tamq-ci / test (push) Failing after 6s
Assistant: codex Assistant-Model: gpt-5.6-sol Assistant-Session: 01a03397-4d51-7fd1-8ff2-946eb22ea2bc
239 lines
9.4 KiB
Python
239 lines
9.4 KiB
Python
import os
|
|
import subprocess
|
|
|
|
from tamq import tmux
|
|
import pytest
|
|
|
|
|
|
def test_tmux_manager_builds_tap_windows(tmp_path, monkeypatch):
|
|
calls = []
|
|
manager = tmux.TmuxManager("tamq-test", command_dir=tmp_path / "commands")
|
|
repo_a = tmp_path / "a"
|
|
repo_b = tmp_path / "b"
|
|
repo_a.mkdir()
|
|
repo_b.mkdir()
|
|
monkeypatch.setattr(tmux, "validate_targets", lambda repos: None)
|
|
monkeypatch.setattr(tmux, "repository_paths", lambda: {"a": str(repo_a), "b": str(repo_b)})
|
|
monkeypatch.setattr(tmux.shutil, "which", lambda command: f"/bin/{command}")
|
|
def run(*args, check=True):
|
|
calls.append(args)
|
|
if args[:2] == ("list-windows", "-t"):
|
|
return "__tamq_boot"
|
|
if args[:2] == ("display-message", "-p"):
|
|
return "42"
|
|
return ""
|
|
monkeypatch.setattr(manager, "_run", run)
|
|
monkeypatch.setattr(tmux.subprocess, "run", lambda *args, **kwargs: type("R", (), {"returncode": 1})())
|
|
plan = manager.preflight(["a", "b"], "codex --quiet")
|
|
endpoint = manager.ensure_plan(plan, tap=True)
|
|
assert endpoint.endpoint_id == "tmux-amq-42"
|
|
assert endpoint.instance_id.startswith("tmux-amq-42-")
|
|
assert endpoint.repos == ["a", "b"]
|
|
new_session = next(call for call in calls if call and call[0] == "new-session")
|
|
assert new_session[-2:] == ("-c", str(repo_a))
|
|
assert "sleep" not in new_session
|
|
new_window = next(call for call in calls if call and call[0] == "new-window")
|
|
assert new_window[-2:] == ("-c", str(repo_b))
|
|
tap_calls = [call for call in calls if call and call[0] == "send-keys"]
|
|
assert len(tap_calls) == 2
|
|
assert all("tamq tap" in " ".join(call) for call in tap_calls)
|
|
assert all(endpoint.instance_id in " ".join(call) for call in tap_calls)
|
|
assert all("codex --quiet" in " ".join(call) for call in tap_calls)
|
|
assert ("select-window", "-t", "tamq-test:a") in calls
|
|
assert any(call and call[0] == "set-option" for call in calls)
|
|
|
|
|
|
def test_tmux_manager_reuses_session_instance_id(tmp_path, monkeypatch):
|
|
repo = tmp_path / "a"
|
|
repo.mkdir()
|
|
manager = tmux.TmuxManager("tamq-test", command_dir=tmp_path / "commands")
|
|
options = {"@tamq_managed": "1"}
|
|
monkeypatch.setattr(tmux, "validate_targets", lambda repos: None)
|
|
monkeypatch.setattr(tmux, "repository_paths", lambda: {"a": str(repo)})
|
|
monkeypatch.setattr(tmux.shutil, "which", lambda command: f"/bin/{command}")
|
|
|
|
def run(*args, check=True):
|
|
if args[:2] == ("display-message", "-p"):
|
|
return "42"
|
|
if args and args[0] == "show-options":
|
|
return options.get(args[-1], "")
|
|
if args and args[0] == "set-option":
|
|
options[args[-2]] = args[-1]
|
|
if args[:2] == ("list-windows", "-t"):
|
|
return "a"
|
|
return ""
|
|
|
|
monkeypatch.setattr(manager, "_run", run)
|
|
monkeypatch.setattr(
|
|
tmux.subprocess,
|
|
"run",
|
|
lambda *args, **kwargs: type("R", (), {"returncode": 0})(),
|
|
)
|
|
|
|
first = manager.ensure(["a"])
|
|
second = manager.ensure(["a"])
|
|
assert first.instance_id == second.instance_id
|
|
|
|
|
|
def test_preflight_rejects_foreign_session(tmp_path, monkeypatch):
|
|
repo = tmp_path / "a"
|
|
repo.mkdir()
|
|
manager = tmux.TmuxManager("tamq-test", command_dir=tmp_path / "commands")
|
|
monkeypatch.setattr(tmux, "validate_targets", lambda repos: None)
|
|
monkeypatch.setattr(tmux, "repository_paths", lambda: {"a": str(repo)})
|
|
monkeypatch.setattr(tmux.shutil, "which", lambda command: f"/bin/{command}")
|
|
monkeypatch.setattr(
|
|
tmux.subprocess,
|
|
"run",
|
|
lambda *args, **kwargs: type("R", (), {"returncode": 0})(),
|
|
)
|
|
|
|
def run(*args, check=True):
|
|
if args[:2] == ("display-message", "-p"):
|
|
return "42"
|
|
return ""
|
|
|
|
monkeypatch.setattr(manager, "_run", run)
|
|
with pytest.raises(tmux.TmuxError, match="not managed by tamq"):
|
|
manager.preflight(["a"])
|
|
|
|
|
|
def test_tmux_only_plan_starts_agent_without_tap(tmp_path, monkeypatch):
|
|
manager = tmux.TmuxManager("tamq-test", command_dir=tmp_path / "commands")
|
|
calls = []
|
|
plan = tmux.LaunchPlan(("a",), {"a": str(tmp_path)}, ("codex", "--quiet"))
|
|
|
|
def run(*args, check=True):
|
|
calls.append(args)
|
|
if args[:2] == ("display-message", "-p"):
|
|
return "42"
|
|
if args[:2] == ("list-windows", "-t"):
|
|
return "__tamq_boot"
|
|
return ""
|
|
|
|
monkeypatch.setattr(manager, "_run", run)
|
|
monkeypatch.setattr(
|
|
tmux.subprocess,
|
|
"run",
|
|
lambda *args, **kwargs: type("R", (), {"returncode": 1})(),
|
|
)
|
|
manager.ensure_plan(plan, tap=False)
|
|
send = next(call for call in calls if call and call[0] == "send-keys")
|
|
assert "codex --quiet" in send
|
|
assert "tamq tap" not in send
|
|
|
|
|
|
def test_neutral_plan_starts_shell_without_sending_keystrokes(tmp_path, monkeypatch):
|
|
command_dir = tmp_path / "commands"
|
|
manager = tmux.TmuxManager("tamq-test", command_dir=command_dir)
|
|
calls = []
|
|
plan = tmux.LaunchPlan(("a", "b"), {"a": str(tmp_path), "b": str(tmp_path)}, ())
|
|
|
|
def run(*args, check=True):
|
|
calls.append(args)
|
|
if args[:2] == ("display-message", "-p"):
|
|
return "42"
|
|
if args[:2] == ("list-windows", "-t"):
|
|
return "__tamq_boot"
|
|
return ""
|
|
|
|
monkeypatch.setattr(manager, "_run", run)
|
|
monkeypatch.setattr(
|
|
tmux.subprocess,
|
|
"run",
|
|
lambda *args, **kwargs: type("R", (), {"returncode": 1})(),
|
|
)
|
|
manager.ensure_plan(plan)
|
|
|
|
assert not any(call and call[0] == "send-keys" for call in calls)
|
|
new_session = next(call for call in calls if call and call[0] == "new-session")
|
|
new_window = next(call for call in calls if call and call[0] == "new-window")
|
|
assert "TAMQ_REPO=a" in new_session
|
|
assert "TAMQ_REPO=b" in new_window
|
|
assert f"PATH={command_dir}{os.pathsep}" in " ".join(new_session)
|
|
assert f"PATH={command_dir}{os.pathsep}" in " ".join(new_window)
|
|
assert (command_dir / "@a").is_file()
|
|
assert (command_dir / "@a:").is_file()
|
|
assert (command_dir / "@b").is_file()
|
|
assert (command_dir / "@b:").is_file()
|
|
assert (command_dir / "@").is_file()
|
|
|
|
|
|
def test_address_commands_preserve_message_arguments(tmp_path):
|
|
command_dir = tmp_path / "commands"
|
|
manager = tmux.TmuxManager(
|
|
"tamq-test", tamq_command=("echo",), command_dir=command_dir,
|
|
)
|
|
manager._install_address_commands(["audit-core"])
|
|
|
|
for name in ("@audit-core", "@audit-core:"):
|
|
result = subprocess.run(
|
|
[str(command_dir / name), "Some message!", "$value", "; literal", "--from"],
|
|
text=True,
|
|
capture_output=True,
|
|
check=True,
|
|
)
|
|
assert result.stdout == "send -- @audit-core: Some message! $value ; literal --from\n"
|
|
|
|
result = subprocess.run(
|
|
[str(command_dir / "@"), "Some reply!", "$value", "--literal"],
|
|
text=True,
|
|
capture_output=True,
|
|
check=True,
|
|
)
|
|
assert result.stdout == "reply -- Some reply! $value --literal\n"
|
|
generic = (command_dir / "@").read_text(encoding="utf-8")
|
|
assert 'TAMQ_RECIPIENTS=' in generic
|
|
assert '["audit-core"]' in generic
|
|
|
|
|
|
def test_address_commands_reject_unsafe_repository_names(tmp_path):
|
|
manager = tmux.TmuxManager("tamq-test", command_dir=tmp_path / "commands")
|
|
with pytest.raises(tmux.TmuxError, match="cannot be exposed"):
|
|
manager._install_address_commands(["../outside"])
|
|
|
|
|
|
def test_address_commands_do_not_replace_unowned_commands(tmp_path):
|
|
command_dir = tmp_path / "commands"
|
|
command_dir.mkdir()
|
|
existing = command_dir / "@audit-core"
|
|
existing.write_text("#!/bin/sh\necho mine\n", encoding="utf-8")
|
|
manager = tmux.TmuxManager("tamq-test", command_dir=command_dir)
|
|
|
|
with pytest.raises(tmux.TmuxError, match="refusing to replace"):
|
|
manager._install_address_commands(["audit-core"])
|
|
assert existing.read_text(encoding="utf-8") == "#!/bin/sh\necho mine\n"
|
|
|
|
|
|
def test_address_commands_do_not_replace_unowned_bare_reply(tmp_path):
|
|
command_dir = tmp_path / "commands"
|
|
command_dir.mkdir()
|
|
existing = command_dir / "@"
|
|
existing.write_text("#!/bin/sh\necho mine\n", encoding="utf-8")
|
|
manager = tmux.TmuxManager("tamq-test", command_dir=command_dir)
|
|
|
|
with pytest.raises(tmux.TmuxError, match="refusing to replace"):
|
|
manager._install_address_commands(["audit-core"])
|
|
assert existing.read_text(encoding="utf-8") == "#!/bin/sh\necho mine\n"
|
|
|
|
|
|
def test_preflight_rejects_missing_agent(tmp_path, monkeypatch):
|
|
repo = tmp_path / "a"
|
|
repo.mkdir()
|
|
manager = tmux.TmuxManager("tamq-test", command_dir=tmp_path / "commands")
|
|
monkeypatch.setattr(tmux, "validate_targets", lambda repos: None)
|
|
monkeypatch.setattr(tmux, "repository_paths", lambda: {"a": str(repo)})
|
|
monkeypatch.setattr(tmux.shutil, "which", lambda command: None if command == "missing-agent" else f"/bin/{command}")
|
|
with pytest.raises(tmux.TmuxError, match="initial command not found"):
|
|
manager.preflight(["a"], "missing-agent")
|
|
|
|
|
|
def test_preflight_deduplicates_repositories(tmp_path, monkeypatch):
|
|
repo = tmp_path / "a"
|
|
repo.mkdir()
|
|
manager = tmux.TmuxManager("tamq-test", command_dir=tmp_path / "commands")
|
|
monkeypatch.setattr(tmux, "validate_targets", lambda repos: None)
|
|
monkeypatch.setattr(tmux, "repository_paths", lambda: {"a": str(repo)})
|
|
monkeypatch.setattr(tmux.shutil, "which", lambda command: f"/bin/{command}")
|
|
plan = manager.preflight(["a", "a"], "codex")
|
|
assert plan.repos == ("a",)
|