Fix initial idle wait

hyperskill · Apr 21, 2022 · 55015e4 · 55015e4
1 parent 16e93d3
commit 55015e4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/hstest/testing/process_wrapper.py b/hstest/testing/process_wrapper.py
@@ -39,7 +39,7 @@ def __init__(self, *args, check_early_finish=False, register_output=True, regist
         self.output_diff_history_max = 2
 
         self.initial_idle_wait = True
-        self.initial_idle_wait_max = 10
+        self.initial_idle_wait_time = 150
 
         self.check_early_finish = check_early_finish
         self.register_output = register_output
@@ -183,6 +183,11 @@ def check_cpuload(self):
             sleep(0.01)
             self.check_alive()
 
+            if self.initial_idle_wait:
+                self.initial_idle_wait_time -= 1
+                if self.initial_idle_wait_time == 0:
+                    self.initial_idle_wait = False
+
     def check_output(self):
         output_len_prev = len(self.stdout)