refactor: replace SQLModel with QueryModel in various models and update query methods

backend/tests/test_organizations_member_remove_api.py

@@ -28,7 +28,6 @@ class _FakeExecResult:
 @dataclass
 class _FakeSession:
     exec_results: list[Any]
-    get_results: dict[tuple[type[Any], Any], Any] = field(default_factory=dict)
 
     executed: list[Any] = field(default_factory=list)
     deleted: list[Any] = field(default_factory=list)
@@ -44,9 +43,6 @@ class _FakeSession:
             raise AssertionError("No more exec_results left for session.exec")
         return self.exec_results.pop(0)
 
-    async def get(self, model: type[Any], key: Any) -> Any:
-        return self.get_results.get((model, key))
-
     async def execute(self, statement: Any) -> None:
         self.executed.append(statement)
 
@@ -79,11 +75,11 @@ async def test_remove_org_member_deletes_member_access_and_member() -> None:
         active_organization_id=org_id,
     )
     session = _FakeSession(
-        exec_results=[_FakeExecResult(first_value=fallback_org_id)],
-        get_results={
-            (OrganizationMember, member_id): member,
-            (User, target_user_id): user,
-        },
+        exec_results=[
+            _FakeExecResult(first_value=member),
+            _FakeExecResult(first_value=user),
+            _FakeExecResult(first_value=fallback_org_id),
+        ],
     )
     ctx = SimpleNamespace(
         organization=SimpleNamespace(id=org_id),
@@ -110,10 +106,7 @@ async def test_remove_org_member_disallows_self_removal() -> None:
         user_id=user_id,
         role="member",
     )
-    session = _FakeSession(
-        exec_results=[],
-        get_results={(OrganizationMember, member.id): member},
-    )
+    session = _FakeSession(exec_results=[_FakeExecResult(first_value=member)])
     ctx = SimpleNamespace(
         organization=SimpleNamespace(id=org_id),
         member=SimpleNamespace(user_id=user_id, role="owner"),
@@ -137,10 +130,7 @@ async def test_remove_org_member_requires_owner_to_remove_owner() -> None:
         user_id=uuid4(),
         role="owner",
     )
-    session = _FakeSession(
-        exec_results=[],
-        get_results={(OrganizationMember, member.id): member},
-    )
+    session = _FakeSession(exec_results=[_FakeExecResult(first_value=member)])
     ctx = SimpleNamespace(
         organization=SimpleNamespace(id=org_id),
         member=SimpleNamespace(user_id=uuid4(), role="admin"),
@@ -165,8 +155,10 @@ async def test_remove_org_member_rejects_removing_last_owner() -> None:
         role="owner",
     )
     session = _FakeSession(
-        exec_results=[_FakeExecResult(all_values=[member.id])],
-        get_results={(OrganizationMember, member.id): member},
+        exec_results=[
+            _FakeExecResult(first_value=member),
+            _FakeExecResult(all_values=[member]),
+        ],
     )
     ctx = SimpleNamespace(
         organization=SimpleNamespace(id=org_id),

backend/tests/test_task_dependencies_integration.py

@@ -5,7 +5,7 @@ from uuid import UUID, uuid4
 import pytest
 from fastapi import HTTPException
 from sqlalchemy.ext.asyncio import AsyncEngine, create_async_engine
-from sqlmodel import SQLModel
+from sqlmodel import SQLModel, col, select
 from sqlmodel.ext.asyncio.session import AsyncSession
 
 from app.models.boards import Board
@@ -122,7 +122,7 @@ async def test_dependency_queries_and_replace_and_dependents() -> None:
         assert deps_map.get(t2, []) == []
 
         # mark t2 done, t3 not
-        task2 = await session.get(Task, t2)
+        task2 = (await session.exec(select(Task).where(col(Task.id) == t2))).first()
         assert task2 is not None
         task2.status = td.DONE_STATUS
         await session.commit()