unittest框架（一）

基本使用

from selenium import webdriver
import unittest

class Baidu(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.driver=webdriver.Firefox()
        cls.driver.implicitly_wait(5)

    def test_baidu_search(self):
        driver = self.driver
        driver.get("http://www.baidu.com")
        driver.find_element_by_css_selector("#kw").send_keys("selenium")
        driver.find_element_by_css_selector("#su").click()


    def test_baidu_set(self):
        driver = self.driver
        driver.get("http://www.baidu.com")
        driver.find_element_by_css_selector("#s_usersetting_top>span").click()


    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()

    if __name__=="__main__":
        unittest.main()

產生報告

#coding=utf-8
import unittest
import HTMLTestRunner
import Baidu

suite=unittest.TestSuite()
suite.addTest(unittest.makeSuite(Baidu.Baidu))
file = open("E:\\result.html","wb")
runner = HTMLTestRunner.HTMLTestRunner(stream=file,title=u'測試報告',description=u'用例執行情況：')
runner.run(suite)