From 0475635025564256e909cc75f077a6a10ccc6719 Mon Sep 17 00:00:00 2001
From: Tobias Tengler <45513122+tobias-tengler@users.noreply.github.com>
Date: Fri, 21 Feb 2025 11:42:21 +0100
Subject: [PATCH] (Hopefully) Fix flakey test (#8056)

---
 .../Configuration/TypeModuleTests.cs             | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/src/HotChocolate/Core/test/Execution.Tests/Configuration/TypeModuleTests.cs b/src/HotChocolate/Core/test/Execution.Tests/Configuration/TypeModuleTests.cs
index 9f647b90150..c96aeb97c6e 100644
--- 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);
     }