didier/tests/test_database/test_crud/test_dad_jokes.py

17 lines
526 B
Python

from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession
from database.crud import dad_jokes as crud
from database.schemas.relational import DadJoke
async def test_add_dad_joke(postgres: AsyncSession):
"""Test creating a new joke"""
statement = select(DadJoke)
result = (await postgres.execute(statement)).scalars().all()
assert len(result) == 0
await crud.add_dad_joke(postgres, "joke")
result = (await postgres.execute(statement)).scalars().all()
assert len(result) == 1