Locust 关联
关联的概念
用户登入后,服务器通常会给用户返回一个sessionID, 用户每次登陆,服务器返回的都会不同。
那么在自动化测试中,让系统自动登入账号就会被中断。那么我们可以通过取得服务器每次返回的session来让账号登入成功。这个过程就叫做关联
目的
在用过Loadrunner后,我们同样可以把其中测试的概念用Locust实现。
Loadrunner中,关联是通过web_reg_save_param函数,通过类似正则表达式查找到session,并通过修改,实现每次连接都会从服务器取到相应的session。
本篇将实现Locust中达到Loadrunner关联相同效果
lxml 库 和 xpath 定位
通过lxml库的etree和xpath方法,我们可以定位html中的任何元素。
自动化测试selenium和分布式爬虫对于html页面的操作通常也可以用xpath来定位元素,so lxml库的应用场景可以放在很多地方,此处我们将用它取得我们得session value。
安装lxml:
pip install lxml
Quick start:
from lxml import etree doc = ''' <div> <ul> <li class="item-0"><a href="link1.html">first item</a></li> <li class="item-1"><a href="link2.html">second item</a></li> </ul> </div> ''' tree = etree.HTML(doc) res = tree.xpath('路径定位') print(res)
Quickstart中,实例化tree后通过xpath方法,定位元素路径,并获得他的值。
外部获取html文件可以用parse方法:
from lxml import etree tree = etree.parse('html文件名') res = tree.xpath('路径定位') print(res)
xpath中路径表达式如下表:
表达式 | 描述 |
---|---|
nodename | 选取此节点的所有子节点。 |
/ | 从根节点选取。 |
// | 从匹配选择的当前节点选择文档中的节点,而不考虑它们的位置。 |
. | 选取当前节点。 |
.. | 选取当前节点的父节点。 |
@ | 选取属性。 |
样例:
取class为“item-0”:
from lxml import etree tree = etree.parse('文件名') res = tree.xpath('//li[@class="item-0"]') print(res)
输出结果:
[<Element li at 0x28ba323fb48>]
取class为“item-0”下,href为“link1.html”:
from lxml import etree tree = etree.parse('文件名') res = tree.xpath('//li[@class="item-0"]/a[@href="link1.html"]') print(res)
输出结果:
[<Element a at 0x1f8f7840b48>]
关联应用
回到关联得使用,通过lxml取得html页面的session
from lxml import etree from locust import TaskSet, task, HttpLocust
class UserBehavior(TaskSet): @staticmethod def get_session(html): tree = etree.HTML(html) return tree.xpath("//div[@class='xxx']/input[@xxx='session']/@value")[0] @task(10) def test_login(self): html = self.client.get('/login').text username = 'xxx' password = 'xxx' session = self.get_session(html) datas = { 'username': username, 'password': password, 'session': session } self.client.post('/login', data=datas)