mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2026-11-06 10:59:59 +00:00
117 lines
3.4 KiB
Python
117 lines
3.4 KiB
Python
import os
|
|
import shutil
|
|
import subprocess
|
|
import tempfile
|
|
import textwrap
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
|
|
REPO_ROOT = Path(__file__).resolve().parents[2]
|
|
SCRIPT_PATH = REPO_ROOT / "scripts" / "meta" / "resolve" / "pr" / "scope.sh"
|
|
|
|
|
|
@unittest.skipUnless(
|
|
shutil.which("jq"), "jq is required for the shell script under test"
|
|
)
|
|
class TestPullRequestScope(unittest.TestCase):
|
|
def _write_fake_gh(self, temp_dir: Path) -> None:
|
|
fake_gh = temp_dir / "gh"
|
|
fake_gh.write_text(
|
|
textwrap.dedent(
|
|
"""\
|
|
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
if [[ "${1:-}" != "api" ]]; then
|
|
echo "Unsupported gh invocation: $*" >&2
|
|
exit 1
|
|
fi
|
|
|
|
printf '%s\n' "${GH_FAKE_PULL_FILES_JSON:-}"
|
|
"""
|
|
),
|
|
encoding="utf-8",
|
|
)
|
|
fake_gh.chmod(0o755)
|
|
|
|
def _run_script(self, *, files_json: str):
|
|
with tempfile.TemporaryDirectory() as temp_dir_str:
|
|
temp_dir = Path(temp_dir_str)
|
|
output_file = temp_dir / "output.txt"
|
|
|
|
self._write_fake_gh(temp_dir)
|
|
|
|
env = os.environ.copy()
|
|
env.update(
|
|
{
|
|
"PR_NUMBER": "42",
|
|
"REPOSITORY": "kevinveenbirkenbach/infinito-nexus",
|
|
"GH_TOKEN": "test-token",
|
|
"GITHUB_OUTPUT": str(output_file),
|
|
"GH_FAKE_PULL_FILES_JSON": files_json,
|
|
"PATH": f"{temp_dir}:{env['PATH']}",
|
|
}
|
|
)
|
|
|
|
result = subprocess.run(
|
|
["bash", str(SCRIPT_PATH)],
|
|
cwd=REPO_ROOT,
|
|
env=env,
|
|
capture_output=True,
|
|
text=True,
|
|
check=True,
|
|
)
|
|
|
|
outputs = {}
|
|
for line in output_file.read_text(encoding="utf-8").splitlines():
|
|
if "=" not in line:
|
|
continue
|
|
key, value = line.split("=", 1)
|
|
outputs[key] = value
|
|
|
|
return result, outputs
|
|
|
|
def test_agents_only_scope_skips_ci_orchestrator(self):
|
|
files_json = """\
|
|
[
|
|
{"filename":"AGENTS.md"},
|
|
{"filename":"docs/agents/action/debug.md"},
|
|
{"filename":"docs/agents/action/develop.md"}
|
|
]
|
|
"""
|
|
_, outputs = self._run_script(files_json=files_json)
|
|
self.assertEqual(outputs["scope"], "agents")
|
|
self.assertEqual(outputs["run_ci_orchestrator"], "false")
|
|
|
|
def test_documentation_only_scope_skips_ci_orchestrator(self):
|
|
files_json = """\
|
|
[
|
|
{"filename":"README.md"},
|
|
{"filename":"docs/contributing/flow/pull-request.md"},
|
|
{"filename":".github/PULL_REQUEST_TEMPLATE/documentation.md"},
|
|
{"filename":"docs/contributing/setup.rst"}
|
|
]
|
|
"""
|
|
_, outputs = self._run_script(files_json=files_json)
|
|
self.assertEqual(outputs["scope"], "documentation")
|
|
self.assertEqual(outputs["run_ci_orchestrator"], "false")
|
|
|
|
def test_renamed_agent_file_falls_back_to_full_ci(self):
|
|
files_json = """\
|
|
[
|
|
{
|
|
"status": "renamed",
|
|
"filename": "docs/contributing/flow/workflow.md",
|
|
"previous_filename": "AGENTS.md"
|
|
}
|
|
]
|
|
"""
|
|
_, outputs = self._run_script(files_json=files_json)
|
|
self.assertEqual(outputs["scope"], "full")
|
|
self.assertEqual(outputs["run_ci_orchestrator"], "true")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|