Python+Selenium笔记（四）：unittest的Test Suite（测试套件）

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)

园子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/