added timer to jobs count

MythicAgents · Feb 16, 2024 · f83314c · f83314c
1 parent 98aa74a
commit f83314c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/Payload_Type/athena/athena/agent_code/Tests/Agent.Tests/TestClasses/TestTaskManager.cs b/Payload_Type/athena/athena/agent_code/Tests/Agent.Tests/TestClasses/TestTaskManager.cs
@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using System.Diagnostics;
 using System.Linq;
 using System.Reflection.Metadata.Ecma335;
 using System.Text;
@@ -38,7 +39,11 @@ public async Task StartTaskAsync(ServerJob job)
 
         public void WaitForNumberOfJobs(int numberOfJobs)
         {
-            while(jobs.Count < numberOfJobs) { }
+            Stopwatch timer = new Stopwatch();
+            timer.Start();
+
+            while (timer.Elapsed.TotalSeconds < 30 && jobs.Count < numberOfJobs) { };
+            timer.Stop();
         }
     }
 }