2、TaskSet类详细讲解

    性能测试工具要模拟用户的业务操作，就需要通过脚本模拟用户的行为。在前面的比喻中说到，TaskSet类好比蝗虫的大脑，控制着蝗虫的具体行为。具体地，TaskSet类实现了虚拟用户所执行任务的调度算法，包括规划任务执行顺序（schedule_task）、挑选下一个任务（execute_next_task）、执行任务（execute_task）、休眠等待（wait）、中断控制（interrupt）等等。在此基础上，我们就可以在TaskSet子类中采用非常简洁的方式来描述虚拟用户的业务测试场景，对虚拟用户的所有行为（任务）进行组织和描述，并可以对不同任务的权重进行配置。在TaskSet子类中定义任务信息时，可以采取两种方式，@task装饰器和tasks属性。

采用@task装饰器定义任务信息时，描述形式如下：

class WebsiteTasks(TaskSet):
    def on_start(self):
        self.client.post("/login", { "username": "test", "password": "123456" })

    @task(2)
    def index(self):
        self.client.get("/")

    @task(1)
    def about(self):
        self.client.get("/about/")

采用tasks属性定义任务信息时，描述形式如下：

class WebsiteTasks(TaskSet):
    def on_start(self):
        self.client.post("/login", { "username": "test", "password": "123456" })

    # @task(2)
    def index(self):
        self.client.get("/")

    # @task(1)
    def about(self):
        self.client.get("/about/")

    tasks = {index: 2, about: 1}  # 与装饰器效果一致

在TaskSet子类中除了定义任务信息，还有一个是经常用到的，那就是on_start函数。这个和LoadRunner中的vuser_init功能相同，在正式执行测试前执行一次，主要用于完成一些初始化的工作。例如，当测试某个搜索功能，而该搜索功能又要求必须为登录态的时候，就可以先在on_start中进行登录操作；前面也提 到，HttpUser使用到了requests.Session，因此后续所有任务执行过程中就都具有登录态了。