selenium~使用unittest測試框架,批次執行測試指令碼,addTest、makeSuite、TestLoader、discover的使用

unittest框架解析

• unittest 是python 的單元測試框架， 在python 的官方檔案中，對unittest有詳細的介紹，想更深一步研究的同學可以到https://www.python.org/doc/ 去了解。
• unittest 單元測試提供了建立測試用例，測試套件以及批次執行的方案， unittest 在安裝pyhton 以後就直接自帶了，直接import unittest 就可以使用。
• 作為單元測試的框架， unittest 也是可以對程式最小模組的一種敏捷化的測試。在自動化測試中，我們雖然不需要做白盒測試，但是必須需要知道所使用語言的單元測試框架。
• 利用單元測試框架，建立一個類，該類繼承unittest的TestCase，這樣可以把每個case看成是一個最小的單元， 由測試容器組織起來，到時候直接執行，同時引入測試報告.

準備工作

• 敲好來個測試類, 繼承unittest.TestCase
• test1.py檔案

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
import unittest
import time
import re


class TestBaiDu1(unittest.TestCase):

    # setUp用來初始化環境 每執行一個測試方法都會先執行這個方法
    # self表示這個類的範例
    def setUp(self) -> None:
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(30)
        self.base_url = "http://www.baidu.com"

    # 測試用例必須用test開頭
    def test_baiDuSearch(self):
        driver = self.driver
        driver.get(self.base_url + "/")
        driver.find_element_by_id("kw").click()
        driver.find_element_by_id("kw").clear()
        # u表示字元編碼
        driver.find_element_by_id("kw").send_keys(u"測試")
        driver.find_element_by_id("su").click()
        driver.find_element_by_id("su").click()

    def test_hao(self):
        driver = self.driver
        driver.get(self.base_url + "/")
        driver.find_element_by_link_text("hao123").click()

    # 清除環境工作 每執行完一個測試用例都會執行這個
    def tearDown(self) -> None:
        time.sleep(3)
        self.driver.close()


if __name__ == "__main__":
    # 執行測試用例
    unittest.main()

• test2.py檔案

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
import unittest, time, re


class TestBaiDu2(unittest.TestCase):

    def setUp(self) -> None:
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(30)
        self.base_url = "http://www.baidu.com"

    def test_baiDuSearch(self):
        driver = self.driver
        driver.get(self.base_url + "/")
        driver.find_element_by_id("kw").click()
        driver.find_element_by_id("kw").clear()
        driver.find_element_by_id("kw").send_keys(u"詹姆斯")
        driver.find_element_by_id("su").click()
        driver.find_element_by_id("su").click()

    def tearDown(self) -> None:
        time.sleep(3)
        self.driver.close()


if __name__ == "__main__":
    unittest.main()

addTest（） 的應用

• 當有多個或者幾百測試用例的時候， 這樣就需要一個測試容器( 測試套件) ，把測試用例放在該容器中進行執行，unittest 模組中提供了TestSuite 類來生成測試套件，使用該類別建構函式可以生成一個測試套件的範例，該類提供了addTest來把每個測試用例加入到測試套件中。
• AddTest.py檔案

import unittest, csv
import test1, test2
import os, sys
import time


# 手工新增測試用例到套件
def createSuite():
    # 初始化一個測試套件
    suite = unittest.TestSuite()
    # 將測試用例新增到測試套件中
    suite.addTest(test1.TestBaiDu1("test_baiDuSearch"))
    suite.addTest(test1.TestBaiDu1("test_hao"))
    suite.addTest(test2.TestBaiDu2("test_baiDuSearch"))
    return suite


if __name__ == "__main__":
    suite = createSuite()
    # 2表示顯示詳細資訊
    runner = unittest.TextTestRunner(verbosity=2)
    # 執行測試套件
    runner.run(suite)

makeSuite（）和TestLoader()的應用

• 上述addTest有一個不方便的就是如果一個類有十幾個甚至幾十個測試用例, 如果每一次多add一次的話就太麻煩了, 所以就有makeSuite（）和TestLoader()他們會自動載入你輸入的類中的所有此時用例

makeSuite（）

• 在unittest 框架中提供了makeSuite() 的方法，makeSuite可以實現把測試用例類內所有的測試case組成的測試套件TestSuite ，unittest 呼叫makeSuite的時候，只需要把測試類名稱傳入即可。
• makeSuitTest.py檔案

import unittest, csv
import os, sys
import time
import test1
import test2


# 手工新增案例到套件，
def createSuite():
    suite = unittest.TestSuite()
    # 將測試用例加入到測試容器（套件）中
    suite.addTest(unittest.makeSuite(test1.TestBaiDu1))
    suite.addTest(unittest.makeSuite(test2.TestBaiDu2))
    return suite


if __name__ == "__main__":
    suite = createSuite()
    runner = unittest.TextTestRunner(verbosity=2)
    runner.run(suite)

TestLoader()

• TestLoader 用於建立類和模組的測試套件，一般的情況下，TestLoader().loadTestsFromTestCase(TestClass)來載入測試類。
• TestLoaderTest.py檔案

import unittest, csv
import os, sys
import time
import test1
import test2


# 手工新增案例到套件，
def createSuite():
    suite = unittest.TestSuite()
    # 將測試用例加入到測試容器（套件）中
    suite1 = unittest.TestLoader().loadTestsFromTestCase(test1.TestBaiDu1)
    suite2 = unittest.TestLoader().loadTestsFromTestCase(test2.TestBaiDu2)
    suite = unittest.TestSuite([suite1, suite2])
    return suite


if __name__ == "__main__":
    suite = createSuite()
    runner = unittest.TextTestRunner(verbosity=2)
    runner.run(suite)

discover（）的應用

• discover 是通過遞迴的方式到其子目錄中從指定的目錄開始， 找到所有測試模組並返回一個包含它們物件的TestSuite ，然後進行載入與模式匹配唯一的測試檔案，discover 引數分別為
  discover(dir,pattern,top_level_dir=None)
• TestDiscover.py檔案

import unittest, csv
import os, sys
import time


# 手工新增案例到套件，
def createSuite():

    # 第一個引數是掃描的路徑 ../表示回到父路徑, 然後遞迴掃描20201020這個檔案下的檔案
    # 第二個引數進行匹配test開頭的.py結尾的檔案, 如果匹配就載入裡面的內容
    # 第三個參數列示測試模組的頂層目錄，如果沒有頂層目錄，預設為None；一般都是為None
    discover = unittest.defaultTestLoader.discover('../20201020', pattern='test*.py', top_level_dir=None)
    print(discover)
    return discover


if __name__ == "__main__":
    suite = createSuite()
    runner = unittest.TextTestRunner(verbosity=2)
    runner.run(suite)