"""Tests for atm.calibrate."""
from __future__ import annotations

from pathlib import Path

import pytest


def _minimal_config_data() -> dict:
    return {
        "window_title": "Test Chart",
        "dot_roi": {"x": 0, "y": 0, "w": 100, "h": 100},
        "chart_roi": {"x": 0, "y": 0, "w": 800, "h": 600},
        "colors": {
            "turquoise":   {"rgb": [0, 200, 200],   "tolerance": 30.0},
            "yellow":      {"rgb": [255, 255, 0],   "tolerance": 30.0},
            "dark_green":  {"rgb": [0, 128, 0],     "tolerance": 30.0},
            "dark_red":    {"rgb": [139, 0, 0],     "tolerance": 30.0},
            "light_green": {"rgb": [144, 238, 144], "tolerance": 30.0},
            "light_red":   {"rgb": [255, 182, 193], "tolerance": 30.0},
            "gray":        {"rgb": [128, 128, 128], "tolerance": 30.0},
        },
        "y_axis": {"p1_y": 100, "p1_price": 10000.0, "p2_y": 200, "p2_price": 9000.0},
        "canary": {
            "roi": {"x": 0, "y": 0, "w": 50, "h": 50},
            "baseline_phash": "abc123",
            "drift_threshold": 8,
        },
        "discord": {"webhook_url": "http://example.com/hook"},
        "telegram": {"bot_token": "123:abc", "chat_id": "456"},
    }


def test_write_config_and_marker(tmp_path: Path) -> None:
    from atm.calibrate import write_config
    from atm.config import Config

    config_path = write_config(_minimal_config_data(), tmp_path)

    assert config_path.exists()
    assert config_path.suffix == ".toml"

    # Must be loadable by Config.load
    cfg = Config.load(config_path)
    assert cfg.window_title == "Test Chart"
    assert cfg.y_axis.p1_price == pytest.approx(10000.0)

    # Marker must point at the filename (basename only)
    marker = tmp_path / "current.txt"
    assert marker.exists()
    assert marker.read_text(encoding="utf-8").strip() == config_path.name

    # Config.load_current should also work
    cfg2 = Config.load_current(tmp_path)
    assert cfg2.window_title == cfg.window_title


def test_import_safe() -> None:
    """Importing atm.calibrate must succeed in a headless environment (no tkinter at top-level)."""
    import importlib  # noqa: F401
    import importlib.util

    spec = importlib.util.find_spec("atm.calibrate")
    assert spec is not None, "atm.calibrate module not found"
    # Actually importing must not raise (tkinter is only used inside run_calibration)
    mod = importlib.import_module("atm.calibrate")
    assert hasattr(mod, "write_config")
    assert hasattr(mod, "run_calibration")