def test_cpu_warning(self):
_monitor_interval = runners.CPU_MONITOR_INTERVAL
runners.CPU_MONITOR_INTERVAL = 2.0
try:
class CpuLocust(Locust):
wait_time = constant(0)
class task_set(TaskSet):
@task
def cpu_task(self):
for i in range(1000000):
_ = 3 / 2
runner = LocalLocustRunner([CpuLocust], mocked_options())
self.assertFalse(runner.cpu_warning_emitted)
runner.spawn_locusts(1, wait=False)
sleep(2.5)
runner.quit()
self.assertTrue(runner.cpu_warning_emitted)
finally:
runners.CPU_MONITOR_INTERVAL = _monitor_interval
def test_cpu_warning(self):
_monitor_interval = runners.CPU_MONITOR_INTERVAL
runners.CPU_MONITOR_INTERVAL = 2.0
try:
class CpuLocust(Locust):
wait_time = constant(0.001)
@task
def cpu_task(self):
for i in range(1000000):
_ = 3 / 2
environment = Environment(
options=mocked_options(),
)
runner = LocalLocustRunner(environment, [CpuLocust])
self.assertFalse(runner.cpu_warning_emitted)
runner.spawn_locusts(1, 1, wait=False)
sleep(2.5)
runner.quit()
self.assertTrue(runner.cpu_warning_emitted)
finally:
runners.CPU_MONITOR_INTERVAL = _monitor_interval
