ironclaw/tests/e2e/provider_operation_github_pull_cases.py

"""GitHub pull request full-path provider operation cases."""

import json

from provider_operation_github_common import (
    BRANCH,
    CODE_PATH,
    OWNER,
    PR_TITLE,
    REPO,
    REPO_PATH,
    pull_request,
    github_request,
    seed_branch,
    seed_pull_request,
)
from provider_operation_types import ProviderOperationCase

CREATED_PR = "REBORN_PROVIDER_CASE_CREATED_PR"
UPDATED_PR = "REBORN_PROVIDER_CASE_UPDATED_PR"
REVIEW_BODY = "REBORN_PROVIDER_CASE_REVIEW"
INLINE_COMMENT = "REBORN_PROVIDER_CASE_INLINE_COMMENT"
BASE_ARGS = {"owner": OWNER, "repo": REPO}


async def _seeded_branch(emulate_url: str) -> None:
    await seed_branch(emulate_url)
    pulls = await github_request(emulate_url, "GET", f"{REPO_PATH}/pulls")
    assert pulls == [], pulls


async def _seeded_pull(emulate_url: str) -> None:
    await seed_pull_request(emulate_url)


async def _seeded_review(emulate_url: str) -> None:
    await seed_pull_request(emulate_url)
    await github_request(
        emulate_url,
        "POST",
        f"{REPO_PATH}/pulls/1/reviews",
        payload={"body": REVIEW_BODY, "event": "COMMENT"},
        expected_status=201,
    )


async def _seeded_inline_comment(emulate_url: str) -> None:
    await seed_pull_request(emulate_url)
    await github_request(
        emulate_url,
        "POST",
        f"{REPO_PATH}/pulls/1/reviews",
        payload={
            "body": REVIEW_BODY,
            "event": "COMMENT",
            "comments": [
                {
                    "path": CODE_PATH,
                    "position": 1,
                    "body": INLINE_COMMENT,
                }
            ],
        },
        expected_status=201,
    )


async def _created_pull_outcome(emulate_url: str, preview: dict) -> None:
    created = await pull_request(emulate_url)
    assert created["title"] == CREATED_PR, created
    assert created["head"]["ref"] == BRANCH, created
    assert CREATED_PR in json.dumps(preview), preview


async def _seeded_pull_outcome(emulate_url: str, preview: dict) -> None:
    seeded = await pull_request(emulate_url)
    assert seeded["title"] == PR_TITLE, seeded
    assert PR_TITLE in json.dumps(preview), preview


async def _updated_pull_outcome(emulate_url: str, preview: dict) -> None:
    updated = await pull_request(emulate_url)
    assert updated["title"] == UPDATED_PR, updated
    assert UPDATED_PR in json.dumps(preview), preview


async def _pull_files_outcome(emulate_url: str, preview: dict) -> None:
    files = await github_request(emulate_url, "GET", f"{REPO_PATH}/pulls/1/files")
    assert isinstance(files, list)
    # Emulate currently models this successful empty state but not generated
    # pull-request diffs. The seeded branch still proves the PR is real.
    assert files == [], files
    assert preview["output_preview"] == "[]", preview


async def _create_review_outcome(emulate_url: str, preview: dict) -> None:
    reviews = await github_request(
        emulate_url, "GET", f"{REPO_PATH}/pulls/1/reviews"
    )
    assert isinstance(reviews, list)
    assert [review["body"] for review in reviews] == [REVIEW_BODY], reviews
    assert REVIEW_BODY in json.dumps(preview), preview


async def _list_comments_outcome(emulate_url: str, preview: dict) -> None:
    comments = await github_request(
        emulate_url, "GET", f"{REPO_PATH}/pulls/1/comments"
    )
    assert isinstance(comments, list)
    assert [comment["body"] for comment in comments] == [
        INLINE_COMMENT
    ], comments
    assert INLINE_COMMENT in json.dumps(preview), preview


async def _merge_outcome(emulate_url: str, preview: dict) -> None:
    merged = await pull_request(emulate_url)
    assert merged["merged"] is True, merged
    assert merged["state"] == "closed", merged
    assert json.loads(preview["output_preview"])["merged"] is True, preview


GITHUB_PULL_PROVIDER_OPERATION_CASES = (
    ProviderOperationCase(
        case_id="github_create_pull_request",
        provider_service="github",
        capability_id="github.create_pull_request",
        arguments={
            **BASE_ARGS,
            "title": CREATED_PR,
            "head": BRANCH,
            "base": "main",
            "body": "Created through the provider operation runner.",
        },
        assert_baseline=_seeded_branch,
        assert_outcome=_created_pull_outcome,
    ),
    ProviderOperationCase(
        case_id="github_get_pull_request",
        provider_service="github",
        capability_id="github.get_pull_request",
        arguments={**BASE_ARGS, "pr_number": 1},
        assert_baseline=_seeded_pull,
        assert_outcome=_seeded_pull_outcome,
    ),
    ProviderOperationCase(
        case_id="github_list_pull_requests",
        provider_service="github",
        capability_id="github.list_pull_requests",
        arguments={**BASE_ARGS, "state": "open"},
        assert_baseline=_seeded_pull,
        assert_outcome=_seeded_pull_outcome,
    ),
    ProviderOperationCase(
        case_id="github_update_pull_request",
        provider_service="github",
        capability_id="github.update_pull_request",
        arguments={**BASE_ARGS, "pr_number": 1, "title": UPDATED_PR},
        assert_baseline=_seeded_pull,
        assert_outcome=_updated_pull_outcome,
    ),
    ProviderOperationCase(
        case_id="github_get_pull_request_files",
        provider_service="github",
        capability_id="github.get_pull_request_files",
        arguments={**BASE_ARGS, "pr_number": 1},
        assert_baseline=_seeded_pull,
        assert_outcome=_pull_files_outcome,
    ),
    ProviderOperationCase(
        case_id="github_create_pr_review",
        provider_service="github",
        capability_id="github.create_pr_review",
        arguments={
            **BASE_ARGS,
            "pr_number": 1,
            "body": REVIEW_BODY,
            "event": "COMMENT",
        },
        assert_baseline=_seeded_pull,
        assert_outcome=_create_review_outcome,
    ),
    ProviderOperationCase(
        case_id="github_get_pull_request_reviews",
        provider_service="github",
        capability_id="github.get_pull_request_reviews",
        arguments={**BASE_ARGS, "pr_number": 1},
        assert_baseline=_seeded_review,
        assert_outcome=_create_review_outcome,
    ),
    ProviderOperationCase(
        case_id="github_list_pull_request_comments",
        provider_service="github",
        capability_id="github.list_pull_request_comments",
        arguments={**BASE_ARGS, "pr_number": 1},
        assert_baseline=_seeded_inline_comment,
        assert_outcome=_list_comments_outcome,
    ),
    ProviderOperationCase(
        case_id="github_merge_pull_request",
        provider_service="github",
        capability_id="github.merge_pull_request",
        arguments={
            **BASE_ARGS,
            "pr_number": 1,
            "merge_method": "squash",
            "commit_title": "Merge provider operation case",
        },
        assert_baseline=_seeded_pull,
        assert_outcome=_merge_outcome,
    ),
)