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