使用Python编写locust压测脚本（学习中。。。。）

最新推荐文章于 2024-09-23 13:34:00 发布

原创

最新推荐文章于 2024-09-23 13:34:00 发布 · 875 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#locust #参数化 #登录 #压力测试

本文介绍如何使用Python的Locust库编写压力测试脚本，包括登录功能的测试，采用参数化从CSV文件读取用户数据，并进行任务执行。通过检查状态码和页面元素来验证登录是否成功。

# -*- coding: utf-8 -*-

from locust import HttpLocust, TaskSet, task
import queue
import pandas as pd
import os
from pyquery import PyQuery as pq

host = 'XXXX'

class WebsiteTasks(TaskSet):
    @staticmethod
    def get_html_id(html):
        d = pq(html)('input').attr('value')
        print(d)
        return d

    def getUrl(self):
        response = self.client.get('/login', allow_redirects=False)
        print('location', response.headers.get("Location"))
        url = response.headers.get("Location").replace(host, "", 1)
        print("获取响应的headers：", url)
        return url

    @task
    def login_in(self):
        try:
            data = self.locust.user_data_queue.get()  # 获取队列里的数据
            print("获取队列的数据：",data)
        except queue.Empty:
            # 队列取空后，直接退出
            print('no data exist')
            exit(0)
        payload = data
        print("测试的用户：", payload)
        with self.client.post(self.getUrl(),data, catch_response=True) as response: