Fixed Identifier Concurrency Test

Author: mckaragoz

src/CodeBeam.UltimateAuth.Core/Abstractions/Stores/InMemoryVersionedStore.cs

@@ -33,7 +33,7 @@ public Task<bool> ExistsAsync(TKey key, CancellationToken ct = default)
         return Task.FromResult(_store.ContainsKey(key));
     }
 
-    public Task AddAsync(TEntity entity, CancellationToken ct = default)
+    public virtual Task AddAsync(TEntity entity, CancellationToken ct = default)
     {
         ct.ThrowIfCancellationRequested();
 
@@ -48,7 +48,7 @@ public Task AddAsync(TEntity entity, CancellationToken ct = default)
         return Task.CompletedTask;
     }
 
-    public Task SaveAsync(TEntity entity, long expectedVersion, CancellationToken ct = default)
+    public virtual Task SaveAsync(TEntity entity, long expectedVersion, CancellationToken ct = default)
     {
         ct.ThrowIfCancellationRequested();
 

src/users/CodeBeam.UltimateAuth.Users.InMemory/Stores/InMemoryUserIdentifierStore.cs

@@ -11,6 +11,7 @@ namespace CodeBeam.UltimateAuth.Users.InMemory;
 public sealed class InMemoryUserIdentifierStore : InMemoryVersionedStore<UserIdentifier, Guid>, IUserIdentifierStore
 {
     protected override Guid GetKey(UserIdentifier entity) => entity.Id;
+    private readonly object _primaryLock = new();
 
     public Task<IdentifierExistenceResult> ExistsAsync(IdentifierExistenceQuery query, CancellationToken ct = default)
     {
@@ -100,6 +101,14 @@ protected override void BeforeAdd(UserIdentifier entity)
         }
     }
 
+    public override Task AddAsync(UserIdentifier entity, CancellationToken ct = default)
+    {
+        lock (_primaryLock)
+        {
+            return base.AddAsync(entity, ct);
+        }
+    }
+
     protected override void BeforeSave(UserIdentifier entity, UserIdentifier current, long expectedVersion)
     {
         if (!entity.IsPrimary)
@@ -117,6 +126,14 @@ protected override void BeforeSave(UserIdentifier entity, UserIdentifier current
         }
     }
 
+    public override Task SaveAsync(UserIdentifier entity, long expectedVersion, CancellationToken ct = default)
+    {
+        lock (_primaryLock)
+        {
+            return base.SaveAsync(entity, expectedVersion, ct);
+        }
+    }
+
     public Task<PagedResult<UserIdentifier>> QueryAsync(TenantKey tenant, UserIdentifierQuery query, CancellationToken ct = default)
     {
         ct.ThrowIfCancellationRequested();