dict

bitfaster · Oct 15, 2023 · 2d5692d · 2d5692d
1 parent abe55ab
commit 2d5692d
Showing 1 changed file with 21 additions and 1 deletion.
diff --git a/BitFaster.Caching.UnitTests/Atomic/AtomicFactorySoakTests.cs b/BitFaster.Caching.UnitTests/Atomic/AtomicFactorySoakTests.cs
@@ -1,5 +1,7 @@
-using System.Linq;
+using System.Collections.Concurrent;
+using System.Linq;
 using System.Threading.Tasks;
+using BitFaster.Caching.Atomic;
 using BitFaster.Caching.Lru;
 using FluentAssertions;
 using Xunit;
@@ -31,5 +33,23 @@ await Threaded.Run(4, (r) =>
             cache.Metrics.Value.Evicted.Should().Be(0);
             counters.Sum(x => x).Should().Be(1024);
         }
+
+        [Fact]
+        public async Task WhenGetOrAddIsConcurrentValuesCreatedAtomically2()
+        {
+            var dictionary = new ConcurrentDictionary<int, AtomicFactory<int, int>>(4, 1024);
+
+            var counters = new int[4];
+
+            await Threaded.Run(4, (r) =>
+            {
+                for (int i = 0; i < 1024; i++)
+                {
+                    dictionary.GetOrAdd(i, k => { counters[r]++; return k; });
+                }
+            });
+
+            counters.Sum(x => x).Should().Be(1024);
+        }
     }
 }