51Testing软件测试论坛

标题: Selenium还能这么玩：重复使用登录状态实战 [打印本页]

作者: lsekfe 时间: 2022-5-16 10:42
标题: Selenium还能这么玩：重复使用登录状态实战
　应用场景
　　自动化测试如果没有效果，还不如不用。但是在 web 自动化测试过程中，为了保证测试脚本运行的稳定性，我们经常会牺牲效率，因此业内人常诟病，web自动化还不如不做。
　　本文介绍一种简单的方法，把 selenium 启动的浏览器状态进行持久化保存，其他的测试脚本可以共用浏览器状态，在一定程度上能提升测试效率，也可以用在其他的场合上。
　　因为使用了持久化保存，所以生成浏览器状态的代码和获取浏览器状态的代码可以分离开，进行独立维护。
　　举例来说，大多数的网站操作都需要登录状态。我们可以先独立编写登录的代码，然后把登录的浏览器状态存储起来，之后我们每次想进行该网站的操作，可以直接把浏览器 session 取出来，直接操作，节省了每次都要重新登录的流程。
　　代码的逻辑非常简单，主要分为几个步骤：
　　·登录
　　· 存储状态
　　· 获取状态
　　· 重复使用
　　登录
　　登录的代码，几乎是每个会 selenium 的小伙伴闭着眼睛都能写出来的。需要注意的是最后 3 行，登录动作完成后，获取 selenium 会话的远程服务地址和 session id。
　　以后测试脚本就靠连接这一次的远程服务地址，通过 session id 得到此次的浏览器对象。注意，一定不要使用 driver.quit() 退出，否则就不能重复使用了。
[attach]138056[/attach]
持久化存储
　　得到服务地址 remote_executor 和 session id 之后，我们要通过持久化的方式把这两个数据保存起来。
　　可以用数据库，可以用 json 文件，可以用 yaml 文件，也可以是其他普通格式的文本，但一定是要可持久化的。
　　在这里我们使用 yaml 文件保存。
[attach]138057[/attach]
现在，通过 login 函数，我们可以登录，然后通过 save_session 函数保存登录后的浏览器状态。如果代码能正常执行，现在你能在项目中看到一个 session.yaml 的文件。

remote_executor, session_id = login()
　　save_session(remote_executor, session_id)

复制代码

　获取持久化的 session
　　接下来，我们要考虑：其他的测试脚本如果想要使用这个 selenium 的 session 对象，首先要从 yaml 文件中读取 session id 和远程服务地址，然后重新初始化一个 selenium 的浏览器对象，连接地址，并把 session id 设置成读取出来的 id。
[attach]138058[/attach]
重复使用 session
　　在具体使用过程中，首先通过 read_service 读取出服务地址和 session id，再通过 reuse_session 函数重新连接之前的 session。

　session_config = read_service()
　　session_id = session_config['session_id']
　　executor = session_config['remote_executor']
　　driver = reuse_session(session_id, executor)

复制代码

只要 session 没有退出，在任何时候都可以通过上面的代码连接，获取浏览器。后面的操作就和普通的 selenium 没有任何区别了。

　# script1
　　driver.get('https://petstore.octoperf.com/actions/Catalog.action')
　　# 不需要登录了
　　driver.find_element('link text', 'My Account').click()
　　# script2
　　driver.get('https://petstore.octoperf.com/actions/Catalog.action')
　　 # 不需要登录了
　　driver.find_element('xpath', "//area[@alt='Fish']").click()

复制代码

并发编程的时候如何处理
　　这种方式的扩展性还是很强的，可以套用生产-消费模型。先初始化多个具备登录态的 session, 再并发执行任务时，可以从队列中每次领取一个 session 对象消费。这里的代码后面有时间再补充，先挖个坑。

欢迎光临 51Testing软件测试论坛 (http://bbs.51testing.com/)