import asyncio
import json

from app.electrum.client import ElectrumClient, ElectrumError


class FakeWriter:
    def __init__(self):
        self.written = b""

    def write(self, data: bytes) -> None:
        self.written += data

    async def drain(self) -> None:
        pass

    def close(self) -> None:
        pass

    async def wait_closed(self) -> None:
        pass


async def _client_with_fake_transport() -> tuple[ElectrumClient, asyncio.StreamReader, FakeWriter]:
    client = ElectrumClient("localhost", 1234)
    reader = asyncio.StreamReader()
    writer = FakeWriter()
    client._reader = reader
    client._writer = writer
    client._read_task = asyncio.create_task(client._read_loop())
    return client, reader, writer


async def test_request_resolves_on_matching_response():
    client, reader, writer = await _client_with_fake_transport()

    task = asyncio.create_task(client.request("blockchain.headers.subscribe"))
    await asyncio.sleep(0)  # let request() write the payload
    sent = json.loads(writer.written.decode())
    assert sent["method"] == "blockchain.headers.subscribe"

    reader.feed_data((json.dumps({"id": sent["id"], "result": {"height": 100}}) + "\n").encode())
    result = await task
    assert result == {"height": 100}

    client._read_task.cancel()


async def test_error_response_raises_electrum_error():
    client, reader, writer = await _client_with_fake_transport()

    task = asyncio.create_task(client.request("blockchain.transaction.broadcast", ["deadbeef"]))
    await asyncio.sleep(0)
    sent = json.loads(writer.written.decode())

    reader.feed_data((json.dumps({"id": sent["id"], "error": "bad tx"}) + "\n").encode())
    try:
        await task
        assert False, "expected ElectrumError"
    except ElectrumError:
        pass

    client._read_task.cancel()


async def test_notification_delivered_to_subscription_queue():
    client, reader, writer = await _client_with_fake_transport()
    queue = client.notifications("blockchain.scripthash.subscribe")

    push = {"method": "blockchain.scripthash.subscribe", "params": ["abcd", "newstatus"]}
    reader.feed_data((json.dumps(push) + "\n").encode())
    await asyncio.sleep(0)

    params = await asyncio.wait_for(queue.get(), timeout=1)
    assert params == ["abcd", "newstatus"]

    client._read_task.cancel()