mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-10-26 21:26:39 +00:00
d48ea2a9a3
* refactor: remove unused proxy policy and private constants * refactor(sandbox): share provisioner runtime mount path
39 lines
2.0 KiB
Python
39 lines
2.0 KiB
Python
"""Keep Gateway/provisioner fixed and dynamic mount contracts in lockstep."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import ast
|
|
from pathlib import Path
|
|
|
|
REPO_ROOT = Path(__file__).resolve().parents[2]
|
|
|
|
|
|
def _literal_assignment(path: Path, name: str):
|
|
tree = ast.parse(path.read_text(encoding="utf-8"), filename=str(path))
|
|
for node in tree.body:
|
|
if isinstance(node, ast.Assign) and any(isinstance(target, ast.Name) and target.id == name for target in node.targets):
|
|
if isinstance(node.value, ast.Set):
|
|
return {_literal_assignment(path, item.id) if isinstance(item, ast.Name) else ast.literal_eval(item) for item in node.value.elts}
|
|
return ast.literal_eval(node.value)
|
|
raise AssertionError(f"{name} not found in {path}")
|
|
|
|
|
|
def test_gateway_and_provisioner_extra_mount_contracts_match() -> None:
|
|
gateway_path = REPO_ROOT / "backend/packages/harness/deerflow/community/aio_sandbox/remote_backend.py"
|
|
provisioner_path = REPO_ROOT / "docker/provisioner/app.py"
|
|
|
|
gateway_paths = _literal_assignment(gateway_path, "_PROVISIONER_EXTRA_MOUNT_PATHS")
|
|
provisioner_paths = _literal_assignment(provisioner_path, "ALLOWED_EXTRA_MOUNT_PATHS")
|
|
gateway_categories = _literal_assignment(gateway_path, "_MANAGED_SKILL_CATEGORY_NAMES")
|
|
provisioner_categories = _literal_assignment(provisioner_path, "MANAGED_SKILL_CATEGORY_NAMES")
|
|
gateway_reserved = _literal_assignment(gateway_path, "_RESERVED_SANDBOX_MOUNT_PATHS")
|
|
provisioner_reserved = _literal_assignment(provisioner_path, "RESERVED_SANDBOX_MOUNT_PATHS")
|
|
|
|
assert gateway_paths == provisioner_paths
|
|
assert gateway_categories == provisioner_categories
|
|
assert gateway_reserved == provisioner_reserved
|
|
assert _literal_assignment(provisioner_path, "DEFAULT_SKILLS_CONTAINER_PATH") == "/mnt/skills"
|
|
assert "/mnt/integrations/lark-cli/runtime" in gateway_paths
|
|
assert "/mnt/integrations/lark-cli/config/locks" in gateway_paths
|
|
assert _literal_assignment(provisioner_path, "MAX_EXTRA_MOUNTS") == 10
|