14 lines
324 B
Python
14 lines
324 B
Python
import pytest
|
|
from app.db import get_engine, make_session_factory, init_db
|
|
|
|
|
|
@pytest.fixture()
|
|
def db_session():
|
|
engine = get_engine("sqlite:///:memory:")
|
|
init_db(engine)
|
|
SessionLocal = make_session_factory(engine)
|
|
session = SessionLocal()
|
|
try:
|
|
yield session
|
|
finally:
|
|
session.close()
|