update 1. GTP-6 Astra, оптимизация кода и фикс
This commit is contained in:
@@ -8,6 +8,7 @@ from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import Session, sessionmaker
|
||||
|
||||
from hockey_data.models import Base
|
||||
from hockey_data.database import HockeyDatabase
|
||||
|
||||
|
||||
class LocalTestDatabase:
|
||||
@@ -29,14 +30,6 @@ class LocalTestDatabase:
|
||||
def create_all(self) -> None:
|
||||
Base.metadata.create_all(self.engine)
|
||||
|
||||
@contextmanager
|
||||
def session(self) -> Iterator[Session]:
|
||||
session = self.SessionFactory()
|
||||
try:
|
||||
yield session
|
||||
session.commit()
|
||||
except Exception:
|
||||
session.rollback()
|
||||
raise
|
||||
finally:
|
||||
session.close()
|
||||
# Exercise the same transaction nesting as production, with a local engine.
|
||||
session = HockeyDatabase.session
|
||||
atomic_session = HockeyDatabase.atomic_session
|
||||
|
||||
Reference in New Issue
Block a user