Implement registration identity model

src/user_engine/adapters/local.py

@@ -18,10 +18,12 @@ from user_engine.domain import (
     Catalog,
     ExternalIdentity,
     FamilyInvitation,
+    IdentityFactor,
     Membership,
     OutboxEvent,
     ProfileScope,
     ProfileValue,
+    RegistrationSession,
     TenantAccount,
     User,
 )
@@ -48,6 +50,10 @@ class InMemoryUserEngineStore:
     bindings: dict[str, ApplicationBinding] = field(default_factory=dict)
     catalogs: dict[str, Catalog] = field(default_factory=dict)
     family_invitations: dict[str, FamilyInvitation] = field(default_factory=dict)
+    registration_sessions: dict[str, RegistrationSession] = field(
+        default_factory=dict
+    )
+    identity_factors: dict[str, IdentityFactor] = field(default_factory=dict)
     profile_values: dict[
         tuple[str, str, ProfileScope, str | None], ProfileValue
     ] = field(default_factory=dict)
@@ -142,6 +148,39 @@ class InMemoryUserEngineStore:
             if invitation.user_id == user_id
         )
 
+    def save_registration_session(self, session: RegistrationSession) -> None:
+        self.registration_sessions[session.registration_id] = session
+
+    def registration_session(
+        self, registration_id: str
+    ) -> RegistrationSession | None:
+        return self.registration_sessions.get(registration_id)
+
+    def all_registration_sessions(self) -> tuple[RegistrationSession, ...]:
+        return tuple(self.registration_sessions.values())
+
+    def save_identity_factor(self, factor: IdentityFactor) -> None:
+        self.identity_factors[factor.factor_id] = factor
+
+    def identity_factor(self, factor_id: str) -> IdentityFactor | None:
+        return self.identity_factors.get(factor_id)
+
+    def factors_for_registration(
+        self, registration_id: str
+    ) -> tuple[IdentityFactor, ...]:
+        return tuple(
+            factor
+            for factor in self.identity_factors.values()
+            if factor.registration_id == registration_id
+        )
+
+    def factors_for_user(self, user_id: str) -> tuple[IdentityFactor, ...]:
+        return tuple(
+            factor
+            for factor in self.identity_factors.values()
+            if factor.user_id == user_id
+        )
+
     def save_profile_value(self, value: ProfileValue) -> None:
         self.profile_values[
             (value.user_id, value.attribute_key, value.scope, value.scope_id)
@@ -199,6 +238,8 @@ class InMemoryUserEngineStore:
             "applications": len(self.applications),
             "catalogs": len(self.catalogs),
             "family_invitations": len(self.family_invitations),
+            "registration_sessions": len(self.registration_sessions),
+            "identity_factors": len(self.identity_factors),
             "profile_values": len(self.profile_values),
             "audit_records": len(self.audit_records),
             "pending_outbox_events": len(self.outbox_events),
@@ -215,6 +256,8 @@ class InMemoryUserEngineStore:
             "bindings": copy.deepcopy(self.bindings),
             "catalogs": copy.deepcopy(self.catalogs),
             "family_invitations": copy.deepcopy(self.family_invitations),
+            "registration_sessions": copy.deepcopy(self.registration_sessions),
+            "identity_factors": copy.deepcopy(self.identity_factors),
             "profile_values": copy.deepcopy(self.profile_values),
             "audit_records": copy.deepcopy(self.audit_records),
             "outbox_events": copy.deepcopy(self.outbox_events),
@@ -238,6 +281,10 @@ class InMemoryUserEngineStore:
         self.family_invitations = snapshot[
             "family_invitations"
         ]  # type: ignore[assignment]
+        self.registration_sessions = snapshot[
+            "registration_sessions"
+        ]  # type: ignore[assignment]
+        self.identity_factors = snapshot["identity_factors"]  # type: ignore[assignment]
         self.profile_values = snapshot["profile_values"]  # type: ignore[assignment]
         self.audit_records = [*snapshot_audit_records, *denied_audit_records]
         self.outbox_events = snapshot["outbox_events"]  # type: ignore[assignment]