paper-dynasty-discord/tests/gameplay_models/test_card_model.py
2024-10-13 20:39:58 -05:00

31 lines
793 B
Python

import datetime
from sqlmodel import Session
from in_game.gameplay_models import CACHE_LIMIT, Card, select, get_card_or_none
from factory import session_fixture
def test_create_card(session: Session):
all_cards = session.exec(select(Card)).all()
assert len(all_cards) == 41
card_1 = session.get(Card, 1)
card_2 = session.get(Card, 12)
assert card_1.player_id == 1
assert card_1.team_id == 31
assert card_2.player_id == 12
assert card_2.team_id == 400
async def test_get_card_or_none(session: Session):
card_1 = session.get(Card, 1)
new_card_1 = await get_card_or_none(session, card_id=card_1.id)
assert card_1.created == new_card_1.created
new_card_2 = await get_card_or_none(session, card_id=55555)
assert new_card_2 is not None