From e6a6819c571e61b5259e5d66ef42aa74e69e4263 Mon Sep 17 00:00:00 2001 From: Esteban Maya Date: Fri, 1 Mar 2024 12:26:05 -0500 Subject: [PATCH] =?UTF-8?q?=E2=9C=85=20Add=20setup=20and=20teardown=20data?= =?UTF-8?q?base=20for=20tests=20(#626)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Sebastián Ramírez --- src/backend/app/tests/conftest.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/src/backend/app/tests/conftest.py b/src/backend/app/tests/conftest.py index ffd097d..74a0fb0 100644 --- a/src/backend/app/tests/conftest.py +++ b/src/backend/app/tests/conftest.py @@ -2,19 +2,27 @@ from collections.abc import Generator import pytest from fastapi.testclient import TestClient -from sqlmodel import Session +from sqlmodel import Session, delete from app.core.config import settings from app.db.engine import engine +from app.db.init_db import init_db from app.main import app +from app.models import Item, User from app.tests.utils.user import authentication_token_from_email from app.tests.utils.utils import get_superuser_token_headers -@pytest.fixture(scope="session") +@pytest.fixture(scope="session", autouse=True) def db() -> Generator: with Session(engine) as session: + init_db(session) yield session + statement = delete(Item) + session.execute(statement) + statement = delete(User) + session.execute(statement) + session.commit() @pytest.fixture(scope="module")