Add timeout to MockQueryFunction to detect deadlocks in tests

Jcparkyn · Apr 25, 2024 · 2c8738c · 2c8738c
1 parent 11aad02
commit 2c8738c
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/test/Phetch.Tests/MockQueryFunction.cs b/test/Phetch.Tests/MockQueryFunction.cs
@@ -2,6 +2,7 @@
 
 using System;
 using System.Collections.Generic;
+using System.Diagnostics;
 using System.Linq;
 using System.Threading.Tasks;
 using Phetch.Core;
@@ -33,6 +34,12 @@ public async Task<TResult> Query(TArg arg)
         var resultTask = Sources[_queryCount].Task;
         _queryCount++;
         Calls.Add(arg);
+
+        if (!Debugger.IsAttached)
+        {
+            // Stop early if we're not debugging and it takes a while, we've probably forgotten to set a result before awaiting.
+            return await resultTask.WaitAsync(TimeSpan.FromSeconds(2));
+        }
         return await resultTask;
     }