為了使程式更規範，且方便後期維護，故把程式分為四個模組：case、suite、HTMLTestRunner、methods

case：用例模組；編寫測試模組的用例資料

suite：測試套件模組；載入需要測試的具體用例

HTMLTestRunner：測試報告模組；定義生成測試報告

methods：方法模組；編寫測試模組的用例方法

------------------------------------------------------------------------------------------------------

case模組：

one.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import unittest
import two

class MyTestCase(unittest.TestCase):

    # 初始化工作
    def setUp(self):
        pass

    # 退出清理工作
    def tearDown(self):
        pass

    # 具體的測試用例，一定要以test開頭
    def test_Case1(self):
        self.assertMultiLineEqual(two.baidu1(), u'百度一下，你就知道')

    def test_Case2(self):
        self.assertMultiLineEqual(two.baidu2(), u'京公網安備11000002000001號')
 

three.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import unittest
import one

def suite1():
    # 建立測試套件
    suite = unittest.TestSuite()
    # 為套件新增測試用例
    suite.addTest(one.MyTestCase("test_Case1"))
    return suite

def suite2():
    # 建立測試套件
    suite = unittest.TestSuite()
    # 為套件新增測試用例
    suite.addTest(one.MyTestCase("test_Case2"))
    return suite

def all_suite():
    suite = unittest.TestSuite((suite2(), suite1()))
    return suite
 

HTMLTestRunner模組：

four.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import HTMLTestRunner

def save_file():
    # 確定生成報告的路徑
    filePath = "pyResult.html"
    fp = file(filePath, 'wb')
    # 生成報告的Title,描述
    runner = HTMLTestRunner.HTMLTestRunner(stream=fp, title='Python TestReport', description='This  is Python  Report')
    return runner
 

two.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import time
from selenium import webdriver

br = webdriver.Chrome()
br.maximize_window()
br.get('http://www.baidu.com')

def baidu1():
    time.sleep(1)
    title = br.title
    return title

def baidu2():
    time.sleep(1)
    jg = br.find_element_by_id('jgwab').text
    return jg

run.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import four
import three
import unittest

if __name__ == '__main__':
    # 呼叫定義HTMLTestRunner的方法
    runner = four.save_file()
    # 執行suite所組裝的測試用例
    # runner = unittest.TextTestRunner()
    # 呼叫測試套件方法
    all_suite = three.all_suite()
    # 執行測試套件
    runner.run(all_suite)