Carlos Florêncio
75 lines
2.8 KiB
Python
75 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
"""Contract tests for the Claude Code fork-review workflow."""
|
|
|
|
from pathlib import Path
|
|
import re
|
|
import unittest
|
|
|
|
|
|
REPO_ROOT = Path(__file__).resolve().parents[2]
|
|
WORKFLOW = REPO_ROOT / ".github" / "workflows" / "claude-code-review.yml"
|
|
MENTION_WORKFLOW = REPO_ROOT / ".github" / "workflows" / "claude.yml"
|
|
ALLOWED_USERS_INPUT = "allowed_non_write_users: ${{ github.event.pull_request.user.login }}"
|
|
|
|
|
|
def activity_types(workflow: str, event: str) -> list[str]:
|
|
trigger_section = workflow.partition("on:\n")[2].partition("\nconcurrency:")[0]
|
|
event_block = trigger_section.partition(f" {event}:\n")[2]
|
|
event_block = re.split(r"\n [a-z_]+:\n", event_block, maxsplit=1)[0]
|
|
match = re.search(r"^ types: \[([^]]+)]$", event_block, re.MULTILINE)
|
|
if match is None:
|
|
raise AssertionError(f"{event} activity types are missing")
|
|
return [activity.strip() for activity in match.group(1).split(",")]
|
|
|
|
|
|
class ClaudeCodeReviewWorkflowContractTest(unittest.TestCase):
|
|
def test_review_workflow_ignores_pr_updates(self) -> None:
|
|
workflow = WORKFLOW.read_text(encoding="utf-8")
|
|
|
|
self.assertEqual(["opened"], activity_types(workflow, "pull_request"))
|
|
self.assertEqual(
|
|
["opened", "labeled"],
|
|
activity_types(workflow, "pull_request_target"),
|
|
)
|
|
self.assertNotIn(" strip-safe-to-review:", workflow)
|
|
|
|
def test_fork_review_uses_only_open_or_safe_to_review_label(self) -> None:
|
|
workflow = WORKFLOW.read_text(encoding="utf-8")
|
|
_, separator, fork_job = workflow.partition(" claude-review-fork:")
|
|
|
|
self.assertTrue(separator, "Claude fork-review job is missing")
|
|
self.assertIn(
|
|
"github.event.action == 'labeled' && github.event.label.name == 'safe-to-review'",
|
|
fork_job,
|
|
)
|
|
self.assertRegex(
|
|
fork_job,
|
|
r"github\.event\.action == 'opened' &&\s+"
|
|
r"vars\.CLAUDE_REVIEW_ALLOWLIST != ''",
|
|
)
|
|
|
|
def test_manual_claude_mention_can_request_another_review(self) -> None:
|
|
workflow = MENTION_WORKFLOW.read_text(encoding="utf-8")
|
|
|
|
self.assertIn("issue_comment:\n types: [created]", workflow)
|
|
self.assertIn(
|
|
"contains(github.event.comment.body, '@claude')",
|
|
workflow,
|
|
)
|
|
|
|
def test_fork_review_forwards_allowlist_to_claude_action(self) -> None:
|
|
workflow = WORKFLOW.read_text(encoding="utf-8")
|
|
_, separator, fork_job = workflow.partition(" claude-review-fork:")
|
|
|
|
self.assertTrue(separator, "Claude fork-review job is missing")
|
|
self.assertIn(
|
|
ALLOWED_USERS_INPUT,
|
|
fork_job,
|
|
"fork review must forward its job-authorized pull request author "
|
|
"to Claude's allowed_non_write_users input",
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|