一、安裝HTMLTestRunner

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

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

1.原因是官網(wǎng)的是python2語法寫的，看官手動(dòng)把官網(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，開始時(shí)，引入HTMLTestRunner模塊報(bào)錯(cuò)。

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):
    #初始化設(shè)置
    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__":
    #定義一個(gè)測(cè)試容器
    test = unittest.TestSuite()

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

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

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

    #運(yùn)行測(cè)試用例
    runner.run(test)
    file_result.close()

運(yùn)行測(cè)試腳本后，發(fā)現(xiàn)報(bào)錯(cuò)：

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

if not rmap.has_key(cls):

所以前往642行修改代碼：

5

運(yùn)行后繼續(xù)報(bào)錯(cuò)：

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

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

6

修改后運(yùn)行，發(fā)現(xiàn)又報(bào)錯(cuò)：

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)整個(gè)程序中，唯一一個(gè)print:

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

這個(gè)是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í)行測(cè)試用例，這一篇介紹unittest下addTest()方法來加載測(cè)試用例到測(cè)試套件中去。
test_get_page_title這個(gè)測(cè)試腳本有一個(gè)測(cè)試用例，test_search3有兩個(gè)測(cè)試用例，一共有三個(gè)測(cè)試用例。
'''
'''
今天來學(xué)習(xí)下，如果加載這兩個(gè)測(cè)試類文件里面的3個(gè)測(cè)試腳本。
在unittest下有一個(gè)管理測(cè)試套件的叫TestSuit(),我們要使用這個(gè)測(cè)試套件，
需要先初始化一個(gè)suite實(shí)例，然后這個(gè)實(shí)例有一個(gè)addTest()的方法，
可以加載不同類里面的不同測(cè)試函數(shù)，格式這樣的 
addTest(測(cè)試類的類名（‘測(cè)試函數(shù)名稱，就是test開頭的函數(shù)’）)，
我們?cè)趖estsuites這個(gè)包下新建一個(gè)TestRunner.py文件用來管理我們的用例啟動(dòng)方式。
'''
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)

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

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

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

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

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

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

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

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

fp.close()

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

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

image.png