mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-09-30 18:33: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.
124 lines
4 KiB
Python
124 lines
4 KiB
Python
from __future__ import annotations
|
|
|
|
import importlib
|
|
from typing import Any
|
|
|
|
import pytest
|
|
|
|
dispatch = importlib.import_module("agent.dispatch")
|
|
|
|
_ABSOLUTE = "https://open-swe-v3-abc.us.langgraph.app/webhooks/run-complete"
|
|
|
|
|
|
def test_is_loopback_webhook_relative() -> None:
|
|
assert dispatch._is_loopback_webhook("/webhooks/run-complete") is True
|
|
|
|
|
|
def test_is_loopback_webhook_localhost() -> None:
|
|
assert dispatch._is_loopback_webhook("http://localhost:2024/webhooks/run-complete") is True
|
|
assert dispatch._is_loopback_webhook("http://127.0.0.1:8000/webhooks/run-complete") is True
|
|
|
|
|
|
def test_is_loopback_webhook_absolute() -> None:
|
|
assert dispatch._is_loopback_webhook(_ABSOLUTE) is False
|
|
|
|
|
|
def test_resolve_no_secret_attaches_nothing() -> None:
|
|
assert dispatch._resolve_completion_webhook_url(_ABSOLUTE, None) is None
|
|
assert dispatch._resolve_completion_webhook_url(_ABSOLUTE, "") is None
|
|
|
|
|
|
def test_resolve_relative_url_degrades_to_none() -> None:
|
|
# Secret set but a loopback URL would 422 every run — attach nothing instead.
|
|
assert dispatch._resolve_completion_webhook_url("/webhooks/run-complete", "s3cret") is None
|
|
|
|
|
|
def test_resolve_localhost_url_degrades_to_none() -> None:
|
|
assert dispatch._resolve_completion_webhook_url("http://localhost/x", "s3cret") is None
|
|
|
|
|
|
def test_resolve_absolute_url_appends_token() -> None:
|
|
assert (
|
|
dispatch._resolve_completion_webhook_url(_ABSOLUTE, "s3cret") == f"{_ABSOLUTE}?token=s3cret"
|
|
)
|
|
|
|
|
|
def test_resolve_absolute_url_with_existing_query_left_as_is() -> None:
|
|
url = f"{_ABSOLUTE}?token=preset"
|
|
assert dispatch._resolve_completion_webhook_url(url, "s3cret") == url
|
|
|
|
|
|
class _FakeRuns:
|
|
def __init__(self) -> None:
|
|
self.created: list[dict[str, Any]] = []
|
|
|
|
async def create(self, thread_id: str, assistant_id: str, **kwargs: Any) -> dict[str, str]:
|
|
self.created.append({"thread_id": thread_id, "assistant_id": assistant_id, **kwargs})
|
|
return {"run_id": "run-1"}
|
|
|
|
|
|
class _FakeClient:
|
|
def __init__(self) -> None:
|
|
self.runs = _FakeRuns()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_durable_run_applies_defaults(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
client = _FakeClient()
|
|
monkeypatch.setattr(dispatch, "COMPLETION_WEBHOOK_URL", "https://app/webhooks/run-complete")
|
|
|
|
run = await dispatch.create_durable_run(
|
|
"thread-1",
|
|
"agent",
|
|
input={"messages": [{"role": "user", "content": "hi"}]},
|
|
source="test",
|
|
config={"configurable": {"thread_id": "thread-1"}, "metadata": {"kind": "test"}},
|
|
client=client,
|
|
)
|
|
|
|
assert run == {"run_id": "run-1"}
|
|
created = client.runs.created[0]
|
|
assert created["durability"] == "sync"
|
|
assert created["multitask_strategy"] == "interrupt"
|
|
assert created["if_not_exists"] == "create"
|
|
assert created["webhook"] == "https://app/webhooks/run-complete"
|
|
assert created["config"]["metadata"] == {"kind": "test"}
|
|
assert created["config"]["configurable"]["thread_id"] == "thread-1"
|
|
assert isinstance(created["config"]["configurable"]["prepare_run_id"], str)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_durable_run_no_webhook_when_none(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
client = _FakeClient()
|
|
monkeypatch.setattr(dispatch, "COMPLETION_WEBHOOK_URL", None)
|
|
|
|
await dispatch.create_durable_run(
|
|
"thread-1",
|
|
"agent",
|
|
input={"messages": []},
|
|
source="test",
|
|
client=client,
|
|
)
|
|
|
|
created = client.runs.created[0]
|
|
assert "webhook" not in created
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_dispatch_agent_run_preserves_multitask_strategy(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
client = _FakeClient()
|
|
monkeypatch.setattr(dispatch, "COMPLETION_WEBHOOK_URL", None)
|
|
|
|
await dispatch.dispatch_agent_run(
|
|
"thread-1",
|
|
"hi",
|
|
{"key": "val"},
|
|
source="test",
|
|
client=client,
|
|
multitask_strategy="reject",
|
|
)
|
|
|
|
created = client.runs.created[0]
|
|
assert created["multitask_strategy"] == "reject"
|