open-swe/tests/test_account_link.py

"""Tests for the Slack account-link prompt."""

from __future__ import annotations

import pytest


@pytest.fixture(autouse=True)
def _jwt_secret(monkeypatch: pytest.MonkeyPatch) -> None:
    monkeypatch.setenv("DASHBOARD_JWT_SECRET", "test-secret")


def test_account_link_prompt_posts_generic_token_free_link(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    """The prompt posts a plain settings link in the thread — no per-user token."""
    import asyncio

    from agent import webapp

    monkeypatch.setenv("DASHBOARD_BASE_URL", "https://app.example.com")
    calls: dict[str, object] = {}

    async def fake_reply(channel_id, thread_ts, text):
        calls["reply"] = {"channel_id": channel_id, "thread_ts": thread_ts, "text": text}
        return True

    monkeypatch.setattr(webapp, "post_slack_thread_reply", fake_reply)

    asyncio.run(webapp._post_account_link_prompt("C1", "1.1", "U1", "d@x.com", reason="unlinked"))
    assert calls["reply"]["channel_id"] == "C1"
    assert calls["reply"]["thread_ts"] == "1.1"
    assert "https://app.example.com/my-settings" in calls["reply"]["text"]
    # No signed account-link token may appear in the public thread.
    assert "link=" not in calls["reply"]["text"]


def test_account_link_prompt_revoked_wording(monkeypatch: pytest.MonkeyPatch) -> None:
    import asyncio

    from agent import webapp

    monkeypatch.setenv("DASHBOARD_BASE_URL", "https://app.example.com")
    calls: dict[str, object] = {}

    async def fake_reply(channel_id, thread_ts, text):
        calls["text"] = text
        return True

    monkeypatch.setattr(webapp, "post_slack_thread_reply", fake_reply)

    asyncio.run(webapp._post_account_link_prompt("C1", "1.1", "U1", "d@x.com", reason="revoked"))
    assert "no longer valid" in calls["text"]
    assert "link=" not in calls["text"]


def test_account_link_prompt_skips_when_dashboard_url_unset(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    import asyncio

    from agent import webapp

    monkeypatch.delenv("DASHBOARD_BASE_URL", raising=False)
    posted = False

    async def fake_reply(channel_id, thread_ts, text):
        nonlocal posted
        posted = True
        return True

    monkeypatch.setattr(webapp, "post_slack_thread_reply", fake_reply)

    asyncio.run(webapp._post_account_link_prompt("C1", "1.1", "U1", "d@x.com", reason="unlinked"))
    assert posted is False