locust学习总结12 - 业务流程场景

1、初始的登录脚本

import json
from locust import HttpLocust, task, TaskSet, between


# 测试类：用户行为
class UserBehavior(TaskSet):
    # 测试任务1：登录
    @task
    def test_login(self):
        # 发送首页请求给服务器
        data = json.dumps({
            "username": "xxx",
            "password": "yyy",
            "From": "WEB",
            "ryption": 1,
        })
        headers = {"Content-Type": "application/json"}
        r = self.client.post("/api/login", data=data, headers=headers)
        # print(r.content)


class WebSiteUser(HttpLocust):
    host = "https://xxx.com"
    task_set = UserBehavior  # 指定测试任务
    wait_time = between(2, 5)
    # min_wait = 2000  # 最小等待时间
    # max_wait = 5000  # 最大等待时间

2、参数化处理：读取csv文件中的username\password值

# 登录脚本参数化
import json
from locust import HttpLocust, task, TaskSet, between
import csv


# 测试类：用户行为
class UserBehavior(TaskSet):
    # 测试任务1：登录脚本参数化
    @task
    def test_login(self):
        # 读取csv中的用户名、密码
        file = open("userdata_100.csv", "r")
        table = csv.reader(file)
        for i in table:
            # 参数化登录的入参
            data = json.dumps({
                "username": f"{i[0]}",
                "password": f"{i[1]}",
                "loginFrom": "jiao",
                "crytion": 1,
            })
            headers = {"Content-Type": "application/json"}
            r = self.client.post("/api/login", data=data, headers=headers)
            # print(r.content)


class WebSiteUser(HttpLocust):
    host = "https://xxx.com"
    task_set = UserBehavior  # 指定测试任务
    wait_time = between(2, 5)
    # min_wait = 2000  # 最小等待时间
    # max_wait = 5000  # 最大等待时间