【pytest框架之web自动化测试】---二维码登录的实现(登录的复用和cookies的使用)

当需要登录的网址中登录方式只包含扫描二维码时，如何用自动化实现呢？

方法1.这里需要用到复用，跳过登录阶段

比如企业微信登录时，可选方式只有一种：手机扫描二维码，验证通过后才可以进入企业微信

 

 

 

 https://work.weixin.qq.com/wework_admin/loginpage_wx

 

 

selenium中的webdriver提供了一种方法

chrome_arg=webdriver.ChromeOptions()
chrome_arg.debugger_address= '127.0.0.1:9222'
self.driver = webdriver.Chrome(options=chrome_arg)
以调试模式打开浏览器。

具体步骤如下：
1.cmd指定Chrome调试地址
chrome -remote-debugging-port=9222

 

 2.验证调试地址是否可以正常打开---注意必须能正常打开，后续pytest文件才可以正常执行

 

 

 3.手机扫描二维码并登录--这里用的是企业微信二维码

4.代码中指定以调试方式打开Chrome--注意ip地址和端口号需要和cmd中对应

# -*- coding:utf-8 -*-
# author:xjw
# date=2021/3/2
#企业微信web端登录
import json
from http import cookies

from time import sleep

from selenium import webdriver


class Testweixin():
    def setup(self):
        #self.driver=webdriver.Chrome()
        chrome_arg=webdriver.ChromeOptions()
        chrome_arg.debugger_address= '127.0.0.1:9222'
        self.driver = webdriver.Chrome(options=chrome_arg)
        self.driver.implicitly_wait(5)
    def teardown(self):
        self.driver.quit()
    # def test_weixinlogin(self):
    # 登录首页登录
    #     self.driver.get('https://work.weixin.qq.com/')
    #     self.driver.find_element_by_css_selector('[class="index_top_operation_loginBtn"]').click()
    #     self.driver.find_element_by_css_selector('[class="login_registerBar_link"]').click()
    #     self.driver.find_element_by_xpath('//*[@id="corp_name"]').send_keys('123')


    def test_after_login(self):
        # 登录后浏览器复用
        self.driver.get('https://work.weixin.qq.com/wework_admin/frame')
        self.driver.find_element_by_xpath('//*[@id="menu_contacts"]').click() #//*[@id="menu_contacts"]/span
        sleep(3)

    # def test_cookies(self):
    #     #使用cookies登录
    #     cookies = self.driver.get_cookies()
    #     # print(cookies)
    #     self.driver.get('https://work.weixin.qq.com/wework_admin/frame')
    #     # with open('tmp.txt','r',encoding='utf-8') as f:
    #     #     #f.write(json.dumps(cookies))
    #     #     raw_cookies=f.read()
    #     #     cookies=json.loads(raw_cookies)
    #     # for cookie in cookies:
    #     #     self.driver.add_cookie(cookie)
    #     self.driver.refresh()
    #     sleep(2)

 

预期结果：
执行代码后，直接跳过登录阶段进入企业微信首页并点击了通讯录。

 

方法2：使用cookies登录

步骤：

1.手机扫描二维码并登录成功

2.代码中获取登录的cookies，携带此cookies

可以先获取cookies存入本地文件，然后从本地文件读取，以下代码可以直接跳过二维码扫描操作，直接访问通讯录

# -*- coding:utf-8 -*-
# author:xjw
# date=2021/3/2
#企业微信web端登录
import json
from http import cookies

from time import sleep

from selenium import webdriver


class Testweixin():
    def setup(self):
        #self.driver=webdriver.Chrome()
        chrome_arg=webdriver.ChromeOptions()
        chrome_arg.debugger_address= '127.0.0.1:9222'
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(5)
    def teardown(self):
        self.driver.quit()
    # def test_weixinlogin(self):
    # 登录首页登录
    #     self.driver.get('https://work.weixin.qq.com/')
    #     self.driver.find_element_by_css_selector('[class="index_top_operation_loginBtn"]').click()
    #     self.driver.find_element_by_css_selector('[class="login_registerBar_link"]').click()
    #     self.driver.find_element_by_xpath('//*[@id="corp_name"]').send_keys('123')


    def test_after_login(self):
        # 登录后浏览器复用
        self.driver.get('https://work.weixin.qq.com/wework_admin/frame')
        self.driver.find_element_by_xpath('//*[@id="menu_contacts"]').click() #//*[@id="menu_contacts"]/span
        sleep(3)

    def test_cookies(self):
        #使用cookies登录

        # #存入cookies  注意此时self.driver = webdriver.Chrome(options=chrome_arg) 要带上options=chrome_arg
        # cookies = self.driver.get_cookies()
        # with open('tmp.txt', 'w', encoding='utf-8') as f:
        #     f.write(json.dumps(cookies))

        #读取cookies  注意此时self.driver = webdriver.Chrome() 要去掉options
        self.driver.get('https://work.weixin.qq.com/wework_admin/frame#contacts')
        # self.driver.find_element_by_xpath('//*[@id="menu_contacts"]').click()
        with open('tmp.txt','r',encoding='utf-8') as f:
            #f.write(json.dumps(cookies))
            raw_cookies=f.read()
            cookies=json.loads(raw_cookies)
        for cookie in cookies:
            self.driver.add_cookie(cookie)
        self.driver.refresh()
        sleep(2)