unittest对单个测试类的多种测试执行方法总结

　　基于unittest测试框架编写的测试脚本，一般单个测试类下会有多个测试方法，unittest也提供多种测试执行方式，下面就不同方式或者需求一一实操并说明：

一、使用unittest下main()方法的运行所有测试方法，实例代码如下：

'''单个测试类的多种运行方式'''
# -*- coding:utf-8 -*-
import unittest
from selenium import webdriver
from time import sleep
import HTMLTestRunner

class BaiduSearch(unittest.TestCase):
    @classmethod   
    def setUpClass(cls):
        #create a new browser session
        cls.driver = webdriver.Chrome()
        cls.driver.implicitly_wait(30)
        cls.driver.maximize_window()

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

    def test_search_selenium(self):
        #get the search textbox and enter search keyword
        search_textbox = self.driver.find_element_by_id('kw')
        search_textbox.clear()
        search_textbox.send_keys("selenium")
        #get the and seacrh button and click
        seacrh_button = self.driver.find_element_by_id('su')
        seacrh_button.click()
        
        #add assert
        sleep(2)
        tag = self.driver.find_element_by_link_text("Selenium Remote Control").text
        self.assertIn("Selenium",tag)

    def test_search_python(self):
        search_textbox = self.driver.find_element_by_id('kw')
        search_textbox.clear()
        search_textbox.send_keys("python")
        seacrh_button = self.driver.find_element_by_id('su')
        seacrh_button.click()

        sleep(3)
        tag = self.driver.find_element_by_link_text("PyPI").text
        self.assertEqual("PyPI",tag)

    @classmethod
    def tearDownClass(cls):
        #close the browser window
        cls.driver.quit()

#使用unittest下main()方法的运行所有测试方法
if __name__ == '__main__':
    unittest.main(verbosity=2)  #传递verbosity参数查看更详细的测试结果展示

　　运行结果如下 ：

　　

二、使用TestLoader和TextTestRunner来创建和运行测试套件，代码如下：

...
suite = unittest.TestLoader().loadTestsFromTestCase(BaiduSearch)     
unittest.TextTestRunner(verbosity=2).run(suite)

　　运行结果同上。

三、使用TestSuite和TextTestRunner来创建和运行测试套件，代码如下：

...
suite = unittest.TestSuite()
suite.addTest(BaiduSearch("test_search_python"))   #使用addTest()添加测试类下面的单个测试方法
suite.addTest(BaiduSearch("test_search_selenium"))
unittest.TextTestRunner(verbosity=2).run(suite)

　　运行结果同上。

　　注意这个方法可以灵活只运行测试类下的任何一个或者几个测试方法，只需要addTest相应测试方法即可，或者在添加完所有方法后注释掉其中不打算测试的方法也可以。

四、使用HTMLTestRunner和TestLoader来创建和运行测试套件并生成漂亮的HTML格式的测试报告，代码如下：

...
suite = unittest.TestLoader().loadTestsFromTestCase(BaiduSearch)
#open the report file
report_file = open('./' + ' BaiduSearch_test.html', 'wb')
#configure HTMLTestRunner options and run the suite
HTMLTestRunner.HTMLTestRunner(stream = report_file,
                              title = "baidu",
                              description = "Baidu Test",
                              verbosity=2).run(suite)

　　运行结果同上且生成一个HTML格式的测试报告在当前目录下。