update
Some checks failed
test / unit (push) Has been cancelled
test / e2e (push) Has been cancelled

This commit is contained in:
2026-05-18 04:05:46 +03:00
parent 082468e2bd
commit 9fb08d035f
5 changed files with 158 additions and 12 deletions

View File

@@ -56,13 +56,23 @@ class ActionExecutor:
if primitive_result is None:
primitive_result = ActionResult(action.id, action.type, "success")
results.append(primitive_result)
self.audit.write(
"action_success",
{"plan": plan.name, "action": action},
)
if primitive_result.status == "failed":
self.audit.write(
"action_failed",
{
"plan": plan.name,
"action": action,
"result": primitive_result,
},
)
else:
self.audit.write(
"action_success",
{"plan": plan.name, "action": action},
)
if action.rollback_policy == RollbackPolicy.BARRIER:
rollback_stack.clear()
elif rollback is not None:
elif primitive_result.status != "failed" and rollback is not None:
rollback_stack.append(rollback)
except Exception as e:
results.append(ActionResult(action.id, action.type, "failed", str(e)))
@@ -288,7 +298,8 @@ class ActionExecutor:
interactive=bool(p.get("interactive", False)),
detach_keys=p.get("detach_keys"),
)
return ActionResult(action.id, action.type, "success", returncode=returncode)
status = "success" if returncode == 0 else "failed"
return ActionResult(action.id, action.type, status, returncode=returncode)
if t == "tmux.new_session":
self.ctx.runtime.tmux.new_session(

View File

@@ -65,8 +65,10 @@ class FileSystem:
shutil.copy2(source, target)
def copy_tree(self, source: Path, target: Path) -> None:
if target.exists() or target.is_symlink():
raise FlowError(f"Copy target already exists: {target}")
self.ensure_dir(target.parent)
shutil.copytree(source, target, dirs_exist_ok=True)
shutil.copytree(source, target)
def create_symlink(
self,
@@ -155,4 +157,3 @@ class FileSystem:
with open(tmp, "w", encoding="utf-8") as f:
json.dump(data, f, indent=2)
os.replace(tmp, path)

View File

@@ -354,8 +354,14 @@ class PackageService:
return ActionPlan(
name="packages.install",
domain_actions=tuple(actions),
primitive_actions=(self._state_write_action(projected),),
domain_actions=(
*actions,
self._state_domain_action(
"package.state.write.install",
"install",
projected,
),
),
)
def _remove_action_plan(
@@ -399,8 +405,14 @@ class PackageService:
projected.packages.pop(op.name, None)
return ActionPlan(
name="packages.remove",
domain_actions=tuple(actions),
primitive_actions=(self._state_write_action(projected),),
domain_actions=(
*actions,
self._state_domain_action(
"package.state.write.remove",
"remove",
projected,
),
),
)
def _binary_install_primitives(
@@ -548,6 +560,21 @@ class PackageService:
rollback_policy=RollbackPolicy.ROLLBACKABLE,
)
def _state_domain_action(
self,
action_id: str,
action: str,
state: InstalledState,
) -> DomainAction:
return DomainAction(
id=action_id,
kind="package",
action=action,
description=f"Write package state to {paths.INSTALLED_STATE}",
payload={"primitive_actions": (self._state_write_action(state),)},
rollback_policy=RollbackPolicy.ROLLBACKABLE,
)
def _executor(self) -> ActionExecutor:
return ActionExecutor(self.ctx, audit_path=paths.INSTALLED_STATE.parent / "actions.jsonl")

View File

@@ -3,6 +3,7 @@
from __future__ import annotations
import json
import subprocess
import sys
import pytest
@@ -210,3 +211,64 @@ def test_executor_dispatches_container_and_tmux_primitives(tmp_path):
assert [
"tmux", "set-option", "-t", "dev-api", "default-command", "flow dev exec api",
] in runner.calls
def test_container_exec_nonzero_result_is_failed(tmp_path):
runner = FakeRunner(
responses={
("exec", "dev-api", "false"): subprocess.CompletedProcess(
["docker", "exec", "dev-api", "false"], 7, stdout="", stderr=""
),
}
)
ctx = _ctx()
ctx.runtime.runner = runner
ctx.runtime.containers = ContainerRuntime(runner, binary="docker")
plan = ActionPlan(
name="container-exec-failure",
primitive_actions=(
PrimitiveAction(
id="container.exec",
type="container.exec",
description="Run failing command in container",
payload={"name": "dev-api", "argv": ("false",)},
),
),
)
summary = ActionExecutor(ctx, audit_path=tmp_path / "actions.jsonl").execute(plan)
assert summary.results[0].status == "failed"
assert summary.results[0].returncode == 7
audit_events = [
json.loads(line)["event"]
for line in (tmp_path / "actions.jsonl").read_text(encoding="utf-8").splitlines()
]
assert "action_failed" in audit_events
assert "action_success" not in audit_events
def test_copy_directory_refuses_existing_target(tmp_path):
source = tmp_path / "source"
target = tmp_path / "target"
source.mkdir()
target.mkdir()
(source / "new.txt").write_text("new", encoding="utf-8")
(target / "existing.txt").write_text("keep", encoding="utf-8")
plan = ActionPlan(
name="copy-dir-existing-target",
primitive_actions=(
PrimitiveAction(
id="copy-dir",
type="file.copy",
description="Copy directory",
payload={"source": source, "target": target},
),
),
)
with pytest.raises(FlowError, match="Copy target already exists"):
ActionExecutor(_ctx(), audit_path=tmp_path / "actions.jsonl").execute(plan)
assert not (target / "new.txt").exists()
assert (target / "existing.txt").read_text(encoding="utf-8") == "keep"

View File

@@ -1,6 +1,7 @@
"""Tests for PackageService."""
import io
import subprocess
import tarfile
import urllib.error
from pathlib import Path
@@ -16,6 +17,7 @@ from flow.core.runtime import SystemRuntime
from flow.core import paths
from flow.domain.packages.models import InstalledPackage, InstalledState, PackageDef
from flow.app.packages import PackageService
from tests.fakes import FakeRunner
def _make_ctx(tmp_path, manifest=None):
@@ -134,6 +136,49 @@ class TestPackageService:
assert (home / ".local" / "share" / "nvim" / "runtime.txt").exists()
assert (home / ".local" / "share" / "man" / "man1" / "nvim.1").exists()
def test_install_does_not_write_state_when_package_manager_install_fails(
self, tmp_path, monkeypatch,
):
state_path = tmp_path / "installed.json"
monkeypatch.setattr(paths, "INSTALLED_STATE", state_path)
monkeypatch.setattr("flow.app.packages.detect_package_manager", lambda: "apt")
class FailingInstallRunner(FakeRunner):
def run(
self, argv, *, cwd=None, env=None, capture_output=True,
check=False, timeout=None,
):
parts = list(argv)
self.calls.append(parts)
self.timeouts.append(timeout)
if parts[:3] == ["sudo", "apt-get", "install"]:
return subprocess.CompletedProcess(
parts, 42, stdout="", stderr="install failed"
)
return subprocess.CompletedProcess(parts, 0, stdout="", stderr="")
rt = SystemRuntime()
rt.runner = FailingInstallRunner()
ctx = FlowContext(
config=AppConfig(),
manifest={},
platform=PlatformInfo(),
console=Console(color=False),
runtime=rt,
)
svc = PackageService(ctx)
pkg = PackageDef(
name="fd", type="pkg", sources={},
source=None, version=None, asset_pattern=None,
platform_map={}, extract_dir=None, install={},
post_install=None,
)
with pytest.raises(FlowError, match="install failed"):
svc.install([pkg])
assert not state_path.exists()
def test_post_install_with_sudo_runs_unchecked(self, tmp_path, monkeypatch):
"""No allow_sudo gate -- post-install scripts run as written."""
ctx = _make_ctx(tmp_path)