refactor: update migration paths and improve database operation handling
This commit is contained in:
@@ -15,6 +15,9 @@ class _FakeSession:
|
||||
executed: list[Any] = field(default_factory=list)
|
||||
committed: int = 0
|
||||
|
||||
async def exec(self, statement: Any) -> None:
|
||||
self.executed.append(statement)
|
||||
|
||||
async def execute(self, statement: Any) -> None:
|
||||
self.executed.append(statement)
|
||||
|
||||
|
||||
@@ -17,7 +17,11 @@ class _FakeSession:
|
||||
deleted: list[Any] = field(default_factory=list)
|
||||
committed: int = 0
|
||||
|
||||
async def exec(self, _statement: Any) -> Any:
|
||||
async def exec(self, statement: Any) -> Any:
|
||||
is_dml = statement.__class__.__name__ in {"Delete", "Update", "Insert"}
|
||||
if is_dml:
|
||||
self.executed.append(statement)
|
||||
return None
|
||||
if not self.exec_results:
|
||||
raise AssertionError("No more exec_results left for session.exec")
|
||||
return self.exec_results.pop(0)
|
||||
|
||||
@@ -16,6 +16,9 @@ class _FakeSession:
|
||||
executed: list[Any] = field(default_factory=list)
|
||||
committed: int = 0
|
||||
|
||||
async def exec(self, statement: Any) -> None:
|
||||
self.executed.append(statement)
|
||||
|
||||
async def execute(self, statement: Any) -> None:
|
||||
self.executed.append(statement)
|
||||
|
||||
|
||||
@@ -36,6 +36,10 @@ class _FakeSession:
|
||||
committed: int = 0
|
||||
|
||||
async def exec(self, _statement: Any) -> Any:
|
||||
is_dml = _statement.__class__.__name__ in {"Delete", "Update", "Insert"}
|
||||
if is_dml:
|
||||
self.executed.append(_statement)
|
||||
return None
|
||||
if not self.exec_results:
|
||||
raise AssertionError("No more exec_results left for session.exec")
|
||||
return self.exec_results.pop(0)
|
||||
|
||||
@@ -51,6 +51,10 @@ class _FakeSession:
|
||||
refreshed: list[Any] = field(default_factory=list)
|
||||
|
||||
async def exec(self, _statement: Any) -> Any:
|
||||
is_dml = _statement.__class__.__name__ in {"Delete", "Update", "Insert"}
|
||||
if is_dml:
|
||||
self.executed.append(_statement)
|
||||
return None
|
||||
if not self.exec_results:
|
||||
raise AssertionError("No more exec_results left for session.exec")
|
||||
return self.exec_results.pop(0)
|
||||
|
||||
@@ -68,6 +68,10 @@ class _FakeSession:
|
||||
added: list[object] = field(default_factory=list)
|
||||
|
||||
async def exec(self, _query):
|
||||
is_dml = _query.__class__.__name__ in {"Delete", "Update", "Insert"}
|
||||
if is_dml:
|
||||
self.executed.append(_query)
|
||||
return None
|
||||
if not self.exec_results:
|
||||
raise AssertionError("No more exec_results left for session.exec")
|
||||
return self.exec_results.pop(0)
|
||||
|
||||
Reference in New Issue
Block a user