feat: use InMemoryDb via lib

Author: emptybutton

pyproject.toml

@@ -19,6 +19,7 @@ dependencies = [
     "sqlalchemy==2.0.38",
     "psycopg[binary]==3.2.5",
     "alembic==1.14.1",
+    "in-memory-db==0.2.0",
 ]
 
 [project.optional-dependencies]

src/app_name_snake_case/infrastructure/adapters/transaction.py

@@ -1,13 +1,9 @@
 from collections.abc import AsyncIterator, Sequence
 from contextlib import asynccontextmanager
-from typing import Any
 
+from in_memory_db import InMemoryDb
 from sqlalchemy.ext.asyncio import AsyncSession
 
-from app_name_snake_case.infrastructure.in_memory_storage import (
-    TransactionalInMemoryStorage,
-)
-
 
 @asynccontextmanager
 async def in_postgres_transaction(session: AsyncSession) -> AsyncIterator[None]:
@@ -17,17 +13,17 @@ async def in_postgres_transaction(session: AsyncSession) -> AsyncIterator[None]:
 
 @asynccontextmanager
 async def in_memory_transaction(
-    storages: Sequence[TransactionalInMemoryStorage[Any]]
+    dbs: Sequence[InMemoryDb]
 ) -> AsyncIterator[None]:
-    for storage in storages:
-        storage.begin()
+    for db in dbs:
+        db.begin()
 
     try:
         yield
     except Exception as error:
-        for storage in storages:
-            storage.rollback()
+        for db in dbs:
+            db.rollback()
         raise error from error
     else:
-        for storage in storages:
-            storage.commit()
+        for db in dbs:
+            db.commit()

src/app_name_snake_case/infrastructure/adapters/users.py

@@ -1,23 +1,23 @@
 from dataclasses import dataclass
 from uuid import UUID
 
+from in_memory_db import InMemoryDb
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from app_name_snake_case.application.ports.users import Users
 from app_name_snake_case.entities.core.user import User
-from app_name_snake_case.infrastructure.in_memory_storage import (
-    TransactionalInMemoryStorage,
-)
 from app_name_snake_case.infrastructure.sqlalchemy import orm  # noqa: F401
 
 
-@dataclass(kw_only=True, slots=True)
-class InMemoryUsers(Users, TransactionalInMemoryStorage[User]):
+@dataclass(frozen=True)
+class InMemoryUsers(Users):
+    db: InMemoryDb
+
     async def user_with_id(self, id: UUID) -> User | None:
-        return self.select_one(lambda user: user.id == id)
+        return self.db.subset(User).select_one(lambda user: user.id == id)
 
 
-@dataclass(kw_only=True, frozen=True, slots=True)
+@dataclass(frozen=True)
 class InPostgresUsers(Users):
     session: AsyncSession