unittest模块使用方法

unittest模块常用属性

1、 unittest.TestCase类：所有的测试用例类继承的基类

定义一个测试用例类，需要继承TestCase，比如：

class BaiduTest(unittest.TestCase):

2、 unittest.main():使用这个可以方便的将一个测试模块变为可直接运行的测试脚本，main()方法使用TestLoader类来搜索所有包含在该模块中以test命名开头的测试方法，并自动执行它们。执行方法的默认顺序是：根据ASCII码的顺序加载测试用例，数字与字母的顺序为：0-9，A-Z，a-z。

3、 unittest.TestSuite()：用来创建测试套件

4、 unittest.TextTestRunner():通过该类的run()方法可运行suite组装的测试用例。

5、 unittest.defaultTestLoader()：通过该类下的discover()方法可以自动根据测试目录匹配查找到的测试用例文件，并将查找到的测试用例组装到测试套件，因此可以直接通过run()方法执行discover。例如：

discover = unittest.defaultTestLoader.discover(test_dir, pattern=’test_*.py’)

第一个参数是测试用例目录，第二个参数是查找以test_开头的py文件

6、 unittest.skip()：装饰器，当运行用例时，有些用例可能不想执行等，可以用装饰器暂时屏蔽该条测试用例。一种常见的用法就是比如说想调试某一个测试用例，想先屏蔽其他用例，就可以用装饰器屏蔽。

a) @unittest.skip(reason):skip(reason)这个装饰器是无条件跳过装饰器所装饰的case，并说明跳过测试的原因

b) @unittest.skipIf(reason):skipIf(condition, reason):条件为真时，跳过装饰的case，并说明跳过原因。

c) @unittest.skipUnless(reason):skipUnless(condition, reason):条件为假时，跳过装饰的测试，并说明跳过测试的原因

d) @unittest.expecteFailure():expecteFailure():测试标记为失败

TestCase类常用属性

setUp()：setUp()方法是用于测试用例执行前的初始化工作。如测试用例需要访问数据库，可以在setUp()中建立数据库连接并初始化，需要登录web，可以先实例化浏览器等。
tearDown()：用于测试用例执行之后的善后工作，比如关闭数据库连接，关闭浏览器等。
assertEqual(a, b, [msg=测试失败时打印的信息]):断言a,b两个值是否相等，相等则测试通过。
assertNotEqual(a, b, [msg=测试失败时打印的信息]):断言a,b两个值是否相等，不相等则测试通过.
assertTrue(x, [msg]):断言x是否True，true则测试通过
assertFalse(x, [msg]):断言x是否False，False则测试通过
assertIs(a, b, [msg])：断言a是否是b，是则测试通过
assertNotIs(a, b, [msg])：断言a是否是b，不是则测试通过
assertIsNone(x, [msg]):断言x是否是None，是None则测试通过
assertIsNotNone(x, [msg]):断言x是否是None，是None则测试通过
assertIn(a, b, [msg])：断言a是否在b中，是则测试通过
assertNotIn(a, b, [msg])：断言a是否在b中，不在则测试通过
assertIsInstance(a, b, [msg])：断言a是否是b的一个实例，是则测试通过
assertNotIsInstance(a, b, [msg])：断言a是否是b的一个实例，不是则测试通过

举例：

class TestSub(unittest.TestCase):

    def setUp(self):
        """为测试方法初始化，每个case运行前执行一次"""
        self.a = 20
        self.b = 10

    def tearDown(self):
        """清理函数，每个case执行后执行一次"""
        pass

    def test_sub(self):
        """验证减法"""
        result = self.a - self.b
        self.assertEqual(result, 10)

 

def test_add(self):
        """验证加法"""
        result = self.a + self.b
        self.assertEqual(result, 30)

TestSuite类常用属性

addTest()：将测试用例添加到测试套件中，比如：

#将test_baidu模块下的BaiduTest类下的test_baidu_serach和test_baidu_login测试用例添加到套件

suite = unittest.TestSuite()

suite.addTest(TestSub ("test_sub"))

suite.addTest(TestSub ("test_add "))

runner = unittest.TextTestRunner()

runner.run(suite)

TextTestRunner类常用属性

run()方法是运行测试套件中的测试用例,参数是一个测试套件，例如：

runner = unittest.TextTestRunner()

runner.run(suite)

if __name__ == '__main__':
    # 可以直接调用main方法执行测试用例
    # unittest.main()

    # 也可以像下面这样先创建测试套件，然后将测试用例放入套件，再使用run方法运行套件
    test_dir = './'
    discover = unittest.defaultTestLoader.discover(test_dir, pattern='test_*.py')
    runner = unittest.TextTestRunner()
    runner.run(discover)

综合实例

实例1：

# coding=utf-8

"""
create on 2019.12.26
@author:suyang
Project:使用unittest编写测试用例
"""
import unittest

# 定义测试类，继承TestCase的所有方法
class Test(unittest.TestCase):
    # 重写父类的setUp方法10

    def setUp(self):
        self.number = input("Enter a number:")
        self.number = int(self.number)

    # 定义第一个测试用例
    def test_case1(self):
        print(self.number)
        self.assertEqual(self.number, 10, "Your input is not 10")

     # 定义第二个测试用例
    @unittest.skip("暂时跳过用例2的测试")
    def test_case2(self):
        print(self.number)
        self.assertEqual(self.number, 20, "Your input is not 20")

    def test_case3(self):
        result = self.number + 10
        self.assertEqual(result, 30, "Your input is not 20")

    def tearDown(self):
        print('test over')

# 3种方法都可以执行测试用例
if __name__ == '__main__':
    # 可以直接调用main方法执行测试用例
    # unittest.main()

    # 先创建一个测试套件，然后将用例依次用addTest方法加入测试套件中，再用run方法执行用例
    # suit = unittest.TestSuite()
    # suit.addTest(Test('test_case1'))
    # suit.addTest(Test('test_case2'))
    # suit.addTest(Test('test_case3'))
    # runner = unittest.TextTestRunner()
    # runner.run(suit)

    # 也可以像下面这样先创建测试套件，然后将测试用例放入套件，再使用run方法运行套件
    test_dir = './'
    discover = unittest.defaultTestLoader.discover(test_dir, pattern='test_*.py')
    runner = unittest.TextTestRunner()
    runner.run(discover)

实例2：

# coding=utf-8
"""
created on 2019.12.26
@author:suyang
Project:测试百度搜索
"""
from selenium import webdriver
import unittest
import time


class BaiduTest(unittest.TestCase):
    def setUp(self):
        # pass
        self.driver = webdriver.Chrome()
        # 隐性等待时间为30秒
        self.driver.implicitly_wait(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()


if __name__ == '__main__':
    unittest.main()