open-swe/tests/test_agent_thread_pr_state.py

"""Unit tests for agent-thread PR-state tracking from PR webhook events."""

from __future__ import annotations

from typing import Any
from unittest.mock import AsyncMock, MagicMock, patch

import pytest

from agent import webapp


def _pr_payload(*, state: str, merged: bool = False, draft: bool = False) -> dict[str, Any]:
    return {
        "pull_request": {
            "html_url": "https://github.com/lc/repo/pull/7",
            "state": state,
            "merged": merged,
            "draft": draft,
        }
    }


def test_pr_state_from_payload_merged() -> None:
    assert webapp._pr_state_from_payload(_pr_payload(state="closed", merged=True)) == "merged"


def test_pr_state_from_payload_closed() -> None:
    assert webapp._pr_state_from_payload(_pr_payload(state="closed")) == "closed"


def test_pr_state_from_payload_draft() -> None:
    assert webapp._pr_state_from_payload(_pr_payload(state="open", draft=True)) == "draft"


def test_pr_state_from_payload_open() -> None:
    assert webapp._pr_state_from_payload(_pr_payload(state="open")) == "open"


def test_pr_state_from_payload_missing_pull_request() -> None:
    assert webapp._pr_state_from_payload({}) is None


@pytest.mark.asyncio
async def test_update_agent_thread_pr_state_updates_matching_thread() -> None:
    fake_client = MagicMock()
    fake_client.threads.search = AsyncMock(
        return_value=[
            {
                "thread_id": "t1",
                "metadata": {"kind": "agent", "pr_state": "draft"},
            }
        ]
    )
    fake_client.threads.update = AsyncMock()

    with patch("agent.webapp.get_client", return_value=fake_client):
        await webapp.update_agent_thread_pr_state(_pr_payload(state="closed"))

    fake_client.threads.search.assert_awaited_once()
    fake_client.threads.update.assert_awaited_once()
    assert fake_client.threads.update.await_args.kwargs["thread_id"] == "t1"
    assert fake_client.threads.update.await_args.kwargs["metadata"] == {"pr_state": "closed"}


@pytest.mark.asyncio
async def test_update_agent_thread_pr_state_skips_reviewer_threads() -> None:
    fake_client = MagicMock()
    fake_client.threads.search = AsyncMock(
        return_value=[{"thread_id": "rev", "metadata": {"kind": "reviewer"}}]
    )
    fake_client.threads.update = AsyncMock()

    with patch("agent.webapp.get_client", return_value=fake_client):
        await webapp.update_agent_thread_pr_state(_pr_payload(state="closed"))

    fake_client.threads.update.assert_not_called()


@pytest.mark.asyncio
async def test_update_agent_thread_pr_state_noop_when_state_unchanged() -> None:
    fake_client = MagicMock()
    fake_client.threads.search = AsyncMock(
        return_value=[{"thread_id": "t1", "metadata": {"pr_state": "merged"}}]
    )
    fake_client.threads.update = AsyncMock()

    with patch("agent.webapp.get_client", return_value=fake_client):
        await webapp.update_agent_thread_pr_state(_pr_payload(state="closed", merged=True))

    fake_client.threads.update.assert_not_called()