refactor: update migration paths and improve database operation handling

This commit is contained in:
Abhimanyu Saharan
2026-02-09 00:51:26 +05:30
parent 8c4bcca603
commit f6bcd1ca5f
43 changed files with 1175 additions and 1445 deletions

View File

@@ -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)

View File

@@ -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)

View File

@@ -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)

View File

@@ -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)

View File

@@ -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)

View File

@@ -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)