selenium+unittest實現(xiàn)web自動化的示例代碼

更新時間：2023年03月02日 08:33:05 作者：小胖虎*

本文主要介紹了selenium+unittest實現(xiàn)web自動化的示例代碼，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

前面我們學習unittest 的目的是用它編寫 Web 自動化測試用例，所以接下來會將 unittest 與Selenium 結合起來寫一個 Web 自動化測試的例子。

我們仍以百度搜索為例，創(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ū)別是搜索的關鍵字和斷言的結果不同。在學習過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ù)為搜索關鍵字，根據(jù)接收的關鍵字執(zhí)行不同內容的搜索。這里的 baidu_search()方法不會被當作測試用例執(zhí)行，因為根據(jù) unittest 查找和執(zhí)行測試用例的規(guī)則，它只會把以“test”開頭的方法當作測試用例。

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

另外我們還發(fā)現(xià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

修改前：