以下是一段跑测试的代码,用HTMLTestRunner生成一个html的测试报告,然后调用发邮件模块把测试报告作为附件发出去
if __name__ == "__main__":
report_path = "../report/htmlreport.html"
fp = open(report_path, "wb")
s = unittest.TestSuite()
s.addTests(unittest.TestLoader().loadTestsFromTestCase(LoginTestCase))
runner = HTR.HTMLTestRunner(stream=fp, title="this is a test report from allen")
runner.run(s)
time.sleep(5)
sen = SendMail()
sen.send_mail(attach_file=report_path)
运行,邮件发送成功,查看附件也有,再看附件大小是0字节,打开是空的。单独执行邮件发送模块,附件是有内容的,问题到底出在哪里呢。
想了好久,试了好久,一直不成功,一开始以为是附件路径的问题,把路径写死为绝对路径,也还是不行。正当怎么也想不明白的时候,忽然想到会不会是文件打开了没有关闭啊,赶紧试试。
if __name__ == "__main__":
report_path = "../report/htmlreport.html"
fp = open(report_path, "wb")
s = unittest.TestSuite()
s.addTests(unittest.TestLoader().loadTestsFromTestCase(LoginTestCase))
runner = HTR.HTMLTestRunner(stream=fp, title="this is a test report from allen")
runner.run(s)
fp.close() # 加了这一句
time.sleep(5)
sen = SendMail()
sen.send_mail(attach_file=report_path)
运行,果然是这个原因,附件现在有内容了。
赶紧把这段代码中打开文件部分修改为with的方式:
if __name__ == "__main__":
report_path = "../report/htmlreport.html"
with open(report_path, "wb") as fp:
s = unittest.TestSuite()
s.addTests(unittest.TestLoader().loadTestsFromTestCase(LoginTestCase))
runner = HTR.HTMLTestRunner(stream=fp, title="this is a test report from allen")
runner.run(s)
time.sleep(5)
sen = SendMail()
sen.send_mail(attach_file=report_path)
原本打开文件自己都会写with的方法的,这次是从别处拷了代码过来没改,结果浪费了一些时间和脑细胞。以后但凡打开文件,还是老老实实的用with吧