雷电芽衣
aa4b8a8bdf
Split the three oversized test files into support modules and topic files, evaluate the rmdir dir_fd capability once at import, and drop unused imports. Claude-Session: https://claude.ai/code/session_01Agoc9bRVE9KL7mmWtey9Vq
84 lines
2.0 KiB
Python
84 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
import argparse
|
|
import base64
|
|
import hashlib
|
|
import json
|
|
import os
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
import pytest
|
|
|
|
from elydora import cli
|
|
|
|
|
|
AGENT_ID = "agent-1"
|
|
AGENT_NAME = "opencode"
|
|
PRIVATE_KEY = base64.urlsafe_b64encode(bytes([19]) * 32).rstrip(b"=").decode()
|
|
|
|
|
|
def set_home(monkeypatch: pytest.MonkeyPatch, home: Path) -> None:
|
|
monkeypatch.setenv("HOME", str(home))
|
|
monkeypatch.setenv("USERPROFILE", str(home))
|
|
|
|
|
|
def write_json(path: Path, value: Any) -> None:
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
path.write_text(json.dumps(value, indent=2) + "\n", encoding="utf-8")
|
|
|
|
|
|
def write_secret(path: Path, value: str) -> None:
|
|
path.write_text(value, encoding="utf-8")
|
|
if os.name != "nt":
|
|
path.chmod(0o600)
|
|
|
|
|
|
def write_error_log(path: Path, chunks: int = 1) -> str:
|
|
digest = hashlib.sha256()
|
|
block = b"opaque historical error\n" * 4096
|
|
with path.open("wb") as file:
|
|
for _index in range(chunks):
|
|
file.write(block)
|
|
digest.update(block)
|
|
if os.name != "nt":
|
|
path.chmod(0o600)
|
|
return digest.hexdigest()
|
|
|
|
|
|
def stream_digest(path: Path) -> str:
|
|
digest = hashlib.sha256()
|
|
with path.open("rb") as file:
|
|
while chunk := file.read(1024 * 1024):
|
|
digest.update(chunk)
|
|
return digest.hexdigest()
|
|
|
|
|
|
def prepare_runtime(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
tmp_path: Path,
|
|
*,
|
|
agent_name: str = AGENT_NAME,
|
|
) -> tuple[Path, Path]:
|
|
home = tmp_path / "home"
|
|
root = home / ".elydora"
|
|
agent = root / AGENT_ID
|
|
set_home(monkeypatch, home)
|
|
write_json(
|
|
agent / "config.json",
|
|
{"agent_id": AGENT_ID, "agent_name": agent_name},
|
|
)
|
|
return root, agent
|
|
|
|
|
|
def uninstall_args(agent_name: str = AGENT_NAME) -> argparse.Namespace:
|
|
return cli.build_parser().parse_args(
|
|
[
|
|
"uninstall",
|
|
"--agent",
|
|
agent_name,
|
|
"--agent_id",
|
|
AGENT_ID,
|
|
]
|
|
)
|