tests: refactor tests RegisterPostEvictionCallback

alastairtree · Sep 19, 2020 · e66522f · e66522f
1 parent 129e798
commit e66522f
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/LazyCache.UnitTests/CachingServiceMemoryCacheProviderTests.cs b/LazyCache.UnitTests/CachingServiceMemoryCacheProviderTests.cs
@@ -800,10 +800,10 @@ public void GetOrAddWithCancellationExpiryBasedOnTimerAndCallbackInTheDelegateDo
                 TestKey,
             entry =>
             {
-                entry.RegisterPostEvictionCallback((key, value, reason, state) => callbackHasFired = true);
                 return new ComplexTestObject();
             }, new MemoryCacheEntryOptions()
-                    .AddExpirationToken(expireToken));
+                    .AddExpirationToken(expireToken)
+                    .RegisterPostEvictionCallback((key, value, reason, state) => callbackHasFired = true));
             // trigger expiry
             Thread.Sleep(TimeSpan.FromMilliseconds(millisecondsCacheDuration + 50));
 
@@ -820,9 +820,10 @@ public void GetOrAddWithImmediateExpirationAndCallbackInTheDelegateDoesExpireIte
                 TestKey,
                 entry =>
                 {
-                    entry.RegisterPostEvictionCallback((key, value, reason, state) => callbackHasFired = true);
                     return new ComplexTestObject();
-                }, LazyCacheEntryOptions.WithImmediateAbsoluteExpiration(TimeSpan.FromMilliseconds(millisecondsCacheDuration)));
+                }, LazyCacheEntryOptions
+                    .WithImmediateAbsoluteExpiration(TimeSpan.FromMilliseconds(millisecondsCacheDuration))
+                    .RegisterPostEvictionCallback((key, value, reason, state) => callbackHasFired = true));
             // trigger expiry
             Thread.Sleep(TimeSpan.FromMilliseconds(millisecondsCacheDuration + 50));
 
@@ -839,9 +840,10 @@ public async Task GetOrAddAsyncWithImmediateExpirationAndCallbackInTheDelegateDo
                 TestKey,
                 entry =>
                 {
-                    entry.RegisterPostEvictionCallback((key, value, reason, state) => callbackHasFired = true);
                     return Task.FromResult(new ComplexTestObject());
-                }, LazyCacheEntryOptions.WithImmediateAbsoluteExpiration(TimeSpan.FromMilliseconds(millisecondsCacheDuration)));
+                }, LazyCacheEntryOptions
+                    .WithImmediateAbsoluteExpiration(TimeSpan.FromMilliseconds(millisecondsCacheDuration))
+                    .RegisterPostEvictionCallback((key, value, reason, state) => callbackHasFired = true));
             // trigger expiry
             Thread.Sleep(TimeSpan.FromMilliseconds(millisecondsCacheDuration + 50));