Selenium自动化测试框架中使用Cookies绕过登录-Toy模板网

这篇具有很好参考价值的文章主要介绍了Selenium自动化测试框架中使用Cookies绕过登录。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

在使用selenium测试后台时常常每个流程都需要走登录流程，这样自然比较浪费时间。如果遇到登录需要输入验证码等情况，就可能出师未捷身先死。

在Web应用中，登录状态通常是通过Cookie中对应的session id验证的。也是就是说，我们只要携带上登录后的Cookies，浏览器就会自动识别我们为登录状态。由于Selenium每次启动的浏览器是一个隔离的环境，不能直接使用本地已经保存的Cookies值，我们需要通过driver的add_cookie()方法手动添加Cookies,来绕过登录。

手工抓取Cookie并绕过登录

如图，先手工登录后，通过开发者工具-网络面板，查看任意请求中的Cookie值，经尝试分析，ECSCP_ID的值为session id，我们只需要添加此条Cookie即可。

Selenium自动化测试框架中使用Cookies绕过登录,selenium,selenium,chrome,前端

Cookie的存储格式，如下图所示，每一条包含name,value,path,secure等字段

Selenium自动化测试框架中使用Cookies绕过登录,selenium,selenium,chrome,前端

WebDriver相关的Cookies操作

get_cookies(self): 获取当前会话中当前域名所有cookies
get_cookie(self, name): 获取当前会话中当前域名指定name对应的cookie值
delete_cookie(self, name): 删除指定cookie
delete_all_cookies(self): 删除所有cookie
add_cookie(self, cookie_dict): 添加cookie

添加Cookie时需要逐条添加，cookie_dict示例：

driver.add_cookie({'name' : 'foo', 'value' : 'bar'})
driver.add_cookie({'name' : 'foo', 'value' : 'bar', 'path' : '/'})
driver.add_cookie({'name' : 'foo', 'value' : 'bar', 'path' : '/', 'secure':True})"

示例代码：

from selenium import webdriver
from time import sleep

driver = webdriver.Chrome()

admin_url = 'http://****/newecshop/admin/index.php'
cookie = dict(name='ECSCP_ID',value='7406d5a6f0c5bdfbebcc29b4e41cea9731d9e9f0')

driver.get(admin_url)
driver.add_cookie(cookie)
driver.get(admin_url)

sleep(3)
driver.quit()

注意：必须先打开一个页面（同域名页面）才能设置cookie，否则会报UnableToSetCookieException

使用Requests配合Selenium绕过登录#

手工抓取的Cookie有一定的有效期，我们可以通过发送登录接口，从响应中拿到Cookies再放入Selenium中使用。

安装requests: pip install requests
示例代码：

import requests
from selenium import webdriver
from time import sleep

def get_cookie(username, password):
    login_url = 'http://****/newecshop/admin/privilege.php'
    data = dict(username=username,password=password, act='signin')
    res = requests.post(login_url, data=data, allow_redirects=False)
    esscp_id = res.cookies.get('ECSCP_ID')
    cookie = dict(name='ECSCP_ID',value=esscp_id)
    return cookie

admin_url = 'http://****/newecshop/admin/index.php'

driver = webdriver.Chrome()
driver.get(admin_url)  
cookie = get_cookie(****, ****)
driver.add_cookie(cookie)
driver.get('http://***/newecshop/admin/order.php?act=list')

sleep(3)
driver.quit()

就可以直接跳过登录，如下图。

Selenium自动化测试框架中使用Cookies绕过登录,selenium,selenium,chrome,前端