本文主要是介绍locust之setup/teardown，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
1. 单用户测试的setup和teardown行为，分别通过taskset类里面的on_start()和on_stop()方法实现。
on_start()：在用户建立前执行，初始化用户相关配置。
on_stop()：在用户测试结束后执行，清理恢复配置。
class MyTaskSet(TaskSet):""" 定义用户行为 """def on_start(self):print("Executing on_start ...")def on_stop(self):print("Executing on_stop ...")2.整个负载测试的setup和teardown行为，则是通过locust.events添加事件监听器来实现。
test_start：在负载测试开始前执行，可以实现整个测试的初始化操作。
test_stop：在负载测试结束后执行，可以实现测试后的配置恢复和清理工作。
@events.test_start.add_listener
def on_test_start(environment, **kwargs):""" 定义负载测试前的行为 """print("Executing on_test_start ...")print(environment.runner.state)@events.test_stop.add_listener
def on_test_stop(environment, **kwargs):""" 定义负载测试结束后的行为 """print("Executing on_test_stop ...")
3.有一个比较特别一点的，就是进程process初始化init，相当于是worker的初始化，每个负载测试可以配置多个worker，实现分布式负载测试。
如下面代码来自用户手册，实现进程监控错误率，超过阈值后自动停止测试，添加检查是在workerrunner初始化时执行。
def checker(environment):""" 检查runner的计数器 """while environment.runner.state not in [STATE_STOPPING, STATE_STOPPED, STATE_CLEANUP]:time.sleep(1)if environment.runner.stats.total.fail_ratio > 0.2:print(f"Fail ratio is {environment.runner.stats.total.fail_ratio}, exit ...")environment.runner.quit()return@events.init.add_listener
def on_init(environment, **kwargs):""" 定义进程执行前的行为 """# 只有独立master模式运行时有效if not isinstance(environment.runner, WorkerRunner):gevent.spawn(checker, environment)
这篇关于locust之setup/teardown的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
