import pytest
from log_parser import strip_ts, classify_line

# ── strip_ts ────────────────────────────────────────────────────────────────

def test_strip_ts_with_prefix():
    line = "[2026-06-22 14:29:49] 0.00.044.403 I srv  init: running without SSL"
    assert strip_ts(line) == "0.00.044.403 I srv  init: running without SSL"

def test_strip_ts_without_prefix():
    line = "0.00.044.403 I srv  init: running without SSL"
    assert strip_ts(line) == "0.00.044.403 I srv  init: running without SSL"

def test_strip_ts_child_line():
    line = "[2026-06-22 14:31:59] [41803] 0.00.041.673 I log_info: verbosity = 3"
    assert strip_ts(line) == "[41803] 0.00.041.673 I log_info: verbosity = 3"

def test_strip_ts_cmd_line():
    line = '[2026-06-22 14:31:59] [41803] cmd_child_to_router:state:{"state":"loading"}'
    assert strip_ts(line) == '[41803] cmd_child_to_router:state:{"state":"loading"}'

# ── classify_line ────────────────────────────────────────────────────────────

def test_classify_router_line():
    kind, port, content = classify_line("0.00.044.403 I srv  init: running without SSL")
    assert kind == "router"
    assert port is None
    assert content == "0.00.044.403 I srv  init: running without SSL"

def test_classify_child_line():
    kind, port, content = classify_line("[41803] 0.00.041.673 I log_info: verbosity = 3")
    assert kind == "child"
    assert port == 41803
    assert content == "0.00.041.673 I log_info: verbosity = 3"

def test_classify_child_cmd_line():
    payload = '{"state":"loading","payload":null}'
    kind, port, content = classify_line(f"[41803] cmd_child_to_router:state:{payload}")
    assert kind == "child"
    assert port == 41803
    assert content == f"cmd_child_to_router:state:{payload}"

def test_classify_router_proxy_line():
    line = "2.10.286.889 I srv  proxy_reques: proxying request to model Qwen3 on port 41803"
    kind, port, content = classify_line(line)
    assert kind == "router"
    assert port is None