mirror of https://github.com/stijndcl/didier
26 lines
862 B
Python
26 lines
862 B
Python
|
from sqlalchemy import select
|
||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
|
||
|
|
from database.crud import users as crud
|
||
|
|
from database.models import User
|
||
|
|
|
||
|
|
|
||
|
|
async def test_get_or_add_non_existing(database_session: AsyncSession):
|
||
|
|
"""Test get_or_add for a user that doesn't exist"""
|
||
|
|
await crud.get_or_add(database_session, 1)
|
||
|
|
statement = select(User)
|
||
|
|
res = (await database_session.execute(statement)).scalars().all()
|
||
|
|
|
||
|
|
assert len(res) == 1
|
||
|
|
assert res[0].bank is not None
|
||
|
|
assert res[0].nightly_data is not None
|
||
|
|
|
||
|
|
|
||
|
|
async def test_get_or_add_existing(database_session: AsyncSession):
|
||
|
|
"""Test get_or_add for a user that does exist"""
|
||
|
|
user = await crud.get_or_add(database_session, 1)
|
||
|
|
bank = user.bank
|
||
|
|
|
||
|
|
assert await crud.get_or_add(database_session, 1) == user
|
||
|
|
assert (await crud.get_or_add(database_session, 1)).bank == bank
|