python单元测试模块unittest

1.unittest_demo.py

# coding=utf-8
import time
import unittest

from HtmlTestRunner import HTMLTestRunner
from selenium import webdriver

class BaiduTest(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(30)  # 隐性等待时间为30秒
        self.base_url = "https://www.baidu.com"

    def test_baidu(self):
        driver = self.driver
        driver.get(self.base_url + "/")
        driver.find_element_by_id("kw").clear()
        driver.find_element_by_id("kw").send_keys("unittest")
        driver.find_element_by_id("su").click()
        time.sleep(3)
        title = driver.title
        self.assertEqual(title, u"unittest_百度搜索")

    def tearDown(self):
        self.driver.quit()

class YoudaoTest(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(10)  # 隐性等待时间为10秒
        self.base_url = "http://www.youdao.com"

    def test_youdao(self):
        driver = self.driver
        driver.get(self.base_url + "/")
        driver.find_element_by_id("translateContent").clear()
        driver.find_element_by_id("translateContent").send_keys(u"你好")
        driver.find_element_by_id("translateContent").submit()
        time.sleep(3)
        page_source = driver.page_source
        self.assertIn("hello", page_source)

    def tearDown(self):
        self.driver.quit()

if __name__ == "__main__":
    """方式一"""
    # unittest.main()方法会搜索该模块下所有以test_xxx的测试用例方法，并自动执行它们
    # unittest.main()
    """方式二"""
    # # 构造测试集
    # suite = unittest.TestSuite()
    # suite.addTest(BaiduTest('test_baidu'))
    # suite.addTest(YoudaoTest('test_youdao'))
    # # 执行测试
    # runner = unittest.TextTestRunner()
    # runner.run(suite)
    """方式三"""
    # 构造测试集
    suite = unittest.TestSuite()
    suite.addTest(BaiduTest('test_baidu'))
    suite.addTest(YoudaoTest('test_youdao'))
    # 定义报告文件权限，wb，表示有读写权限
    runner = HTMLTestRunner(report_title="wtf_title", report_name="wtf_name", descriptions="wtf_description", add_timestamp=False,
                            open_in_browser=True)
    # 执行测试
    runner.run(suite)

2.使用代码中的‘方式三’生成测试报告

　　pycharm 在运行测试用例的时候，默认是以unittest 框架来运行的，所以不能生成测试报告。只要运行环境是unittest in ...的，就都是以unittest框架运行的。只要不以unittest框架方式运行就可解决不能生成报告的问题

ps:

　　HtmlTestRunner模块安装：pip install html-testRunner