Locust性能测试4-参数关联

前言

前面【Locust性能测试2-先登录场景案例】讲了登录的案例,这种是直接传账号和密码就能登录了,有些登录的网站会复杂一点,
需要先从页面上动态获取参数,作为登录接口的请求参数,如【学信网:https://account.chsi.com.cn/passport/login】的登录接口请求参数

请求参数

需要先发个get请求,从返回的html页面中解析出需要的数据

  • lt : LT-277623-5ldGTLqQhP4foKihHUlgfKPeGGyWVI
  • execution: e1s1

备注:
lt 参数是每次打开浏览器,访问登录首页时服务端会返回一个新的数据
execution 参数是表示网站刷新次数,可以刷新下再登录,就变成 e2s1了

<input class="btn_login" name="submit" accesskey="l" value="登录" tabindex="4" type="submit" title="登录" />

<div class="account-oprate clearfix">
<a class="find-yhm" href="https://account.chsi.com.cn/account/password!rtvlgname">找回用户名</a>
<a class="find-mm" href="https://account.chsi.com.cn/account/password!retrive">找回密码</a>
<a href="https://account.chsi.com.cn/account/preregister.action?from=account-login" class="regist-btn">注册</a>
</div>
<input type="hidden" name="lt" value="LT-279621-fnisPBt0FVGNFrfWzJJqhTEyw6VkfH" />
<input type="hidden" name="execution" value="e2s1" />
<input type="hidden" name="_eventId" value="submit" />

locustfile3.py代码

前面用篇专门讲了requests实现接口的参数关联案例,这里直接转化成locust脚本就行了

# coding:utf-8
from locust import HttpLocust, TaskSet, task
from lxml import etree

class LoginDemo(TaskSet):
'''用户行为描述'''
def get_it_execution(self):
result = {}
h1 = {
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/68.0.3440.106 Safari/537.36",
}
self.client.headers.update(h1)
r = self.client.get("/passport/login", verify=False)
# 从返回html页面,解析出lt、execution
dom = etree.HTML(r.content.decode("utf-8"))
try:
result["lt"] = dom.xpath('//input[@name="lt"]')[0].get("value")
result["execution"] = dom.xpath('//input[@name="execution"]')[0].get("value")
print(result)
except:
print("lt、execution参数获取失败!")
return result

<span class="hljs-function"><span class="hljs-keyword">def</span> <span class="hljs-title">login</span><span class="hljs-params">(self, user, psw)</span>:</span>
    result = self.get_it_execution()
    loginurl = <span class="hljs-string">"/passport/login"</span>
    h2 = {
        <span class="hljs-string">"Referer"</span>: loginurl,
        <span class="hljs-string">"User-Agent"</span>: <span class="hljs-string">"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/68.0.3440.106 Safari/537.36"</span>,
        <span class="hljs-string">"Accept"</span>: <span class="hljs-string">"text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8"</span>,
        <span class="hljs-string">"Origin"</span>: <span class="hljs-string">"https://account.chsi.com.cn"</span>,
        <span class="hljs-string">"Content-Length"</span>: <span class="hljs-string">"119"</span>,
        <span class="hljs-string">"Cache-Control"</span>: <span class="hljs-string">"max-age=0"</span>,
        <span class="hljs-string">"Upgrade-Insecure-Requests"</span>: <span class="hljs-string">"1"</span>,
        <span class="hljs-string">"Content-Type"</span>: <span class="hljs-string">"application/x-www-form-urlencoded"</span>
        }
    body = {
        <span class="hljs-string">"username"</span>: user,
        <span class="hljs-string">"password"</span>: psw,
        <span class="hljs-string">"rememberMe"</span>: <span class="hljs-string">"true"</span>,
        <span class="hljs-string">"lt"</span>: result[<span class="hljs-string">"lt"</span>],
        <span class="hljs-string">"execution"</span>: result[<span class="hljs-string">"execution"</span>],
        <span class="hljs-string">"_eventId"</span>: <span class="hljs-string">"submit"</span>
    }
    self.client.headers.update(h2)
    print(self.client.headers)
    r1 = self.client.post(loginurl, data=body, verify=<span class="hljs-keyword">False</span>)
    <span class="hljs-comment"># print(r1.text)</span>

@task(1)
def test_login(self):
# 测试数据
user = "13888888888"
psw = "111111"
self.login(user, psw)

class websitUser(HttpLocust):
task_set = LoginDemo
host = "https://account.chsi.com.cn"
min_wait = 3000 # 单位毫秒
max_wait = 6000 # 单位毫秒

if name == "main":
import os
os.system("locust -f locustfile3.py")

*征得博主同意转发,转发链接https://www.cnblogs.com/yoyoketang/p/9642242.html

posted @ 2020-02-06 10:25  SunJoker  阅读(287)  评论(0编辑  收藏  举报