Skip to content

Commit

Permalink
test: use sqlite for unit tests
Browse files Browse the repository at this point in the history
Co-Authored-By: Chris Weaver <chris@onyx.app>
  • Loading branch information
devin-ai-integration[bot] and Chris Weaver committed Jan 30, 2025
1 parent 4675859 commit 0051a20
Showing 1 changed file with 11 additions and 3 deletions.
14 changes: 11 additions & 3 deletions backend/tests/unit/onyx/db/conftest.py
Original file line number Diff line number Diff line change
@@ -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)

0 comments on commit 0051a20

Please sign in to comment.