"""Sync shared stream controller tests."""

from __future__ import annotations

import httpx

from langgraph_sdk._sync.http import SyncHttpClient
from langgraph_sdk._sync.threads import SyncThreadsClient
from langgraph_sdk.stream.sync_controller import SyncStreamController
from langgraph_sdk.stream.transport.sync_http import SyncProtocolSseTransport
from streaming._events import values_event
from streaming._sync_fake_server import SyncFakeServer, SyncStreamScript


def test_sync_controller_fans_out_to_subscription():
    fake = SyncFakeServer()
    fake.script([values_event(seq=1, counter=1)])
    with httpx.Client(transport=fake.transport, base_url="http://test") as raw:
        transport = SyncProtocolSseTransport(client=raw, thread_id="t-1")
        controller = SyncStreamController(transport)
        sub = controller.register_subscription({"channels": ["values"]})
        controller.reconcile_stream({"channels": ["values"]})
        controller.ensure_fanout_running()

        assert sub.queue.get(timeout=1) == values_event(seq=1, counter=1)
        assert sub.queue.get(timeout=1) is None
        controller.close()


def test_sync_send_command_applied_through_seq_seeds_shared_stream_since():
    fake = SyncFakeServer()
    fake.script_sequence([SyncStreamScript(events=[]), SyncStreamScript(events=[])])
    fake.script_command_response(
        {
            "type": "success",
            "id": None,
            "result": {"run_id": "run-1"},
            "meta": {"applied_through_seq": 17},
        }
    )
    with httpx.Client(transport=fake.transport, base_url="http://test") as raw:
        threads = SyncThreadsClient(SyncHttpClient(raw))
        with threads.stream(thread_id="t-1", assistant_id="agent") as thread:
            thread.run.start(input={})
            assert list(thread.subscribe(["values"])) == []

    values_requests = [
        b for b in fake.stream_request_bodies if b.get("channels") == ["values"]
    ]
    assert len(values_requests) == 1
    assert values_requests[0]["since"] == 17