sysadminandClaude Opus 4.8
ffbd21db8e
Store per remote/org/repo/issue locks under GITEA_ISSUE_LOCK_DIR with atomic writes and per-session binding. Integrate own-branch adoption for lock recovery, update worktree-start and cleanup reconcile, and add tests documenting the ban on manual global lock seeding. Co-Authored-By: Claude Opus 4.8 (1M context) <[email protected]>
112 lines
3.8 KiB
Python
112 lines
3.8 KiB
Python
"""Tests for agent temp artifact detection and preflight warnings (#261)."""
|
|
import json
|
|
import os
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
sys.path.insert(0, str(__import__("pathlib").Path(__file__).resolve().parent.parent))
|
|
|
|
import agent_temp_artifacts
|
|
import mcp_server
|
|
|
|
|
|
class TestAgentTempArtifactPatterns(unittest.TestCase):
|
|
def test_detects_root_level_encode_emit_inline(self):
|
|
porcelain = (
|
|
"?? _encode_commit_payload.py\n"
|
|
"?? _emit_payload.py\n"
|
|
"?? _inline_b64.py\n"
|
|
)
|
|
self.assertEqual(
|
|
agent_temp_artifacts.find_agent_temp_artifacts_from_porcelain(porcelain),
|
|
["_emit_payload.py", "_encode_commit_payload.py", "_inline_b64.py"],
|
|
)
|
|
|
|
def test_ignores_nested_and_unrelated_untracked(self):
|
|
porcelain = (
|
|
"?? scripts/_encode_x.py\n"
|
|
"?? README-draft.md\n"
|
|
" M task_capability_map.py\n"
|
|
)
|
|
self.assertEqual(
|
|
agent_temp_artifacts.find_agent_temp_artifacts_from_porcelain(porcelain),
|
|
[],
|
|
)
|
|
|
|
|
|
class TestPreflightWarnings(unittest.TestCase):
|
|
def setUp(self):
|
|
self._snapshot = (
|
|
mcp_server._preflight_whoami_called,
|
|
mcp_server._preflight_capability_called,
|
|
)
|
|
mcp_server._preflight_whoami_called = True
|
|
mcp_server._preflight_capability_called = True
|
|
|
|
def tearDown(self):
|
|
mcp_server._preflight_whoami_called, mcp_server._preflight_capability_called = (
|
|
self._snapshot
|
|
)
|
|
|
|
@patch(
|
|
"mcp_server._get_workspace_porcelain",
|
|
return_value="?? _encode_commit_payload.py\n",
|
|
)
|
|
def test_assess_preflight_surfaces_warning_not_block(self, _porcelain):
|
|
status = mcp_server.assess_preflight_status()
|
|
self.assertTrue(status["preflight_ready"])
|
|
self.assertEqual(status["preflight_block_reasons"], [])
|
|
self.assertEqual(len(status["preflight_warnings"]), 1)
|
|
self.assertIn("_encode_commit_payload.py", status["preflight_warnings"][0])
|
|
|
|
|
|
# Issue-write tools are profile-gated (#69); gitea_lock_issue requires
|
|
# gitea.issue.comment (see task_capability_map), so the gate must be
|
|
# seeded exactly like tests/test_mcp_server.py::TestIssueLocking (#359).
|
|
ISSUE_WRITE_ENV = {
|
|
"GITEA_ALLOWED_OPERATIONS": (
|
|
"gitea.issue.create,gitea.issue.close,gitea.issue.comment"
|
|
),
|
|
}
|
|
|
|
|
|
class TestIssueLockArtifactWarning(unittest.TestCase):
|
|
def setUp(self):
|
|
self._lock_dir = tempfile.TemporaryDirectory()
|
|
env = {**ISSUE_WRITE_ENV, "GITEA_ISSUE_LOCK_DIR": self._lock_dir.name}
|
|
self._env_patcher = patch.dict(os.environ, env, clear=True)
|
|
self._env_patcher.start()
|
|
|
|
def tearDown(self):
|
|
self._env_patcher.stop()
|
|
self._lock_dir.cleanup()
|
|
|
|
@patch(
|
|
"mcp_server.issue_duplicate_context_fetcher",
|
|
return_value=([], [], {"status": "not_claimed"}),
|
|
)
|
|
@patch("mcp_server._auth", return_value="token x")
|
|
@patch("mcp_server._resolve", return_value=("h", "o", "r"))
|
|
@patch("issue_lock_worktree.read_worktree_git_state")
|
|
def test_lock_success_includes_artifact_warning(self, mock_state, *_mocks):
|
|
mock_state.return_value = {
|
|
"current_branch": "master",
|
|
"porcelain_status": "?? _emit_payload.py\n",
|
|
"base_equivalent": True,
|
|
"inspected_git_root": "/scratch/wt",
|
|
"base_branch": "origin/master",
|
|
}
|
|
result = mcp_server.gitea_lock_issue(
|
|
issue_number=261,
|
|
branch_name="docs/issue-261-agent-artifact-cleanup",
|
|
remote="prgs",
|
|
)
|
|
self.assertTrue(result["success"])
|
|
self.assertIn("warnings", result)
|
|
self.assertIn("_emit_payload.py", result["warnings"][0])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main() |