import pytest
from sqlmodel import Session, select, func

from dice import ab_roll
from in_game.gameplay_models import Game
from tests.factory import session_fixture


def test_ab_roll(session: Session):
    game_1 = session.get(Game, 1)
    play_2 = game_1.initialize_play(session)

    this_roll = ab_roll(game_1.away_team, game_1, allow_chaos=False)

    assert this_roll.d_six_one is not None
    assert this_roll.d_six_two is not None
    assert this_roll.d_six_three is not None
    assert this_roll.d_twenty is not None
    assert this_roll.roll_message is not None