249 lines
8.3 KiB
Python
249 lines
8.3 KiB
Python
from __future__ import annotations
|
|
|
|
import plistlib
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from k1link.local_service_launchd import (
|
|
MissionCoreLaunchAgentError,
|
|
plan_mission_core_launch_agent,
|
|
)
|
|
|
|
|
|
def _write_agent(
|
|
*,
|
|
path: Path,
|
|
repository: Path,
|
|
uv_entrypoint: Path,
|
|
environment: dict[str, str] | None = None,
|
|
) -> None:
|
|
path.write_bytes(
|
|
plistlib.dumps(
|
|
{
|
|
"Label": "com.nodedc.mission-core.local",
|
|
"ProgramArguments": [str(uv_entrypoint), "run", "k1link", "serve"],
|
|
"WorkingDirectory": str(repository),
|
|
"EnvironmentVariables": environment or {"PATH": "/usr/bin:/bin"},
|
|
}
|
|
)
|
|
)
|
|
path.chmod(0o600)
|
|
|
|
|
|
def test_launch_agent_plan_disables_sync_and_enables_watchdog(tmp_path: Path) -> None:
|
|
repository = tmp_path / "repo"
|
|
repository.mkdir()
|
|
uv_entrypoint = tmp_path / "uv"
|
|
uv_entrypoint.write_text("#!/bin/sh\n")
|
|
uv_entrypoint.chmod(0o700)
|
|
agent = tmp_path / "agent.plist"
|
|
_write_agent(path=agent, repository=repository, uv_entrypoint=uv_entrypoint)
|
|
|
|
plan = plan_mission_core_launch_agent(
|
|
repository_root=repository,
|
|
agent_path=agent,
|
|
)
|
|
desired = plistlib.loads(plan.desired_payload)
|
|
|
|
assert desired["ProgramArguments"] == [
|
|
str(uv_entrypoint),
|
|
"run",
|
|
"--no-sync",
|
|
"k1link",
|
|
"serve",
|
|
]
|
|
assert desired["EnvironmentVariables"]["MISSIONCORE_SERVICE_WATCHDOG"] == "1"
|
|
assert desired["KeepAlive"] is True
|
|
assert desired["RunAtLoad"] is True
|
|
assert desired["AbandonProcessGroup"] is False
|
|
assert desired["ExitTimeOut"] == 20
|
|
assert plan.current_sha256 != plan.desired_sha256
|
|
assert plan.current_working_directory == repository
|
|
assert plan.desired_working_directory == repository
|
|
assert plan.preserved_data_directory is None
|
|
assert "MISSIONCORE_DATA_DIR" not in desired["EnvironmentVariables"]
|
|
assert plan.to_dict()["changes"]["repository_migration"] is False
|
|
assert plan.to_dict()["changes"]["data_directory_preserved"] is False
|
|
|
|
|
|
def test_launch_agent_plan_rejects_cross_repository_migration_by_default(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
current_repository = tmp_path / "current-repo"
|
|
current_repository.mkdir()
|
|
desired_repository = tmp_path / "desired-repo"
|
|
desired_repository.mkdir()
|
|
uv_entrypoint = tmp_path / "uv"
|
|
uv_entrypoint.write_text("#!/bin/sh\n")
|
|
uv_entrypoint.chmod(0o700)
|
|
agent = tmp_path / "agent.plist"
|
|
_write_agent(
|
|
path=agent,
|
|
repository=current_repository,
|
|
uv_entrypoint=uv_entrypoint,
|
|
)
|
|
|
|
with pytest.raises(
|
|
MissionCoreLaunchAgentError,
|
|
match="current launch agent targets another repository",
|
|
):
|
|
plan_mission_core_launch_agent(
|
|
repository_root=desired_repository,
|
|
agent_path=agent,
|
|
)
|
|
|
|
|
|
def test_launch_agent_plan_binds_explicit_cross_repository_migration(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
current_repository = tmp_path / "current-repo"
|
|
current_repository.mkdir()
|
|
desired_repository = tmp_path / "desired-repo"
|
|
desired_repository.mkdir()
|
|
other_repository = tmp_path / "other-repo"
|
|
other_repository.mkdir()
|
|
data_directory = current_repository / ".runtime" / "mission-core"
|
|
data_directory.mkdir(parents=True, mode=0o700)
|
|
uv_entrypoint = tmp_path / "uv"
|
|
uv_entrypoint.write_text("#!/bin/sh\n")
|
|
uv_entrypoint.chmod(0o700)
|
|
agent = tmp_path / "agent.plist"
|
|
_write_agent(
|
|
path=agent,
|
|
repository=current_repository,
|
|
uv_entrypoint=uv_entrypoint,
|
|
)
|
|
|
|
with pytest.raises(
|
|
MissionCoreLaunchAgentError,
|
|
match="does not target the expected current repository",
|
|
):
|
|
plan_mission_core_launch_agent(
|
|
repository_root=desired_repository,
|
|
agent_path=agent,
|
|
expected_current_repository_root=other_repository,
|
|
)
|
|
|
|
plan = plan_mission_core_launch_agent(
|
|
repository_root=desired_repository,
|
|
agent_path=agent,
|
|
expected_current_repository_root=current_repository,
|
|
)
|
|
desired = plistlib.loads(plan.desired_payload)
|
|
document = plan.to_dict()
|
|
|
|
assert plan.current_working_directory == current_repository
|
|
assert plan.desired_working_directory == desired_repository
|
|
assert plan.preserved_data_directory == data_directory
|
|
assert desired["WorkingDirectory"] == str(desired_repository)
|
|
assert desired["EnvironmentVariables"]["MISSIONCORE_DATA_DIR"] == str(data_directory)
|
|
assert document["current_working_directory"] == str(current_repository)
|
|
assert document["desired_working_directory"] == str(desired_repository)
|
|
assert document["preserved_data_directory"] == str(data_directory)
|
|
assert document["changes"]["repository_migration"] is True
|
|
assert document["changes"]["data_directory_preserved"] is True
|
|
assert document["changes"]["preserved_data_directory"] == str(data_directory)
|
|
|
|
|
|
def test_launch_agent_migration_retains_existing_private_data_directory(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
current_repository = tmp_path / "current-repo"
|
|
current_repository.mkdir()
|
|
desired_repository = tmp_path / "desired-repo"
|
|
desired_repository.mkdir()
|
|
configured_data_directory = tmp_path / "canonical-data"
|
|
configured_data_directory.mkdir(mode=0o700)
|
|
uv_entrypoint = tmp_path / "uv"
|
|
uv_entrypoint.write_text("#!/bin/sh\n")
|
|
uv_entrypoint.chmod(0o700)
|
|
agent = tmp_path / "agent.plist"
|
|
_write_agent(
|
|
path=agent,
|
|
repository=current_repository,
|
|
uv_entrypoint=uv_entrypoint,
|
|
environment={
|
|
"PATH": "/usr/bin:/bin",
|
|
"MISSIONCORE_DATA_DIR": str(configured_data_directory),
|
|
},
|
|
)
|
|
|
|
plan = plan_mission_core_launch_agent(
|
|
repository_root=desired_repository,
|
|
agent_path=agent,
|
|
expected_current_repository_root=current_repository,
|
|
)
|
|
desired = plistlib.loads(plan.desired_payload)
|
|
|
|
assert plan.preserved_data_directory == configured_data_directory
|
|
assert (
|
|
desired["EnvironmentVariables"]["MISSIONCORE_DATA_DIR"]
|
|
== str(configured_data_directory)
|
|
)
|
|
|
|
|
|
def test_launch_agent_migration_rejects_symlinked_data_directory(tmp_path: Path) -> None:
|
|
current_repository = tmp_path / "current-repo"
|
|
current_repository.mkdir()
|
|
desired_repository = tmp_path / "desired-repo"
|
|
desired_repository.mkdir()
|
|
private_directory = tmp_path / "private-data"
|
|
private_directory.mkdir(mode=0o700)
|
|
symlinked_directory = tmp_path / "data-link"
|
|
symlinked_directory.symlink_to(private_directory, target_is_directory=True)
|
|
uv_entrypoint = tmp_path / "uv"
|
|
uv_entrypoint.write_text("#!/bin/sh\n")
|
|
uv_entrypoint.chmod(0o700)
|
|
agent = tmp_path / "agent.plist"
|
|
_write_agent(
|
|
path=agent,
|
|
repository=current_repository,
|
|
uv_entrypoint=uv_entrypoint,
|
|
environment={
|
|
"PATH": "/usr/bin:/bin",
|
|
"MISSIONCORE_DATA_DIR": str(symlinked_directory),
|
|
},
|
|
)
|
|
|
|
with pytest.raises(
|
|
MissionCoreLaunchAgentError,
|
|
match="data directory is not a private canonical directory",
|
|
):
|
|
plan_mission_core_launch_agent(
|
|
repository_root=desired_repository,
|
|
agent_path=agent,
|
|
expected_current_repository_root=current_repository,
|
|
)
|
|
|
|
|
|
def test_launch_agent_migration_rejects_nonprivate_default_data_directory(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
current_repository = tmp_path / "current-repo"
|
|
current_repository.mkdir()
|
|
desired_repository = tmp_path / "desired-repo"
|
|
desired_repository.mkdir()
|
|
data_directory = current_repository / ".runtime" / "mission-core"
|
|
data_directory.mkdir(parents=True, mode=0o755)
|
|
data_directory.chmod(0o755)
|
|
uv_entrypoint = tmp_path / "uv"
|
|
uv_entrypoint.write_text("#!/bin/sh\n")
|
|
uv_entrypoint.chmod(0o700)
|
|
agent = tmp_path / "agent.plist"
|
|
_write_agent(
|
|
path=agent,
|
|
repository=current_repository,
|
|
uv_entrypoint=uv_entrypoint,
|
|
)
|
|
|
|
with pytest.raises(
|
|
MissionCoreLaunchAgentError,
|
|
match="data directory is not a private canonical directory",
|
|
):
|
|
plan_mission_core_launch_agent(
|
|
repository_root=desired_repository,
|
|
agent_path=agent,
|
|
expected_current_repository_root=current_repository,
|
|
)
|