Update DatabaseLevelTests.cs

Fix up exception type for cancelled requests
HicServices · Jun 6, 2024 · 885bce7 · 885bce7
1 parent 83ca4f5
commit 885bce7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Tests/FAnsiTests/Database/DatabaseLevelTests.cs b/Tests/FAnsiTests/Database/DatabaseLevelTests.cs
@@ -57,9 +57,12 @@ public void Test_CreateSchema(DatabaseType type)
     public void TestListDatabasesAsync(DatabaseType type)
     {
         var db = GetTestDatabase(type, false);
+
+        var exceptionType = type == DatabaseType.MySql ? Throws.TypeOf<OperationCanceledException>() : Throws.TypeOf<TaskCanceledException>();
+
         Assert.That(
             () => db.Server.Helper.ListDatabasesAsync(db.Server.Builder, new CancellationToken(true))
-                .ToBlockingEnumerable().ToList(), Throws.TypeOf<TaskCanceledException>());
+                .ToBlockingEnumerable().ToList(), exceptionType);
         var databases = db.Server.Helper.ListDatabasesAsync(db.Server.Builder, CancellationToken.None).ToBlockingEnumerable().ToList();
         Assert.That(databases, Has.Member(db.GetRuntimeName()));
     }