⚗️ put a retry around a time-sensitive test

fluffynuts · Nov 14, 2023 · 2f85810 · 2f85810
1 parent a787028
commit 2f85810
Showing 1 changed file with 32 additions and 24 deletions.
diff --git a/source/TempDb/PeanutButter.TempDb.MySql.Connector.Tests/TestTempDbMySqlConnector.cs b/source/TempDb/PeanutButter.TempDb.MySql.Connector.Tests/TestTempDbMySqlConnector.cs
@@ -377,37 +377,45 @@ TimeSpan maxWait
             public void ConnectionUseShouldExtendLifetime()
             {
                 // Arrange
-                var disposed = new ConcurrentQueue<bool>();
-                using (var db = Create(inactivityTimeout: TimeSpan.FromSeconds(1)))
+                Retry.Max(3).Times(() =>
                 {
-                    db.Disposed += (o, e) => disposed.Enqueue(true);
-                    // Act
-                    for (var i = 0; i < 5; i++)
+                    var disposed = new ConcurrentQueue<bool>();
+                    using (var db = Create(inactivityTimeout: TimeSpan.FromSeconds(1)))
                     {
-                        Expect(
-                            () =>
-                            {
-                                using var conn = db.OpenConnection();
-                                Thread.Sleep(500);
-                            }
-                        ).Not.To.Throw();
-                    }
+                        db.Disposed += (o, e) => disposed.Enqueue(true);
+                        // Act
+                        for (var i = 0; i < 5; i++)
+                        {
+                            Expect(
+                                () =>
+                                {
+                                    using var conn = db.OpenConnection();
+                                    Thread.Sleep(500);
+                                }
+                            ).Not.To.Throw();
+                        }
 
-                    WaitFor(() => disposed.Any(), 10000);
+                        WaitFor(() => disposed.Any(), 10000);
 
 
-                    Expect(
-                            () =>
+                        Expect(
+                                () =>
+                                {
+                                    using var conn = db.OpenConnection();
+                                }
+                            ).To.Throw<InvalidOperationException>()
+                            .With.Message.Containing("not running");
+                    }
+
+                    // Assert
+                    Expect(disposed.ToArray())
+                        .To.Equal(
+                            new[]
                             {
-                                using var conn = db.OpenConnection();
+                                true
                             }
-                        ).To.Throw<InvalidOperationException>()
-                        .With.Message.Containing("not running");
-                }
-
-                // Assert
-                Expect(disposed.ToArray())
-                    .To.Equal(new[] { true });
+                        );
+                });
             }
 
             [Test]