【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)