sysadminandClaude Opus 4.8
850b7c34b6
Add runbook section, root-level .gitignore patterns, and detection helper for _encode_/_emit_/_inline_ throwaway scripts. Surface non-blocking warnings from assess_preflight_status and gitea_lock_issue. Closes #261 Co-Authored-By: Claude Opus 4.8 (1M context) <[email protected]>
87 lines
3.0 KiB
Python
87 lines
3.0 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])
|
|
|
|
|
|
class TestIssueLockArtifactWarning(unittest.TestCase):
|
|
@patch("mcp_server.api_get_all", return_value=[])
|
|
@patch("mcp_server._auth", return_value="token x")
|
|
@patch("mcp_server._resolve", return_value=("h", "o", "r"))
|
|
@patch("mcp_server.ISSUE_LOCK_FILE", new_callable=lambda: tempfile.mktemp())
|
|
@patch("issue_lock_worktree.read_worktree_git_state")
|
|
def test_lock_success_includes_artifact_warning(self, mock_state, _lock_file, *_mocks):
|
|
mock_state.return_value = {
|
|
"current_branch": "master",
|
|
"porcelain_status": "?? _emit_payload.py\n",
|
|
}
|
|
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() |