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)