mirror of
https://github.com/283375/arcaea-offline.git
synced 2025-04-19 14:10:18 +00:00
28 lines
665 B
Python
28 lines
665 B
Python
import pytest
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import Session, sessionmaker
|
|
|
|
# region sqlalchemy fixtures
|
|
# from https://medium.com/@vittorio.camisa/agile-database-integration-tests-with-python-sqlalchemy-and-factory-boy-6824e8fe33a1
|
|
engine = create_engine("sqlite:///:memory:")
|
|
Session = sessionmaker()
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def db_conn():
|
|
connection = engine.connect()
|
|
yield connection
|
|
connection.close()
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def db_session(db_conn):
|
|
transaction = db_conn.begin()
|
|
session = Session(bind=db_conn)
|
|
yield session
|
|
session.close()
|
|
transaction.rollback()
|
|
|
|
|
|
# endregion
|