Python3中HTMLTestRunner报ImportError: No module named 'StringIO'的解决方案

Python3 HTMLTestRunner StringIO错误解决方案

在Python3环境中使用HTMLTestRunner.py时，若遇到ImportError: No module named 'StringIO'报错，可按以下步骤解决。

该错误的根本原因在于官网提供的HTMLTestRunner.py是基于Python2语法编写的。因此，需要手动将该文件修改为兼容Python3的语法。

相关参考链接：http://bbs.chinaunix.net/thread-4154743-1-1.html

官方下载地址：http://tungwaiyip.info/software/HTMLTestRunner.html

修改后的Python3兼容版下载地址：HTMLTestRunner_jb51.rar（为方便用户，可直接下载此版本）。

以下是具体的代码修改汇总：

第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源码的第94行中，原代码使用了StringIO。然而Python3已移除StringIO模块，取而代之的是io.StringIO。因此，需将该行修改为import io。

在HTMLTestRunner源码的第539行中，将self.outputBuffer = StringIO.StringIO()修改为self.outputBuffer = io.StringIO()。

完成上述修改后，模块即可成功引入。

运行测试脚本时的后续报错处理

执行测试脚本代码：

# -*- 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"用例执行情况")

  #运行测试用例
  runner.run(test)
  file_result.close()

运行测试脚本后，发现出现新的报错信息。

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

if not rmap.has_key(cls):

因此需要前往第642行修改相关代码。

再次运行后，继续出现报错。

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

需要前往第766行和第772行继续修改（注意：第766行处理的是uo变量，而第772行处理的是ue变量。此前因未注意区分这两个变量导致报出莫名其妙的错误，耗费了大量时间排查）。

修改后再次运行，发现又出现新的报错。

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行查看，发现这是整个程序中唯一一处使用了旧版print语法的地方。

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

这是Python 2.x版本的写法，需要将其修改为Python 3.x的print语法，具体修改如下。

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

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持本站。