Enhance Game testing

This commit is contained in:
Cal Corum 2024-10-12 18:21:15 -05:00
parent 5fe91c0729
commit c5b24d8c8e
2 changed files with 8 additions and 3 deletions

View File

@ -4,7 +4,7 @@ from sqlmodel import Session, SQLModel, create_engine
from sqlmodel.pool import StaticPool
from typing import Literal
from in_game.gameplay_db import Game, Lineup, Team
from in_game.gameplay_models import Game, Lineup, Team
@pytest.fixture(name='session')
@ -19,7 +19,8 @@ def session_fixture():
RAW_GAMES = [
Game(away_team_id=1, home_team_id=2, channel_id=1234, season=9),
Game(away_team_id=3, home_team_id=4, channel_id=5678, season=9, active=False, is_pd=False, ranked=True, week_num=6, game_num=9, away_roster_id=69, home_roster_id=420, first_message=12345678, ai_team='home', game_type='minor-league')
Game(away_team_id=4, home_team_id=3, channel_id=5678, season=9, active=False, is_pd=True, ranked=True, week_num=6, game_num=9, away_roster_id=69, home_roster_id=420, first_message=12345678, ai_team='home', game_type='minor-league'),
Game(away_team_id=3, home_team_id=4, channel_id=5678, season=9, active=True, is_pd=True, ranked=True, week_num=6, game_num=10, away_roster_id=69, home_roster_id=420, first_message=34567890, ai_team='home', game_type='minor-league')
]

View File

@ -2,7 +2,7 @@ import datetime
from sqlmodel import Session, select
from sqlite3 import IntegrityError
from in_game.gameplay_db import Team, get_team
from in_game.gameplay_models import Team, get_team
from factory import session_fixture, new_teams_fixture, pytest
def test_create_team(session: Session, new_teams: list[Team]):
@ -33,11 +33,15 @@ def test_create_incomplete_team(session: Session, new_teams: list[Team]):
async def test_team_cache(session: Session, new_teams: list[Team]):
team_1 = new_teams[0]
team_2 = new_teams[1]
team_3 = new_teams[3]
session.add(team_1)
session.add(team_2)
session.add(team_3)
session.commit()
assert (datetime.datetime.now() - team_3.created).total_seconds() > 1209600
new_team_1 = await get_team(session, team_id=team_1.id)
new_team_2 = await get_team(session, team_id=team_2.id)
new_team_3 = await get_team(session, team_abbrev='BAL')