(Hopefully) Fix flakey test (#8056)

ChilliCream · Feb 27, 2025 · 0475635 · 0475635
1 parent 06365e4
commit 0475635
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/HotChocolate/Core/test/Execution.Tests/Configuration/TypeModuleTests.cs b/src/HotChocolate/Core/test/Execution.Tests/Configuration/TypeModuleTests.cs
@@ -71,7 +71,7 @@ public async Task Ensure_Warmups_Are_Triggered_An_Appropriate_Number_Of_Times()
         // arrange
         var typeModule = new TriggerableTypeModule();
         var warmups = 0;
-        var resetEvent = new AutoResetEvent(false);
+        var warmupResetEvent = new AutoResetEvent(false);
 
         var services = new ServiceCollection();
         services
@@ -80,7 +80,7 @@ public async Task Ensure_Warmups_Are_Triggered_An_Appropriate_Number_Of_Times()
             .InitializeOnStartup(keepWarm: true, warmup: (_, _) =>
             {
                 warmups++;
-                resetEvent.Set();
+                warmupResetEvent.Set();
                 return Task.CompletedTask;
             })
             .AddQueryType(d => d.Field("foo").Resolve(""));
@@ -99,15 +99,19 @@ public async Task Ensure_Warmups_Are_Triggered_An_Appropriate_Number_Of_Times()
 
         // act
         // assert
+        warmupResetEvent.WaitOne();
+
+        Assert.Equal(1, warmups);
+        warmupResetEvent.Reset();
+
         typeModule.TriggerChange();
-        resetEvent.WaitOne();
+        warmupResetEvent.WaitOne();
 
-        // 2 since we have the initial warmup at "startup" and the one triggered above.
         Assert.Equal(2, warmups);
+        warmupResetEvent.Reset();
 
-        resetEvent.Reset();
         typeModule.TriggerChange();
-        resetEvent.WaitOne();
+        warmupResetEvent.WaitOne();
 
         Assert.Equal(3, warmups);
     }