怎么使用selenium+unittest實現(xiàn)web自動化

今天小編給大家分享一下怎么使用selenium+unittest實現(xiàn)web自動化的相關(guān)知識點，內(nèi)容詳細，邏輯清晰，相信大部分人都還太了解這方面的知識，所以分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后有所收獲，下面我們一起來了解一下吧。

創(chuàng)建 test_baidu.py 文件

# _*_ coding:utf-8 _*_
"""
name:zhangxingzai
date:2023/2/25
"""
 
import unittest
from time import sleep
from selenium import webdriver
from selenium.webdriver.common.by import By
 
 
class TestBaidu(unittest.TestCase):
 
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.baidu_url = 'https://www.baidu.com'
 
    def test_search_key_unittest(self):
        self.driver.get(self.baidu_url)
        self.driver.find_element(By.ID, 'kw').send_keys('unittest')
        self.driver.find_element(By.ID, 'su').click()
        sleep(2)
        title = self.driver.title
        self.assertEqual(title, 'unittest_百度搜索')
 
    def test_search_key_selenium(self):
        self.driver.get(self.baidu_url)
        self.driver.find_element(By.ID, 'kw').send_keys('selenium')
        self.driver.find_element(By.ID, 'su').click()
        sleep(2)
        title = self.driver.title
        self.assertEqual(title, 'selenium_百度搜索')
 
    def tearDown(self):
        self.driver.close()
 
 
if __name__ == '__main__':
    unittest.main

對上面的代碼不做過多介紹，都是以 unittest 創(chuàng)建測試類和方法的。方法中的代碼是Selenium 腳本。不過，這里的代碼存在一些問題，我們來一一改進。

首先，觀察代碼可以發(fā)現(xiàn)，兩個測試用例中的步驟是一樣的，唯一的區(qū)別是搜索的關(guān)鍵字和斷言的結(jié)果不同。在學習過python模塊化，所以這里把操作步驟封裝成一個方法。

代碼如下：

# _*_ coding:utf-8 _*_
"""
name:zhangxingzai
date:2023/2/25
"""
 
import unittest
from time import sleep
from selenium import webdriver
from selenium.webdriver.common.by import By
 
 
class TestBaidu(unittest.TestCase):
 
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.baidu_url = 'https://www.baidu.com'
    
    # 將百度首頁的訪問和搜索過程封裝成一個baidu_search()方法
    def baidu_search(self, search_key):
        self.driver.get(self.baidu_url)
        self.driver.find_element(By.ID, 'kw').send_keys(search_key)
        self.driver.find_element(By.ID, 'su').click()
        sleep(2)
 
    def test_search_key_unittest(self):
        search_key = 'unittest'
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title, search_key + '百度搜索')
 
    def test_search_key_selenium(self):
        search_key = "selenium"
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title, search_key + "百度搜索")
    
    def tearDown(self):
        self.driver.close()
 
if __name__ == '__main__':
    unittest.main

這里將百度首頁的訪問和搜索過程封裝成一個 baidu_search()方法，并定義 search_key參數(shù)為搜索關(guān)鍵字，根據(jù)接收的關(guān)鍵字執(zhí)行不同內(nèi)容的搜索。這里的 baidu_search()方法不會被當作測試用例執(zhí)行，因為根據(jù) unittest 查找和執(zhí)行測試用例的規(guī)則，它只會把以“test”開頭的方法當作測試用例。

另一個值得討論的問題是，測試用例的斷言要不要寫在封裝的方法中？從前面的代碼可以看出，測試的斷言點是一樣的。這里更傾向于把斷言寫在每一條測試用例里面。因為很多時候就算操作步驟是一樣的，斷言點也不完全一樣。從設(shè)計的角度來看，斷言寫在每一個測試用例中也會更加清晰。

另外我們還發(fā)現(xiàn)每一條測試用例都要啟動和關(guān)閉一次瀏覽器，這是非常耗時的，那么如何減少瀏覽器的啟動和關(guān)閉次數(shù)呢？利用前面學過的setUpClass/tearDownClass可以解決這個問題。

# _*_ coding:utf-8 _*_
"""
name:zhangxingzai
date:2023/2/25
"""
 
import unittest
from time import sleep
from selenium import webdriver
from selenium.webdriver.common.by import By
 
 
class TestBaidu(unittest.TestCase):
 
    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Firefox()
        cls.baidu_url = 'https://www.baidu.com'
 
    def baidu_search(self, search_key):
        self.driver.get(self.baidu_url)
        self.driver.find_element(By.ID, 'kw').send_keys(search_key)
        self.driver.find_element(By.ID, 'su').click()
        sleep(2)
 
    def test_search_key_unittest(self):
        search_key = 'unittest'
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title, search_key + '_百度搜索')
 
    def test_search_key_selenium(self):
        search_key = 'selenium'
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title, search_key + '_百度搜索')
 
    @classmethod
    def tearDownClass(cls):
        cls.driver.close()
 
 
if __name__ == '__main__':
    unittest.main

修改前：

修改后：

可以看到修改后省去16秒之多，這里雖然我們將 driver 驅(qū)動定義為 cls.driver，但是在每個測試用例中使用時依然為self.driver。

當整個測試類中的所有測試用例都運行完成后，會調(diào)用 cls.driver.quit()關(guān)閉瀏覽器。當一個測試類中有多條測試用例時，這種方式將會大大縮短測試用例的執(zhí)行時間。

以上就是“怎么使用selenium+unittest實現(xiàn)web自動化”這篇文章的所有內(nèi)容，感謝各位的閱讀！相信大家閱讀完這篇文章都有很大的收獲，小編每天都會為大家更新不同的知識，如果還想學習更多的知識，請關(guān)注億速云行業(yè)資訊頻道。