From c5b24d8c8e609a9449a5e5a8e7aa10c365b7077c Mon Sep 17 00:00:00 2001 From: Cal Corum Date: Sat, 12 Oct 2024 18:21:15 -0500 Subject: [PATCH] Enhance Game testing --- tests/factory.py | 5 +++-- tests/test_gameplay_db_team.py | 6 +++++- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/tests/factory.py b/tests/factory.py index 89fc7d8..833277e 100644 --- a/tests/factory.py +++ b/tests/factory.py @@ -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') ] diff --git a/tests/test_gameplay_db_team.py b/tests/test_gameplay_db_team.py index e383b02..c137246 100644 --- a/tests/test_gameplay_db_team.py +++ b/tests/test_gameplay_db_team.py @@ -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')