【Python】locust框架接口性能测试（一）

　　本人工作中主要对接口与web进行性能测试，而接口测试主要为http协议接口和webservice接口，本文主要对locust框架http接口测试先进行简单介绍。

1、测试需求

对某系统登录接口进行测试，登录前需进行身份认证（摘要认证）。

2、环境准备

pycharm

Python 3.6

Windows 7

3、基本功能实现

对该系统进行摘要认证，实现登录功能，代码如下：

 import requests
 #导入HTTPDigestAuth,实现摘要认证
 from requests.auth import HTTPDigestAuth
 url = r'http://172.20.x.xxx:xxxx/login'
 #请求头部
 header = {"APP-Version":"ios-1.1"}
 #摘要认证，post方法请求服务器
 r=requests.post(url,auth=HTTPDigestAuth('username','password'),headers = header)
 #打印服务器返回
 print("返回码："+str(r.status_code)+"\r\n"+"响应报文："+"\n"+str(r.text))

4、压力发起

结合locust框架，形成压测脚本，对该系统发起压力，具体代码如下：

 from locust import HttpLocust,TaskSet,task
 import subprocess
 import json
 from requests.auth import HTTPDigestAuth
 import requests
 #TaskSet类.
 class UserBehavior(TaskSet):
     #开始前执行
     def on_start(self):
         pass
     #等同于loadrunner事务
     @task(1)
     def login(self):
         header = {"APP-Version": "ios-1.1"}
         r = self.client.post("/login", auth=HTTPDigestAuth('username', 'password'), headers=header)
         if json.loads(r.content)["tel_phone"] == ' ':
             print("Got wrong response:"+str(r.content))

 #This is another HttpLocust class.
 class MobileUserLocust(HttpLocust):
     weight = 3
     task_set = UserBehavior
     host = 'http://1.82.238.163:3001'
     min_wait = 3000
     max_wait = 6000

进入终端，执行命令：

locust -f D:\Locust\http接口测试.py

在浏览器输入localhost:8089，输入模拟用户数，点击“Start swarming”开始测试

进入主界面，实时TPS监控效果图：

响应时间效果图：

可以导出需要的数据：

以上对http接口测试locust框架进行了简单的介绍，locust性能测试框架部署及使用详解请参考：http://www.cnblogs.com/zhang-zhi/p/7642017.html，如有错误请大神指出！