open-swe/tests/test_review_styles_store.py

from __future__ import annotations

from unittest.mock import AsyncMock, patch

import pytest

from agent.dashboard.review_styles import (
    create_review_style,
    get_repo_custom_prompt,
    set_custom_prompt,
)


@pytest.mark.asyncio
async def test_get_repo_custom_prompt_returns_trimmed_text() -> None:
    with patch(
        "agent.dashboard.review_styles.get_review_style",
        new_callable=AsyncMock,
        return_value={"custom_prompt": "  Flag nil deref aggressively.\n"},
    ):
        prompt = await get_repo_custom_prompt("acme", "repo")
    assert prompt == "Flag nil deref aggressively."


@pytest.mark.asyncio
async def test_create_review_style_puts_new_record() -> None:
    mock_put = AsyncMock()
    with (
        patch(
            "agent.dashboard.review_styles._get_value", new_callable=AsyncMock, return_value=None
        ),
        patch("agent.dashboard.review_styles._client") as mock_client,
    ):
        mock_client.return_value.store.put_item = mock_put
        record = await create_review_style("acme/repo", "octo")
    assert record["full_name"] == "acme/repo"
    assert record["status"] == "idle"
    mock_put.assert_awaited_once()


@pytest.mark.asyncio
async def test_set_custom_prompt_updates_store() -> None:
    with (
        patch(
            "agent.dashboard.review_styles.get_review_style",
            new_callable=AsyncMock,
            return_value={"full_name": "acme/repo", "status": "completed"},
        ),
        patch(
            "agent.dashboard.review_styles.update_review_style", new_callable=AsyncMock
        ) as mock_up,
    ):
        await set_custom_prompt("acme/repo", "Use direct tone.")
    mock_up.assert_awaited_once()