一、安裝HTMLTestRunner

HTMLTestRunner 是 Python 標準庫的 unittest 模塊的一個擴展，它可以生成 HTML的 測試報告。
首先要下 HTMLTestRunner.py 文件，將下載的文件放入…\python\Lib目錄下
python2版下載地址：http://tungwaiyip.info/software/HTMLTestRunner.html
python3版下載地址：http://pan.baidu.com/s/1dEZQ0pz

python3中用HTMLTestRunner.py報ImportError: No module named 'StringIO'的解決方法：

1.原因是官網(wǎng)的是python2語法寫的，看官手動把官網(wǎng)的HTMLTestRunner.py改成python3的語法：

參考：http://bbs.chinaunix.net/thread-4154743-1-1.html

下載地址：http://tungwaiyip.info/software/HTMLTestRunner.html

修改后下載地址：http://pan.baidu.com/s/1dEZQ0pz （懶人直接下載吧）

2.修改匯總：

第94行，將import StringIO修改成import io

第539行，將self.outputBuffer = StringIO.StringIO()修改成self.outputBuffer = io.StringIO()

第642行，將if not rmap.has_key(cls):修改成if not cls in rmap:

第766行，將uo = o.decode('latin-1')修改成uo = e

第775行，將ue = e.decode('latin-1')修改成ue = e

第631行，將print >> sys.stderr, '\nTime Elapsed: %s' % (self.stopTime-self.startTime)修改成print(sys.stderr, '\nTime Elapsed: %s' % (self.stopTime-self.startTime))

在Python3.4下使用HTMLTestRunner，開始時，引入HTMLTestRunner模塊報錯。

1

在HTMLTestRunner的94行中，是使用的StringIO，但是Python3中，已經(jīng)沒有StringIO了。取而代之的是io.StringIO。所以將此行修改成import io

2

在HTMLTestRunner的539行中，self.outputBuffer = StringIO.StringIO()修改成self.outputBuffer = io.StringIO()

3

修改以后，成功引入模塊了

4

執(zhí)行腳本代碼：

# -*- coding: utf-8 -*-
#引入webdriver和unittest所需要的包
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
import unittest, time, re

#引入HTMLTestRunner包
import HTMLTestRunner   

class Baidu(unittest.TestCase):
    #初始化設置
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.implicitly_wait(30)
        self.base_url = "http://www.baidu.com/"
        self.verificationErrors = []
        self.accept_next_alert = True

    #百度搜索用例
    def test_baidu(self):
        driver = self.driver
        driver.get(self.base_url)
        driver.find_element_by_id("kw").click()
        driver.find_element_by_id("kw").clear()
        driver.find_element_by_id("kw").send_keys("Selenium Webdriver")
        driver.find_element_by_id("su").click()
        time.sleep(2)
        driver.close()

    def tearDown(self):
        self.driver.quit()
        self.assertEqual([], self.verificationErrors)

if __name__ == "__main__":
    #定義一個測試容器
    test = unittest.TestSuite()

    #將測試用例，加入到測試容器中
    test.addTest(Baidu("test_baidu"))

    #定義個報告存放的路徑，支持相對路徑
    file_path = "F:\\RobotTest\\result.html"
    file_result= open(file_path, 'wb')

    #定義測試報告
    runner = HTMLTestRunner.HTMLTestRunner(stream = file_result, title = u"百度搜索測試報告", description = u"用例執(zhí)行情況")

    #運行測試用例
    runner.run(test)
    file_result.close()

運行測試腳本后，發(fā)現(xiàn)報錯：

File "C:\Python34\lib\HTMLTestRunner.py", line 642, in sortResult

if not rmap.has_key(cls):

所以前往642行修改代碼：

5

運行后繼續(xù)報錯：

AttributeError: 'str' object has no attribute 'decode'

前往766, 772行繼續(xù)修改（注意：766行是uo而772行是ue，當時眼瞎，沒有注意到這些，以為是一樣的，導致報了一些莫名其妙的錯誤，折騰的半天）：

6

修改后運行，發(fā)現(xiàn)又報錯：

File "C:\Python34\lib\HTMLTestRunner.py", line 631, in run

print >> sys.stderr, '\nTime Elapsed: %s' % (self.stopTime-self.startTime)

TypeError: unsupported operand type(s) for >>: 'builtin_function_or_method' and '_io.TextIOWrapper'

前往631查看，發(fā)現(xiàn)整個程序中，唯一一個print:

print >> sys.stderr, '\nTime Elapsed: %s' % (self.stopTime-self.startTime

這個是2.x的寫法，咱們修改成3.x的print，修改如下：

print(sys.stderr, '\nTime Elapsed: %s' % (self.stopTime-self.startTime))

將HTMLTestRunner.py保存如Lib文件下：

image.png

image.png

二、編寫TestRunner.py

'''
本文開始介紹如何通過unittest來管理和執(zhí)行測試用例，這一篇介紹unittest下addTest()方法來加載測試用例到測試套件中去。
test_get_page_title這個測試腳本有一個測試用例，test_search3有兩個測試用例，一共有三個測試用例。
'''
'''
今天來學習下，如果加載這兩個測試類文件里面的3個測試腳本。
在unittest下有一個管理測試套件的叫TestSuit(),我們要使用這個測試套件，
需要先初始化一個suite實例，然后這個實例有一個addTest()的方法，
可以加載不同類里面的不同測試函數(shù)，格式這樣的 
addTest(測試類的類名（‘測試函數(shù)名稱，就是test開頭的函數(shù)’）)，
我們在testsuites這個包下新建一個TestRunner.py文件用來管理我們的用例啟動方式。
'''
import unittest
import test_suit
from test_suit.test_search3 import BaiduSearch3
from test_suit.test_get_page_title import GetPageTitle

from Lib.HTMLTestRunner import HTMLTestRunner
import os
import time


# suite = unittest.TestSuite()
# suite.addTest(BaiduSearch3('test_baidu_search'))
# suite.addTest(BaiduSearch3('test_search2'))
# suite.addTest(GetPageTitle('test_get_title'))
#
# if __name__ == '__main__':
#     #  執(zhí)行用例
#     runner = unittest.TextTestRunner()
#     runner.run(suite)

'''
這個方法有沒有局限性呢？當然有，加入你有幾百個測試類，你是不是也一直這樣手動去添加？
有沒有其他更好的方法去加載我們的測試腳本，關于這個問題，請看下一篇文章。
'''
# suite = unittest.TestSuite(unittest.makeSuite(BaiduSearch3))
#
# if __name__ == '__main__':
#     #  執(zhí)行用例
#     runner = unittest.TextTestRunner()
#     runner.run(suite)

'''
缺陷: makeSuite()方法雖然比前面文章的addTest()方法有了一定的效率提升，在添加測試用例到測試套件過程。
但是這個方法也有很多缺點，我不可能把所有項目腳本都放一個測試類文件吧。
在unittest中還是有很多方法可以用來添加測試用例到suite中，makeSuite（類名稱），只是一種，
'''

'''
test_suit是可以包名，也可以是一個文件夾名稱，
在實際腳本開發(fā)過程中，最后都采用這個方法來批量管理和執(zhí)行幾百上千的測試用例。
'''

# suite = unittest.TestLoader().discover("test_suit")
# if __name__ == '__main__':
#     #  執(zhí)行用例
#     runner = unittest.TextTestRunner()
#     runner.run(suite)

#  設置報告文件保存路徑
report_path = os.path.dirname(os.path.abspath('.')) + '/test_report/'
#  獲取系統(tǒng)當前時間
now = time.strftime("%Y-%m-%d-%H_%M_%S", time.localtime(time.time()))

#  設置報告名稱格式
HtmlFile = report_path + now + "HTMLtemplate.html"
print(HtmlFile)
#  python3.0不支持file函數(shù)，可以用open()來替代
# fp = file(HtmlFile, "wb")
fp = open(HtmlFile, "wb")  #  寫入

#  構建suite
suite = unittest.TestLoader().discover("test_suit")

if __name__ == '__main__':
    #  初始化一個HTMLTestRunner實力對象，用來生成報告
    runner = HTMLTestRunner(stream=fp, title=u"某某項目測試報告", description=u"用例測試情況")
    #  開始執(zhí)行測試套件
    runner.run(suite)

fp.close()

補充：
Python open() 函數(shù)

最后，用瀏覽器打開test_report，如下圖。

image.png