From e24f9833490cf9fc85482b47c70ca5a79b25c569 Mon Sep 17 00:00:00 2001
From: Sebastian Stehle <sebastian@squidex.io>
Date: Thu, 4 Apr 2019 15:36:41 +0200
Subject: [PATCH] Tests fixed.

---
 src/Squidex.Infrastructure/Log/LockingLogStore.cs             | 2 +-
 tests/Squidex.Infrastructure.Tests/InMemoryPubSubTests.cs     | 4 ++--
 .../MongoDb/MongoExtensionsTests.cs                           | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/Squidex.Infrastructure/Log/LockingLogStore.cs b/src/Squidex.Infrastructure/Log/LockingLogStore.cs
index 83829d129..a5b7e02dc 100644
--- a/src/Squidex.Infrastructure/Log/LockingLogStore.cs
+++ b/src/Squidex.Infrastructure/Log/LockingLogStore.cs
@@ -56,7 +56,7 @@ namespace Squidex.Infrastructure.Log
                     {
                         await Task.Delay(2000, cts.Token);
                     }
-                    catch (TaskCanceledException)
+                    catch (OperationCanceledException)
                     {
                         break;
                     }
diff --git a/tests/Squidex.Infrastructure.Tests/InMemoryPubSubTests.cs b/tests/Squidex.Infrastructure.Tests/InMemoryPubSubTests.cs
index 3071a1cc8..4d38962f6 100644
--- a/tests/Squidex.Infrastructure.Tests/InMemoryPubSubTests.cs
+++ b/tests/Squidex.Infrastructure.Tests/InMemoryPubSubTests.cs
@@ -81,13 +81,13 @@ namespace Squidex.Infrastructure
                 return x + x;
             }, true);
 
-            await Assert.ThrowsAsync<TaskCanceledException>(() => sut.RequestAsync<int, int>(1, TimeSpan.FromSeconds(0.5), true));
+            await Assert.ThrowsAnyAsync<OperationCanceledException>(() => sut.RequestAsync<int, int>(1, TimeSpan.FromSeconds(0.5), true));
         }
 
         [Fact]
         public async Task Should_timeout_when_nobody_responds()
         {
-            await Assert.ThrowsAsync<TaskCanceledException>(() => sut.RequestAsync<int, int>(2, TimeSpan.FromSeconds(0.5), true));
+            await Assert.ThrowsAnyAsync<OperationCanceledException>(() => sut.RequestAsync<int, int>(2, TimeSpan.FromSeconds(0.5), true));
         }
     }
 }
diff --git a/tests/Squidex.Infrastructure.Tests/MongoDb/MongoExtensionsTests.cs b/tests/Squidex.Infrastructure.Tests/MongoDb/MongoExtensionsTests.cs
index e844bee66..f18a537d0 100644
--- a/tests/Squidex.Infrastructure.Tests/MongoDb/MongoExtensionsTests.cs
+++ b/tests/Squidex.Infrastructure.Tests/MongoDb/MongoExtensionsTests.cs
@@ -141,7 +141,7 @@ namespace Squidex.Infrastructure.MongoDb
 
             var cursor = new Cursor<int>().Add(0).Add(1).Add(1).Add(2).Add(3).Add(5);
 
-            await Assert.ThrowsAsync<TaskCanceledException>(() =>
+            await Assert.ThrowsAnyAsync<OperationCanceledException>(() =>
             {
                 return cursor.ForEachPipelineAsync(x =>
                 {