from sqlmodel import Session, select

from in_game.gameplay_models import Game, Lineup, ManagerAi, Play
from tests.factory import session_fixture
from in_game.managerai_responses import JumpResponse


def test_create_ai(session: Session):
    all_ai = session.exec(select(ManagerAi)).all()

    assert len(all_ai) == 3
    assert ManagerAi.create_ai(session) == True

    all_ai = session.exec(select(ManagerAi)).all()

    assert len(all_ai) == 3


def test_check_jump(session: Session):
    balanced_ai = session.exec(
        select(ManagerAi).where(ManagerAi.name == "Balanced")
    ).one()
    aggressive_ai = session.exec(
        select(ManagerAi).where(ManagerAi.name == "Yolo")
    ).one()

    this_game = session.get(Game, 1)
    runner = session.get(Lineup, 5)
    this_play = session.get(Play, 2)

    this_play.on_first = runner

    assert this_play.starting_outs == 1
    assert balanced_ai.check_jump(session, this_game, to_base=2) == JumpResponse(
        ai_note="- SEND **Player 4** to second if they get the jump", min_safe=16
    )
    assert aggressive_ai.check_jump(session, this_game, to_base=2) == JumpResponse(
        ai_note="- SEND **Player 4** to second if they get the jump",
        min_safe=13,
        run_if_auto_jump=True,
    )

    this_play.on_third = runner

    assert balanced_ai.check_jump(session, this_game, to_base=4) == JumpResponse(
        min_safe=8
    )
    assert aggressive_ai.check_jump(session, this_game, to_base=4) == JumpResponse(
        min_safe=5
    )


def test_tag_from_second(session: Session):
    balanced_ai = session.exec(
        select(ManagerAi).where(ManagerAi.name == "Balanced")
    ).one()
    aggressive_ai = session.exec(
        select(ManagerAi).where(ManagerAi.name == "Yolo")
    ).one()

    this_game = session.get(Game, 1)
    runner = session.get(Lineup, 5)
    this_play = session.get(Play, 2)
    this_play.on_second = runner

    assert this_play.starting_outs == 1

    balanced_resp = balanced_ai.tag_from_second(session, this_game)
    aggressive_resp = aggressive_ai.tag_from_second(session, this_game)

    assert balanced_resp.min_safe == 5
    assert aggressive_resp.min_safe == 2


def test_gb_decide_run(session: Session):
    """
    Verifies that gb_decide_run returns a min_safe threshold based on self.running
    plus an aggression modifier, with outs adjustment applied.

    With 1 out (no outs adjustment):
      - Balanced (running=5, behind_aggression=5): adjusted_running=5 → tier ≥5 → min_safe=6
      - Yolo (running=10, behind_aggression=10): adjusted_running=15 → tier ≥8 → min_safe=4
    """
    balanced_ai = session.exec(
        select(ManagerAi).where(ManagerAi.name == "Balanced")
    ).one()
    aggressive_ai = session.exec(
        select(ManagerAi).where(ManagerAi.name == "Yolo")
    ).one()

    this_game = session.get(Game, 1)
    this_play = session.get(Play, 2)

    assert this_play.starting_outs == 1

    balanced_resp = balanced_ai.gb_decide_run(session, this_game)
    aggressive_resp = aggressive_ai.gb_decide_run(session, this_game)

    assert balanced_resp.min_safe == 6
    assert aggressive_resp.min_safe == 4