From 0051a20365c1fa7df5fd96355e88704f67c3be4c Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Thu, 30 Jan 2025 03:21:28 +0000 Subject: [PATCH] test: use sqlite for unit tests Co-Authored-By: Chris Weaver --- backend/tests/unit/onyx/db/conftest.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/backend/tests/unit/onyx/db/conftest.py b/backend/tests/unit/onyx/db/conftest.py index 769524d968b..7bda8b8eb46 100644 --- a/backend/tests/unit/onyx/db/conftest.py +++ b/backend/tests/unit/onyx/db/conftest.py @@ -1,10 +1,18 @@ import pytest -from sqlalchemy.orm import Session +from sqlalchemy import create_engine +from sqlalchemy.orm import Session, sessionmaker -from onyx.db.engine import get_session_context_manager +from onyx.db.models import Base @pytest.fixture def db_session() -> Session: - with get_session_context_manager() as session: + engine = create_engine("sqlite:///:memory:") + Base.metadata.create_all(engine) + SessionLocal = sessionmaker(bind=engine) + session = SessionLocal() + try: yield session + finally: + session.close() + Base.metadata.drop_all(engine)