paper-dynasty-discord/tests/test_gameplay_db_game.py

46 lines
1.3 KiB
Python

from sqlmodel import Session
from in_game.gameplay_db import Game, select
from factory import session_fixture, new_games_fixture
def test_create_game(session: Session, new_games: list[Game]):
game_1 = new_games[0]
game_2 = new_games[1]
session.add(game_1)
session.add(game_2)
session.commit()
assert game_1.away_team_id == 1
assert game_1.home_team_id == 2
assert game_1.channel_id == 1234
assert game_1.season == 9
assert game_1.active == True
assert game_1.is_pd == True
assert game_1.ranked == False
assert game_1.week_num == None
assert game_1.game_num == None
assert game_1.away_roster_id == None
assert game_1.home_roster_id == None
assert game_1.first_message == None
assert game_1.ai_team == None
assert game_1.game_type == None
assert game_2.active == False
assert game_2.is_pd == False
assert game_2.ranked == True
assert game_2.week_num == 6
assert game_2.game_num == 9
assert game_2.away_roster_id == 69
assert game_2.home_roster_id == 420
assert game_2.first_message == 12345678
assert game_2.ai_team == 'home'
assert game_2.game_type == 'minor-league'
def test_select_all_empty(session: Session):
games = session.exec(select(Game)).all()
assert len(games) == 0