mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-09-30 13: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.
110 lines
4.3 KiB
Python
110 lines
4.3 KiB
Python
from __future__ import annotations
|
|
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import httpx
|
|
import pytest
|
|
|
|
from agent.utils import agents_md
|
|
|
|
|
|
def _make_response(status: int, text: str = "") -> MagicMock:
|
|
resp = MagicMock()
|
|
resp.status_code = status
|
|
resp.text = text
|
|
return resp
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_fetch_agents_md_returns_content() -> None:
|
|
with patch("httpx.AsyncClient") as mock_client_cls:
|
|
client = MagicMock()
|
|
client.get = AsyncMock(return_value=_make_response(200, "# AGENTS.md\nrules"))
|
|
mock_client_cls.return_value.__aenter__ = AsyncMock(return_value=client)
|
|
mock_client_cls.return_value.__aexit__ = AsyncMock(return_value=None)
|
|
result = await agents_md.fetch_agents_md("acme", "repo", "main", token="tok")
|
|
assert result == "# AGENTS.md\nrules"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_fetch_agents_md_falls_back_to_claude_md() -> None:
|
|
with patch("httpx.AsyncClient") as mock_client_cls:
|
|
client = MagicMock()
|
|
client.get = AsyncMock(
|
|
side_effect=[
|
|
_make_response(404),
|
|
_make_response(200, "# CLAUDE.md\nrules"),
|
|
]
|
|
)
|
|
mock_client_cls.return_value.__aenter__ = AsyncMock(return_value=client)
|
|
mock_client_cls.return_value.__aexit__ = AsyncMock(return_value=None)
|
|
result = await agents_md.fetch_agents_md("acme", "repo", "main", token="tok")
|
|
assert result == "# CLAUDE.md\nrules"
|
|
assert client.get.await_count == 2
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_fetch_agents_md_returns_none_when_both_missing() -> None:
|
|
with patch("httpx.AsyncClient") as mock_client_cls:
|
|
client = MagicMock()
|
|
client.get = AsyncMock(
|
|
side_effect=[
|
|
_make_response(404),
|
|
_make_response(404),
|
|
]
|
|
)
|
|
mock_client_cls.return_value.__aenter__ = AsyncMock(return_value=client)
|
|
mock_client_cls.return_value.__aexit__ = AsyncMock(return_value=None)
|
|
result = await agents_md.fetch_agents_md("acme", "repo", "main", token="tok")
|
|
assert result is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_fetch_agents_md_skips_oversized_file() -> None:
|
|
big = "x" * (agents_md._MAX_AGENTS_MD_BYTES + 1)
|
|
with patch("httpx.AsyncClient") as mock_client_cls:
|
|
client = MagicMock()
|
|
client.get = AsyncMock(return_value=_make_response(200, big))
|
|
mock_client_cls.return_value.__aenter__ = AsyncMock(return_value=client)
|
|
mock_client_cls.return_value.__aexit__ = AsyncMock(return_value=None)
|
|
result = await agents_md.fetch_agents_md("acme", "repo", "main", token="tok")
|
|
assert result is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_fetch_agents_md_oversized_agents_md_does_not_fall_back_to_claude_md() -> None:
|
|
big = "x" * (agents_md._MAX_AGENTS_MD_BYTES + 1)
|
|
with patch("httpx.AsyncClient") as mock_client_cls:
|
|
client = MagicMock()
|
|
client.get = AsyncMock(
|
|
side_effect=[
|
|
_make_response(200, big),
|
|
_make_response(200, "# CLAUDE.md\nrules"),
|
|
]
|
|
)
|
|
mock_client_cls.return_value.__aenter__ = AsyncMock(return_value=client)
|
|
mock_client_cls.return_value.__aexit__ = AsyncMock(return_value=None)
|
|
result = await agents_md.fetch_agents_md("acme", "repo", "main", token="tok")
|
|
assert result is None
|
|
assert client.get.await_count == 1
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_fetch_agents_md_handles_http_error() -> None:
|
|
with patch("httpx.AsyncClient") as mock_client_cls:
|
|
client = MagicMock()
|
|
client.get = AsyncMock(side_effect=httpx.HTTPError("boom"))
|
|
mock_client_cls.return_value.__aenter__ = AsyncMock(return_value=client)
|
|
mock_client_cls.return_value.__aexit__ = AsyncMock(return_value=None)
|
|
result = await agents_md.fetch_agents_md("acme", "repo", "main", token="tok")
|
|
assert result is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_fetch_agents_md_returns_none_for_missing_params() -> None:
|
|
result = await agents_md.fetch_agents_md("", "repo", "main", token="tok")
|
|
assert result is None
|
|
result = await agents_md.fetch_agents_md("acme", "", "main", token="tok")
|
|
assert result is None
|
|
result = await agents_md.fetch_agents_md("acme", "repo", "", token="tok")
|
|
assert result is None
|