selenium实现绕过登录

痛点：在做web自动化的时候，每一个测试类都要去进行登录，这实在是太耗费性能，想着在整个测试周期内，只用登录一次就好了，完美

解决方案：我们一般是用的是用cookie来控制登录状态，然而在selenium中有针对cookie的方法，下面做下简单实现

思路：

1、做登录操作

2、获取目前的cookies，将其存储在一个文件中

3、删除目前所有的cookie

4、在每一个测试类的类前置去拿到cookie并将其添加到driver中

下面是实现代码（以课堂派来做个列子）

import json
from time import sleep
from selenium import webdriver
import pytest



class TestLogin:
    def test_login(self):

        dir = webdriver.Chrome()
        dir.get("https://v4.ketangpai.com/")
        dir.maximize_window()
        sleep(2)
        dir.find_element_by_xpath('//a[text()="登录"]').click()
        sleep(1)
        dir.find_element_by_xpath('//input[contains(@placeholder,"手机号")]').send_keys('XXXXXX@163.com')
        dir.find_element_by_xpath('//input[contains(@placeholder,"密码")]').send_keys("XXXXXXX")
        sleep(1)

        dir.find_element_by_xpath('//a[contains(@class,"btn-btn")]').click()
        cookie = dir.get_cookies()   # 获取目前多有的cookies

        with open("cookie.txt","w+") as f:
            f.write(json.dumps(cookie))    # 将目前所有的cookies存储到文件中
        dir.delete_all_cookies()    # 删除目前内存中所有的cookies



class TestA:
    def setup_class(self):
        self.driver = webdriver.Chrome()
        self.url = "https://v4.ketangpai.com/"
        self.driver.get(self.url)   # 这里必须调用下域名，让selenium知道你是想添加哪一个域名下的cookies

        with open("./cookie.txt") as f:
            data = json.loads(f.read())
            self.driver.delete_all_cookies()
            for i in data:
                self.driver.add_cookie(i)     # 将文件中的cookie取出来添加到driver中

    def teardown_class(self):
        self.driver.close()


    def test_1(self):
        self.driver.get("https://v4.ketangpai.com/Resource/index.html")
        sleep(2)
        assert self.driver.find_element_by_xpath('//a[text()="个人备课"]').text == '个人备课'



class TestB:
    def setup_class(self):
        self.driver = webdriver.Chrome()
        self.url = "https://v4.ketangpai.com/"
        self.driver.get(self.url)    # 这里必须调用下域名，让selenium知道你是想添加哪一个域名下的cookies

        with open("./cookie.txt") as f:
            data = json.loads(f.read())
            self.driver.delete_all_cookies()
            for i in data:
                self.driver.add_cookie(i)    # 将文件中的cookie取出来添加到driver中

    def teardown_class(self):
        self.driver.close()

    def test_2(self):
        self.driver.get("https://v4.ketangpai.com/Mooc/Mooc/index.html")
        sleep(2)
        assert self.driver.find_element_by_xpath('//li[text()="我的在线课程"]').text == "我的在线课程"


if __name__ == '__main__':
    pytest.main()

 

从上面的代码执行结果我们可以发现，在做TestA和TestB的时候是没有做登录操作的，也可以实现页面的直接跳转进行测试