mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-10-01 22:07:56 +00:00
Adam Moussa
ae1f883b4c
Applies the plan's C5 step: git mv every test per the domain-reorg
move-map (movemap-m50.txt) into tests/{agent,analyzer,auth,dashboard,
github,middleware,models,reviewer,sandbox,slack,tools,webhooks}/, plus
the 13 fork-only placements from the scoping report §2c (Atlassian
webhook tests -> tests/webhooks/, test_atlassian_connect.py and
test_auth_error_leak.py -> tests/auth/, jira/confluence util tests ->
tests/tools/, test_repo_binding_isolation.py -> tests/sandbox/,
bot-identity/autofix tests -> tests/github/).
Path-only move: the only content edits are parents[1] -> parents[2]
fixes in test_e2b_integration.py and test_daytona_integration.py,
required because their __file__-relative ROOT path gained one more
directory level in the move.
Monkeypatch retargets for these files were already completed in C4;
none remained outstanding here.
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
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()
|