from datetime import datetime, timedelta from freezegun import freeze_time from sqlalchemy.ext.asyncio import AsyncSession from database.crud import birthdays as crud from database.crud import users from database.models import User async def test_add_birthday_not_present(database_session: AsyncSession, user: User): """Test setting a user's birthday when it doesn't exist yet""" assert user.birthday is None bd_date = datetime.today().date() await crud.add_birthday(database_session, user.user_id, bd_date) await database_session.refresh(user) assert user.birthday is not None assert user.birthday.birthday == bd_date async def test_add_birthday_overwrite(database_session: AsyncSession, user: User): """Test that setting a user's birthday when it already exists overwrites it""" bd_date = datetime.today().date() await crud.add_birthday(database_session, user.user_id, bd_date) await database_session.refresh(user) assert user.birthday is not None new_bd_date = bd_date + timedelta(weeks=1) await crud.add_birthday(database_session, user.user_id, new_bd_date) await database_session.refresh(user) assert user.birthday.birthday == new_bd_date async def test_get_birthday_exists(database_session: AsyncSession, user: User): """Test getting a user's birthday when it exists""" bd_date = datetime.today().date() await crud.add_birthday(database_session, user.user_id, bd_date) await database_session.refresh(user) bd = await crud.get_birthday_for_user(database_session, user.user_id) assert bd is not None assert bd.birthday == bd_date async def test_get_birthday_not_exists(database_session: AsyncSession, user: User): """Test getting a user's birthday when it doesn't exist""" bd = await crud.get_birthday_for_user(database_session, user.user_id) assert bd is None @freeze_time("2022/07/23") async def test_get_birthdays_on_day(database_session: AsyncSession, user: User): """Test getting all birthdays on a given day""" await crud.add_birthday(database_session, user.user_id, datetime.today().replace(year=2001)) user_2 = await users.get_or_add(database_session, user.user_id + 1) await crud.add_birthday(database_session, user_2.user_id, datetime.today() + timedelta(weeks=1)) birthdays = await crud.get_birthdays_on_day(database_session, datetime.today()) assert len(birthdays) == 1 assert birthdays[0].user_id == user.user_id @freeze_time("2022/07/23") async def test_get_birthdays_none_present(database_session: AsyncSession): """Test getting all birthdays when there are none""" birthdays = await crud.get_birthdays_on_day(database_session, datetime.today()) assert len(birthdays) == 0 # Add a random birthday that is not today await crud.add_birthday(database_session, 1, datetime.today() + timedelta(days=1)) birthdays = await crud.get_birthdays_on_day(database_session, datetime.today()) assert len(birthdays) == 0