locust 压测之脚本检查点

Posted 2021-02-03 jchome

检查点简而言之是用来检查脚本是否达到事先预想的结果，因为locus判断请求是否成功只是通过请求响应的状态码是否为200或2xx来进行，缺乏可靠性。因此，有时请求返回的是200，但是却并未是测试想要的结果。所以需要使用检查点来检查请求返回的数据是否正确，增加脚本的可靠性和说服力。

要获取到请求返回的数据信息，可以使用正则表达式来获取页面中想要的内容，如果是对http接口的压测，可以直接获取到接口的返回信息进行检验。可根据实际需要选择哪种方式校验脚本是否运行成功。

下面以登录业务的压测脚本为例：

登录脚本：login.py
#-*-encoding:utf-8-*-
from locust import HttpLocust,TaskSet,task
import re


def login(l):
	l.head={ #头部信息
	"Accept":"text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8",
	"Accept-Encoding":"gzip, deflate, br",
	"Accept-Language":"zh-CN,zh;q=0.9",
	"User-Agent":"Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/63.0.3239.132 Safari/537.36"
	}
	l.loginparameter={  #参数信息
	"formhash":"1f3a2bb2",
	"referer":"http://localhost:8090/discuz/",
	"loginfield":"username",
	"username":"A12345",
	"password":"123456",
	"questionid":"0",
	"answer":""
	}
	with l.client.post("/logging.php?action=login&loginsubmit=yes&floatlogin=yes&inajax=1",
		headers=l.head,data=l.loginparameter,catch_response=True) as response:
		#使用正则表达式获取返回的数据，进行检查点判断
		pattern=re.compile(r"(‘messageleft‘).innerHTML = ‘<p>(.+)?</p>")
		page_content=response.content
		# print type(page_content)
		# print page_content
		reval=pattern.search(page_content)
		print reval.group()
		if reval.group()!="(‘messageleft‘).innerHTML = ‘<p>欢迎您回来 新手上路 A12345</p>":
			response.failure(u"登录失败！")


def index(l):
	l.client.get("/")


def profile(l):
	l.client.get("/index.php")



class UserBebavior(TaskSet):
	tasks ={index:2,profile:1}
	


	def on_start(self):
		login(self)


class WebsiteUser(HttpLocust):
	task_set=UserBebavior
	min_wait=5000
	max_wait=9000



执行脚本：run_test.py
#-*-encoding:utf-8-*-
import os
os.system("locust-script.py -f e:/py/login.py --host=http://localhost:8090/test ")

 

执行run_test.py结果如下：

 

现在将检查点的判断条件改为：

将检查点判断条件改为：获取到返回的页面内容为：欢迎您回来...等的信息后，请求运行报错，

再次运行结果如下：