接口API用例自动转locust测试用例

做接口测试是必要的，写完接口测试用例，再写locust压测脚本，其实差异不大：

写个简单的py，把接口测试脚本转为locust压测脚本，本例只是简单的示范：

原接口校验脚本：

# -*- coding = utf-8 -*-
# ------------------------------
# @time: 2020/8/24 15:59
# @Author: drew_gg
# @File: exhibition_exhibit_info.py
# @Software: api_locust
# ------------------------------

import requests


# need to locust #
class KBH:

    host = "https://XXXXer.cn/api"
    url = "/XXXition/XXibit/info"

    @classmethod
    def get_exhibition_exhibit_info(cls):
        """
        :return:
        """

        data = {"exhibitId": 34}
        # 请求参数组装 ## r_url:固定参数
        r_url = KBH.host + KBH.url
        # 发起请求
        r = requests.get(r_url, params=data)
        if r.json()['data']['info']['id'] and r.status_code == 200:
            print("success")
        else:
            print('error')


if __name__ == "__main__":
    KBH().get_exhibition_exhibit_info()

1.代码里需要有一下标识：

（1）@File:

（2）import requests

（3）need to locust

（4）class 

（5）url

（6）@classmethod

（7）cls

（8）host

（9）r = requests.

大致这些标识

转换后的代码：

# -*- coding = utf-8 -*-
# ------------------------------
# @time: 2020/8/24 15:59
# @Author: drew_gg
# @File: locust_XXXxhibit_info.py
# @Software: api_locust
# ------------------------------

import requests
from locust import task, constant 
from locust.contrib.fasthttp import FastHttpUser


# need to locust #
class KBH(FastHttpUser):

    host = "https://XXXXer.cn/api"
    url = "/exXXion/XXXibit/info"
    wait_time = constant(1)

    @task
    def get_exhibition_exhibit_info(self):
        """
        :return:
        """

        data = {"exhibitId": 34}
        # 请求参数组装 ## r_url:固定参数
        r_url = KBH.url
        # 发起请求
        with self.client.get(r_url, params=data, timeout=1, catch_response=True) as r:
            if r.content == b"":
                r.failure("No data")
            if r.status_code != 200:
                r.failure("request error")
            print(r.json())

 

如果有多个压测方法的话，按照这个类似循环修改吧，这里只是做个简单的字符匹配与替换

# -*- coding = utf-8 -*-
# ------------------------------
# @time: 2020/8/21 11:54
# @Author: drew_gg
# @File: case_to_locust.py
# @Software: api_locust
# ------------------------------

import os

pl = os.getcwd().split('api_locust')
path_to_do = pl[0] + "api_locust\\locust_view\\kbh_api\\api\\"
path_to_end = pl[0] + "api_locust\\locust_view\\kbh_api\\locust_api\\"


def search(path, name):
    """
    遍历文档目录
    :param path:
    :param name:
    :return:
    """
    file_l = []
    for root, dirs, files in os.walk(path):
        root = str(root)
        if files:
            for i in files:
                if name in i:
                    if '__init__' not in i:
                        file_l.append(root + i)
    return file_l


fl = search(path_to_do, '.py')

for fi in fl:
    with open(fi, 'r', encoding="utf-8") as f:
        py_file = path_to_end + 'locust_' + fi.split('\\')[-1]
        f_new = open(py_file, 'w', encoding='utf-8')
        f = f.readlines()
        class_host = '&&&&&&&&@@@'
        for i in f:
            if "need to locust" in i:
                for line in f:
                    if "@File:" in line:
                        b = "# @File: " + 'locust_' + fi.split('\\')[-1] + '\n'
                        line = line.replace(line, b)
                    if "import" in line:
                        b = line + "from locust import task, constant \nfrom locust.contrib.fasthttp import FastHttpUser\n"
                        line = line.replace(line, b)
                    if "class " in line:
                        b = line.split(":")[0] + "(FastHttpUser):\n"
                        class_name = line.split('class ')[1].split(":")[0]
                        class_host = class_name + ".host + "
                        line = line.replace(line, b)
                    if 'url = "' in line:
                        b = line + "    wait_time = constant(1)\n"
                        line = line.replace(line, b)
                    if "@classmethod" in line:
                        line = line.replace(line, "    @task\n")
                    if "cls" in line:
                        b = line.split("cls")[0] + 'self' + line.split("cls")[1]
                        line = line.replace(line, b)
                    if class_host in line:
                        b = line.split("KBH.host + ")[0] + line.split("KBH.host + ")[1]
                        line = line.replace(line, b)
                    if "r = requests." in line:
                        r_d = line.split('(')[1].split(")")[0]
                        r_m = line.split('.')[1].split('(')[0]
                        if r_m == "get":
                            b = "        with self.client.get(%s, timeout=1, catch_response=True) as r:\n" % r_d
                        if r_m == "post":
                            b = "        with self.client.post(%s, timeout=1, catch_response=True) as r:\n" % r_d
                        line = line.replace(line, b)
                        f_new.write(line)
                        b = """            if r.content == b"":
                r.failure("No data")
            if r.status_code != 200:
                r.failure("request error")
            print(r.json())
"""
                        f_new.write(b)
                        break
                    f_new.write(line)
        f_new.close()

 

应该有其他更好的方式，欢迎交流

Locust QQ 群：