Skip to content

Commit b5ac53d

Browse files
committed
Tweak cpu warning test for reliability with new solution.
1 parent 55b274a commit b5ac53d

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

locust/test/test_runners.py

+6-5
Original file line numberDiff line numberDiff line change
@@ -161,16 +161,16 @@ def t(self):
161161

162162
def test_cpu_warning(self):
163163
_monitor_interval = runners.CPU_MONITOR_INTERVAL
164-
runners.CPU_MONITOR_INTERVAL = 2.0
164+
runners.CPU_MONITOR_INTERVAL = 0.1
165165
try:
166166

167167
class CpuUser(User):
168-
wait_time = constant(0.001)
169-
170168
@task
171169
def cpu_task(self):
172-
for i in range(1000000):
173-
_ = 3 / 2
170+
for i in range(10):
171+
for j in range(1000000):
172+
_ = 3 / 2
173+
time.sleep(0.0001) # let other greenlets run, like the cpu monitor
174174

175175
environment = Environment(user_classes=[CpuUser])
176176
environment._cpu_warning_event_triggered = False
@@ -181,6 +181,7 @@ def cpu_warning(environment, cpu_usage, **kwargs):
181181

182182
environment.events.cpu_warning.add_listener(cpu_warning)
183183
runner = LocalRunner(environment)
184+
time.sleep(0.2) # let first checks run
184185
self.assertFalse(runner.cpu_warning_emitted)
185186
runner.spawn_users({CpuUser.__name__: 1}, wait=False)
186187
sleep(2.5)

0 commit comments

Comments
 (0)