selenium使用cookie实现免密登录

L
前言:UI自动化测试的时候,我们经常会遇到这样子的问题,一般的登录网站在登录的时候会使用验证码(包括图片点击验证等验证手段),每次登录都需要获取验证码,但是对于验证码控件的获取,这对我们自动化测试来说也是一个难题,这时候,我们可以取浏览器cookie,跳过验证码这个步骤,简单地解决了这个问题。

一、手动获取cookie

这里,我们以简书登录为例,先手动获取简书登录的cookie,代码如下:

import requests
import time
import json
from selenium import webdriver


#手动获取cookies
driver = webdriver.Chrome()
url = 'https://www.jianshu.com/sign_in'
driver.get(url)
time.sleep(2)
driver.find_element_by_id("session_email_or_mobile_number").clear()
driver.find_element_by_id("session_password").clear()
#输入简书登录账号
driver.find_element_by_id("session_email_or_mobile_number").send_keys("******")
#输入简书登录密码
driver.find_element_by_id("session_password").send_keys("**********")
#点击登录按钮
driver.find_element_by_id("sign-in-form-submit-btn").click()
#这里设置停留15秒钟,有足够的时间让我们手动去点击图片验证
time.sleep(15)
#获取cookie
cookies = driver.get_cookies()
#cookie保存到cookies.txt文件
f1 = open("cookies.txt","w")
f1.write(json.dumps(cookies))
f1.close
print(cookies)
print(type(cookies))

二、使用cookie免密登录

import requests
import time
import json
from selenium import webdriver

driver = webdriver.Chrome()
url = 'https://www.jianshu.com/sign_in'
driver.get(url)
time.sleep(2)

#从cookies.txt文件读取cookies
f2 = open("cookies.txt")
cookies = json.loads(f2.read())
#使用cookies登录
for cook in cookies:
    driver.add_cookie(cook)
#刷新页面
driver.refresh()
time.sleep(2)
driver.quit()

至此,我们可以使用cookies进行免密登录,跳过了登录每次都需要点击图片识别验证,解决了UI自动化测试的一个难题。
以上代码笔者测试通过,感谢阅读!

你可能感兴趣的:(自动化测试,selenium)