import pytest

from engine.harness import calibrate_cprofile_overhead, collect_environment, load_source


def test_calibrate_returns_well_formed_calibration():
    """校准应该返回一个 well-formed Calibration:ratio > 0,两段耗时 > 0,workloadName 已知。

    不强求 ratio > 1.0:在极快的机器上校准本身 < 5ms,会走 fallback ratio=1.0;
    此时 instrumentedWorkloadSec 可能比 clean 略小(perf_counter 在 1ms 量级有 noise),
    也不强求 instrumented >= clean。这条测试只锁住「返回结构对 + 数值非负」。
    """
    calib = calibrate_cprofile_overhead()
    assert calib.ratio > 0, f"ratio 必须正:{calib.ratio}"
    assert calib.ratio <= 10.0, f"ratio 太大,可能校准失效:{calib.ratio}"
    assert calib.cleanWorkloadSec > 0
    assert calib.instrumentedWorkloadSec > 0
    assert calib.workloadName == "tight-loop"


def test_calibrate_fallback_when_workload_too_fast(monkeypatch):
    """校准本身耗时 < 5ms 时应回退 ratio=1.0 —— 见 calibrate_cprofile_overhead 的边界处理。

    直接 mock calibrate_cprofile_overhead 内部的 _exec_once_* 行为不可行(它们是模块内 helper),
    改成 monkeypatch time.perf_counter 返回极短耗时:第一次返回 t0=0.0,第二次返回 t0+1e-6。
    """
    from engine import harness

    counter = {"n": 0}
    base = {"t": 1000.0}

    def fake_perf_counter():
        counter["n"] += 1
        # 第一次调用返回 0,之后每次 + 极小 delta,让 elapsed = ~3us (远小于 5ms)
        idx = counter["n"]
        return base["t"] + idx * 1e-6

    monkeypatch.setattr(harness.time, "perf_counter", fake_perf_counter)
    calib = harness.calibrate_cprofile_overhead()
    assert calib.ratio == 1.0, f"短耗时下应回退 ratio=1.0,实际:{calib.ratio}"


def test_calibrate_does_not_leak_files(monkeypatch, tmp_path):
    """校准自己 scratch 一个临时文件,跑完应清干净 —— 不污染系统临时目录。"""
    # 让 mkstemp 落到 tmp_path 方便断言
    import tempfile
    monkeypatch.setattr(tempfile, "gettempdir", lambda: str(tmp_path))
    before = set(tmp_path.iterdir())
    calibrate_cprofile_overhead()
    after = set(tmp_path.iterdir())
    # 不应有残留 —— 之前 / 之后 iterdir 集合应相等
    assert after == before, f"残留文件: {after - before}"


def test_collect_environment():
    env = collect_environment()
    assert env.python.count(".") >= 2
    assert env.timerResolution > 0


def test_load_source_strips_bom(tmp_path):
    """load_source 必须剥掉 BOM —— 引擎内所有读源码的地方都依赖这一点。"""
    script = tmp_path / "bom.py"
    script.write_text("x = 1\n", encoding="utf-8-sig")
    assert script.read_bytes().startswith(b"\xef\xbb\xbf")
    src = load_source(str(script))
    assert not src.startswith("")
    # 剥离后必须能直接过 compile()，这正是 runner 语法预检做的事
    compile(src, str(script), "exec")


def test_load_source_raises_on_binary(tmp_path):
    """二进制文件抛 UnicodeDecodeError（而不是 OSError）—— runner 依赖这个类型来归类错误。"""
    script = tmp_path / "bin.py"
    script.write_bytes(b"\xff\xfe\x00\x01\x80\x81")
    with pytest.raises(UnicodeDecodeError):
        load_source(str(script))