5 locust 参数化3 - 使用

1 脚本

# 新建任务集
from locust import TaskSet, task

from test.test_get_list_queue1 import csv_to_list
from test.test_get_params1 import get_random_param


class TestLogin(TaskSet):
    # on_start 方法在用例执行前自动执行一次
    def on_start(self):
        # 生成 list
        self.uid_list = csv_to_list('/csvfile/report-test1.csv', 'uid')

    @task
    def req_login(self):
        # 随机取值
        uid = get_random_param(self.uid_list)
        print(uid)
        data = {
            "kw": uid
        }
        login_response = self.client.post('/', data=data)

# 访问设置
class BD(HttpLocust):
    task_set = TestIndex　　# 指定测试集合，类名
    wait_time = between(1, 2)　　# 接口完成一次请求后等待的时间
    host = 'https://www.baidu.com'　　# 指定访问那个网站