diff --git a/backend/src/app/core/db/sql_service.py b/backend/src/app/core/db/sql_service.py index 655250c72..9a98a3d28 100644 --- a/backend/src/app/core/db/sql_service.py +++ b/backend/src/app/core/db/sql_service.py @@ -8,7 +8,6 @@ from sqlalchemy.orm import Session, sessionmaker from sqlalchemy_utils import create_database, database_exists, drop_database -from app.core.data.orm.orm_base import ORMBase from app.core.db.import_all_orms import * # noqa: F401, F403 from app.util.singleton_meta import SingletonMeta from config import conf @@ -47,20 +46,9 @@ def __new__(cls, *args, **kwargs): def __del__(self): self.__engine.dispose() - # This method is unused and only left here for historic reference - def _create_database_and_tables(self, drop_if_exists: bool = False) -> None: - logger.info("Setting up PostgresSQL DB and tables...") - if drop_if_exists and database_exists(self.__engine.url): - logger.warning("Dropping existing DB!") - drop_database(self.__engine.url) - - self.create_database_if_not_exists() - - # create all tables from SQLAlchemy ORM Models - ORMBase.metadata.create_all(self.__engine) - logger.debug("Created Tables!") - - logger.info("Done setting up PostgresSQL DB and tables!") + def drop_database(self): + logger.warning("Dropping existing DB!") + drop_database(self.__engine.url) def create_database_if_not_exists(self): if not database_exists(self.__engine.url): diff --git a/backend/src/test/conftest.py b/backend/src/test/conftest.py index fa3c2ee3a..4946d6e49 100755 --- a/backend/src/test/conftest.py +++ b/backend/src/test/conftest.py @@ -41,6 +41,11 @@ from app.core.data.dto.user import UserCreate, UserRead +def pytest_sessionfinish(): + # Make sure the next test session starts with a clean database + SQLService().drop_database() + + # Always use the asyncio backend for async tests @pytest.fixture def anyio_backend():