calendar/tests/quotes_fixture.py at da920035a3ff190b4ddec14571e2b7e081685630 · PythonFreeCourse/calendar · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import pytest
from sqlalchemy.orm import Session

from app.database.models import Quote
from app.internal.utils import create_model, delete_instance


def add_quote(
        session: Session, id_quote: int, text: str, author: str, is_favorite: bool) -> Quote:
    quote = create_model(
        session, Quote,
        id=id_quote,
        text=text,
        author=author,
        is_favorite=is_favorite
    )
    yield quote
    delete_instance(session, quote)


@pytest.fixture
def quote1(session: Session) -> Quote:
    yield from add_quote(
        session=session,
        id_quote=1,
        text='You have to believe in yourself.',
        author='Sun Tzu',
        is_favorite=False
    )


@pytest.fixture
def quote2(session: Session) -> Quote:
    yield from add_quote(
        session=session,
        id_quote=2,
        text='Wisdom begins in wonder.',
        author='Socrates',
        is_favorite=False
    )