import importlib.util
import json
from pathlib import Path

import pytest

SPEC = importlib.util.spec_from_file_location("openrouter_key_check", Path(__file__).resolve().parents[1] / "tools/openrouter_key_check.py")
probe = importlib.util.module_from_spec(SPEC)
SPEC.loader.exec_module(probe)


@pytest.mark.parametrize("status,body,result", [
    (200, b'{"data":{"label":"SYNTHETIC-KEY","usage":999}}', "authenticated"),
    (401, b'SYNTHETIC-KEY', "refused"),
    (302, b'SYNTHETIC-KEY', "refused"),
    (500, b'SYNTHETIC-KEY', "refused"),
    (200, b'bad SYNTHETIC-KEY', "check_failed"),
    (200, b'{"data":null}', "invalid_response"),
    (200, b'x' * (probe.MAX_BODY + 1), "invalid_response"),
])
def test_probe_is_one_fixed_read_and_never_returns_provider_content(status, body, result):
    calls = []
    class Connection:
        def __init__(self, host, timeout):
            assert (host, timeout) == ("openrouter.ai", 10)
        def request(self, method, path, headers):
            calls.append((method, path))
            assert headers["Authorization"] == "Bearer SYNTHETIC-KEY"
        def getresponse(self): return self
        def read(self, size):
            assert size == probe.MAX_BODY + 1
            return body[:size]
        def close(self): calls.append("closed")
    Connection.status = status
    output = probe.check("SYNTHETIC-KEY", connection_factory=Connection)
    assert output["result"] == result
    assert "SYNTHETIC-KEY" not in json.dumps(output)
    assert calls == [("GET", "/api/v1/key"), "closed"]


def test_transport_error_is_sanitized_and_connection_closed():
    closed = []
    class Broken:
        def __init__(self, *a, **kw): pass
        def request(self, *a, **kw): raise OSError("SYNTHETIC-KEY")
        def close(self): closed.append(True)
    assert probe.check("SYNTHETIC-KEY", connection_factory=Broken) == {"result": "check_failed"}
    assert closed == [True]


@pytest.mark.parametrize("key", ["", "key\r\nInjected: value", "key with spaces", "non-ascii-ä"])
def test_bad_input_never_opens_connection(key):
    def forbidden(*a, **kw): pytest.fail("must not connect")
    assert probe.check(key, connection_factory=forbidden) == {"result": "invalid_input"}