Python+Selenium筆記（四）：unittest的Test Suite（測試套件）

（一） Test Suite測試套件

一個測試套件是多個測試或測試用例的集合，是針對被測程式的對應的功能和模塊創建的一組測試，一個測試套件內的測試用例將一起執行。

應用unittest的TestSuites特性，可以將不同的測試組成一個邏輯組，然後設置統一的測試套件，並通過一個命令來執行測試。這都是通過TestSuites、TestLoader和TestRunn類來實現的。

（二） 類級別的setUp()方法和tearDown()方法

使用setUpClass()方法和tearDownClass()方法及@classmethod標識來實現各個測試方法共用初始化數據。具體看下麵的代碼。

（三） searchtest.py

import unittest

from selenium import webdriver

from selenium.webdriver.common.action_chains import ActionChainsimport time

class SearchTest(unittest.TestCase):

    '''通過setUpClass()和@classmethod標識,實現

     在類級別初始化數據，所有測試方法共用這些初始化數據.

     不使用這個的話，每個測試方法都會單獨創建一個實例'''

    @classmethod

    def setUpClass(cls):

        cls.driver = webdriver.Firefox()

        cls.driver.implicitly_wait(10)

        cls.driver.maximize_window()

        cls.driver.get("https://www.cnblogs.com/")

    def test_search_by_category(self):

        #讀取category.txt文件，返回一個字典

        with open('data/category.txt', encoding='UTF-8') as category_file:

            category_dict = dict()

            category_data = category_file.readline().strip().split(',')

            the_class = category_data.pop(0)

            category_dict[the_class] = category_data

        #定位首頁網站分類中的編程語言

        self.search_class = self.driver.find_element_by_xpath('//li/a[@href="/cate/2/"]')

        #游標懸停在“編程語言”上

        ActionChains(self.driver).move_to_element(self.search_class).perform()

        # 以列表形式返回編程語言下的所有小類

        self.search_small = self.driver.find_elements_by_xpath(

            '//div[@id="cate_content_block_2"]/div[@class="cate_content_block"]/ul/li')

        #休眠3秒

        time.sleep(3)

        small_cate = []

        for s in self.search_small:

            #去掉小類最後面的（0），並添加到列表small_cate中

            small = str(s.text).split('(')

            small_cate.append(small[0])

        #檢查表達式是否為true（此處檢查編程語言下的小類是否與預期結果一致）

        self.assertTrue(small_cate == category_dict["編程語言"])

        # self.assertEqual(small_cate,category_dict["編程語言"])

    def test_search_by_look(self):

        self.seach_class = self.driver.find_element_by_xpath('//li/a[@href="/cate/2/"]')

        #定位編程語言下的小類Python

        self.seach_small =self.driver.find_element_by_xpath('//li/a[@href="/cate/python/"]')

        ActionChains(self.driver).move_to_element(self.seach_class).perform()

        self.seach_small.click()

        #檢查打開的網頁標題是不是 Python - 網站分類 - 博客園

        # assert "Python - 網站分類 - 博客園" in self.driver.title

        self.assertEqual(self.driver.title,"Python - 網站分類 - 博客園" )

    '''通過tearDownClass()和@classmethod標識,實現

     在類級別初始化數據，所有測試方法共用這些初始化數據'''

    @classmethod

    def tearDownClass(cls):

        cls.driver.quit()

#加上下麵2句，可以通過命令行運行測試，不加的話不影響通過IDE運行測試

if __name__ == '__main__':

    #加verbosity=2參數，在命令行中顯示具體的測試方法

    unittest.main(verbosity=2)

（四） homepagetest.py

import  unittest

from selenium import  webdriver

from selenium.common.exceptions import NoSuchElementException

from selenium.webdriver.common.by import By

class HomePageTest(unittest.TestCase):

    '''通過setUpClass()和@classmethod標識,實現

     在類級別初始化數據，所有測試方法共用這些初始化數據.

     不使用這個的話，每個測試方法都會單獨創建一個實例'''

    @classmethod

    def setUpClass(cls):

        cls.driver = webdriver.Firefox()

        cls.driver.implicitly_wait(10)

        cls.driver.maximize_window()

        cls.driver.get("https://www.cnblogs.com/")

    def test_search_field(self):

        #通過by,檢查博客園首頁有沒有搜索框,is_element_present()是自定義的方法

        self.assertTrue(self.is_element_present(By.ID,"zzk_q"))

    def test_search_btn(self):

        # 通過by,檢查博客園首頁有沒有找找看按鈕

        self.assertTrue(self.is_element_present(By.CLASS_NAME,"search_btn"))

    def test_menu(self):

        #該方法檢查博客園首頁菜單欄信息是否與預期一致

        #讀取menu.txt文件數據

        with open('data/menu.txt',encoding='UTF-8') as menu_file:

            menu_data = menu_file.readline().strip().split(',')

        #以列表形式返回博客園首頁菜單欄信息

        self.check_menu = self.driver.find_elements_by_xpath('//div[@id="nav_menu"]/a')

        the_menu = []

        for c in self.check_menu:

            #將博客園首頁的菜單名稱和URL添加到列表the_menu

            the_menu.append(c.text + c.get_attribute('href'))

        #檢查2個列表是否一致（檢查博客園首頁的菜單名稱及URL是否和預期一致）

        self.assertListEqual(the_menu,menu_data)

    def is_element_present(self,how,what):

        #找到元素，返回True,否則返回False

        try:

            self.driver.find_element(by=how,value = what)

        except NoSuchElementException as e:

            return False

        return True

    @classmethod

    def tearDownClass(cls):

        cls.driver.quit()

# 加上下麵2句，可以通過命令行運行測試，不加的話不影響通過IDE運行測試

if __name__ == '__main__':

    # 加verbosity=2參數，在命令行中顯示具體的測試方法

    unittest.main(verbosity=2)

（五） smoketests.py （測試套件）

import unittest

from searchtest import SearchTest

from homepagetest import HomePageTest

#獲取SearchTest類 和 HomePageTest類中的所有測試方法

search_test = unittest.TestLoader().loadTestsFromTestCase(SearchTest)

home_page_test = unittest.TestLoader().loadTestsFromTestCase(HomePageTest)

#創建一個包括SearchTest和HomePageTest的測試套件

smoke_tests = unittest.TestSuite([home_page_test,search_test])

#運行測試套件

unittest.TextTestRunner(verbosity=2).run(smoke_tests)

（六） 其他說明

#另外一個文件在上一篇中已經說明

要完全使用上面的代碼，需要創建一個文件menu.txt，輸入以下數據：

園子https://home.cnblogs.com/,新聞https://news.cnblogs.com/,博問https://q.cnblogs.com/,快閃記憶體https://ing.cnblogs.com/,小組https://group.cnblogs.com/,收藏https://wz.cnblogs.com/,招聘https://job.cnblogs.com/,班級https://edu.cnblogs.com/,找找看http://zzk.cnblogs.com/

（七） 運行smoketests.py