"""点数预检/熔断/暂停/删除 专项测试。"""
from sqlmodel import Session, SQLModel, create_engine

from app.models import (Agent, Campaign, CampaignRun, PointAccount, Segment)
from app.points_guard import points_preflight, should_break_run


def _session():
    engine = create_engine("sqlite://", connect_args={"check_same_thread": False})
    SQLModel.metadata.create_all(engine)
    return Session(engine)


def _campaign(session, owner_id=1, org_id=1):
    camp = Campaign(name="测试任务", owner_user_id=owner_id, organization_id=org_id)
    session.add(camp)
    session.commit()
    session.refresh(camp)
    return camp


# ---------------- 启动预检 ----------------

def test_preflight_blocks_when_balance_below_estimate():
    with _session() as s:
        camp = _campaign(s)
        s.add(PointAccount(user_id=1, organization_id=1, balance_points=1000))
        s.commit()
        result = points_preflight(s, camp, estimated_points=100_000)
        assert not result["ok"]
        assert "点数不足" in result["blocker"]
        assert "20% 缓冲" in result["blocker"]
        assert result["info"]["required_points"] == 120_000


def test_preflight_passes_with_buffer():
    with _session() as s:
        camp = _campaign(s)
        s.add(PointAccount(user_id=1, organization_id=1, balance_points=130_000))
        s.commit()
        result = points_preflight(s, camp, estimated_points=100_000)
        assert result["ok"]
        assert result["info"]["required_points"] == 120_000


def test_preflight_blocks_at_exact_boundary():
    with _session() as s:
        camp = _campaign(s)
        s.add(PointAccount(user_id=1, organization_id=1, balance_points=119_999))
        s.commit()
        assert not points_preflight(s, camp, 100_000)["ok"]
        s.get(PointAccount, 1).balance_points = 120_000
        s.commit()
        assert points_preflight(s, camp, 100_000)["ok"]


# ---------------- 运行中熔断判断 ----------------

def test_break_on_balance_error():
    brk, _ = should_break_run({"type": "", "detail": "点数余额不足，至少需要 10 点数"})
    assert brk


def test_break_on_auth_error():
    brk, _ = should_break_run({"type": "billing", "detail": ""})
    assert brk
    brk2, _ = should_break_run({"type": "", "detail": "Error code: 401 - Invalid API Key"})
    assert brk2


def test_no_break_on_transient_error():
    brk, _ = should_break_run({"type": "", "detail": "timeout: 请求超时"})
    assert not brk
    brk2, _ = should_break_run(None)
    assert not brk2


# ---------------- run pause_requested 字段 ----------------

def test_run_pause_fields_exist():
    with _session() as s:
        camp = _campaign(s)
        run = CampaignRun(campaign_id=camp.id, status="running")
        s.add(run)
        s.commit()
        s.refresh(run)
        assert run.pause_requested is False
        run.pause_requested = True
        s.add(run)
        s.commit()
        s.refresh(run)
        assert run.pause_requested is True