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)