mirror of https://github.com/stijndcl/didier
26 lines
818 B
Python
26 lines
818 B
Python
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from database.crud import users as crud
|
|
from database.schemas.relational import User
|
|
|
|
|
|
async def test_get_or_add_non_existing(postgres: AsyncSession):
|
|
"""Test get_or_add for a user that doesn't exist"""
|
|
await crud.get_or_add(postgres, 1)
|
|
statement = select(User)
|
|
res = (await postgres.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(postgres: AsyncSession):
|
|
"""Test get_or_add for a user that does exist"""
|
|
user = await crud.get_or_add(postgres, 1)
|
|
bank = user.bank
|
|
|
|
assert await crud.get_or_add(postgres, 1) == user
|
|
assert (await crud.get_or_add(postgres, 1)).bank == bank
|