【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，如有錯誤請大神指出！