mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-09-30 20:53:15 +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.
96 lines
3.4 KiB
Python
96 lines
3.4 KiB
Python
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
from langchain_core.messages import AIMessage, HumanMessage
|
|
|
|
from agent.middleware.notify_step_limit import notify_step_limit_reached
|
|
|
|
|
|
class TestNotifyStepLimitReached:
|
|
def _make_runtime(self) -> MagicMock:
|
|
return MagicMock()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_posts_slack_reply_when_limit_marker_present(self) -> None:
|
|
state = {"messages": [AIMessage(content="Model call limits exceeded: run limit reached")]}
|
|
|
|
with (
|
|
patch(
|
|
"agent.middleware.notify_step_limit.get_config",
|
|
return_value={
|
|
"configurable": {"slack_thread": {"channel_id": "C123", "thread_ts": "171.123"}}
|
|
},
|
|
),
|
|
patch(
|
|
"agent.middleware.notify_step_limit.post_slack_thread_reply",
|
|
new_callable=AsyncMock,
|
|
) as mock_post,
|
|
):
|
|
result = await notify_step_limit_reached.aafter_agent(state, self._make_runtime())
|
|
|
|
assert result is None
|
|
mock_post.assert_awaited_once()
|
|
assert mock_post.await_args.args[0:2] == ("C123", "171.123")
|
|
assert "maximum step limit" in mock_post.await_args.args[2]
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_posts_slack_reply_for_list_content_with_limit_marker(self) -> None:
|
|
state = {
|
|
"messages": [
|
|
AIMessage(
|
|
content=[
|
|
{"type": "text", "text": "Model call limits exceeded:"},
|
|
{"type": "text", "text": "run limit reached"},
|
|
]
|
|
)
|
|
]
|
|
}
|
|
|
|
with (
|
|
patch(
|
|
"agent.middleware.notify_step_limit.get_config",
|
|
return_value={
|
|
"configurable": {"slack_thread": {"channel_id": "C123", "thread_ts": "171.123"}}
|
|
},
|
|
),
|
|
patch(
|
|
"agent.middleware.notify_step_limit.post_slack_thread_reply",
|
|
new_callable=AsyncMock,
|
|
) as mock_post,
|
|
):
|
|
result = await notify_step_limit_reached.aafter_agent(state, self._make_runtime())
|
|
|
|
assert result is None
|
|
mock_post.assert_awaited_once()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_skips_when_limit_marker_absent(self) -> None:
|
|
state = {"messages": [HumanMessage(content="keep going")]}
|
|
|
|
with patch(
|
|
"agent.middleware.notify_step_limit.post_slack_thread_reply",
|
|
new_callable=AsyncMock,
|
|
) as mock_post:
|
|
result = await notify_step_limit_reached.aafter_agent(state, self._make_runtime())
|
|
|
|
assert result is None
|
|
mock_post.assert_not_called()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_skips_when_slack_thread_config_missing(self) -> None:
|
|
state = {"messages": [AIMessage(content="Model call limits exceeded: run limit reached")]}
|
|
|
|
with (
|
|
patch(
|
|
"agent.middleware.notify_step_limit.get_config",
|
|
return_value={"configurable": {}},
|
|
),
|
|
patch(
|
|
"agent.middleware.notify_step_limit.post_slack_thread_reply",
|
|
new_callable=AsyncMock,
|
|
) as mock_post,
|
|
):
|
|
result = await notify_step_limit_reached.aafter_agent(state, self._make_runtime())
|
|
|
|
assert result is None
|
|
mock_post.assert_not_called()
|