unittest单元测试框架

单元测试是指对软件中的最小可测试单元进行检查和验证。不同编程语言有不同的单元测试框架，Java有Junit,TestNg。python中有unittest,Pyunit,testtools。
单元测试框架的作用：提供用例组织与执行；提供丰富的断言方法；提供丰富的日志与测试结果。

一、unittest核心要素
1、TestCase
一个TestCase的实例就是一个测试用例。测试用例就是一个完整的测试流程，包括测试前准备环境的准备(setUp)，执行测试代码(run)，以及测试后环境的还原(teardDown)。单元测试(unit test)的本质也就在这里，一个测试用例是一个完整的测试单元，通过运行这个测试单元，可以对某一个问题进行验证。
2、TestSuite
多个测试用例集合在一起，就是TestSuite，而且TestSuite也可以嵌套TestSuite。TestLoader是用来加载测试用例到TestSuite中的。
3、TextTestRunner
TextTestRunner是来执行测试用例的，其中的run()会执行TestSuite或TestCase中的run(result)方法。测试的结果会保存到TextTestResult实例中，包括运行了多少测试用例，成功了多少，失败了多少等信息。
4、Fixture
对一个测试用例环境的搭建和销毁是一个fixture。

calendar.py

class Math(object):
    def __init__(self, a, b):
        self.a = int(a)
        self.b = int(b)

    def add(self):
        return self.a + self.b

 test_Math.py

from calendar import Math
import unittest


class TestMath(unittest.TestCase):
    def setUp(self):
        print("test start")

    def test_add(self):
        j = Math(5, 10)
        self.assertEqual(j.add(), 15)
        # 用例失败场景
        # self.assertEqual(j.add(),12)

    def tearDown(self):
        print("test end")


if __name__ == '__main__':
    # 构造测试集
    suite = unittest.TestSuite()
    suite.addTest(TestMath("test_add"))

    # 执行测试
    runner = unittest.TextTestRunner()
    runner.run(suite)

 二、断言

断言内容是自动化脚本的重要内容，正确设置断言以后才能帮助我们判断测试用例执行结果。
断言方法：
assertEqual(a,b) 判断 a==b
assertNotEqual(a,b) 判断 a!=b
assertTrue(x) bool(x) is True
assertFalse(x) bool(y) is False
assertIs(a,b) a is b
assertIsNot(a,b) a is not b
assertIsNone(x) x is None
assertIsNotNone(x) x is not None
assertIn(a,b) a in b
assertNotIn(a,b) a not in b
assertIsInstance(a,b) isinstance(a,b)
assertNotIsInstance(a,b) not isinstance(a,b)

from calendar import Math
import unittest


class TestMath(unittest.TestCase):
    def setUp(self):
        print("test start")

    def test_add(self):
        j = Math(5, 10)
        self.assertEqual(j.add(), 15)
        # 用例失败场景
        # self.assertEqual(j.add(),12)

    def test_add_not_equal(self):
        j = Math(5, 10)
        self.assertNotEqual(j.add(), 12)

    def test_add_true(self):
        j=Math(5,10)
        self.assertTrue(j.add()>10)

    def test_in(self):
        self.assertIn("zxw","wo yao zxw")

    def test_is(self):
        self.assertIs("zxw","zxwa")

    def tearDown(self):
        print("test end")


if __name__ == '__main__':
    # 构造测试集
    suite = unittest.TestSuite()
    suite.addTest(TestMath("test_add"))
    suite.addTest(TestMath("test_add_not_equal"))
    suite.addTest(TestMath("test_add_true"))
    suite.addTest(TestMath("test_in"))
    suite.addTest(TestMath("test_is"))

    # 执行测试
    runner = unittest.TextTestRunner()
    runner.run(suite)

三、新增用例管理

前面是针对单个add方法来进行单元测试，如果需要多个方法来进行测试，该如何处理？如新增一个sub方法来进行单元测试验证。

import unittest
from calendar import *


class Test_add(unittest.TestCase):
    def setUp(self):
        print("test is start")

    def test_add(self):
        j = Math(5, 5)
        self.assertEqual(j.add(), 10)

    def test_add1(self):
        j = Math(10, 20)
        self.assertEqual(j.add(), 30)

    def tearDown(self):
        print("test is end")


class Test_sub(unittest.TestCase):
    def setUp(self):
        print("sub test is start")

    def test_sub(self):
        i = Math(8, 8)
        self.assertEqual(i.sub(), 0)

    def test_sub1(self):
        i = Math(5, 3)
        self.assertEqual(i.sub(), 2)

    def tearDown(self):
        print("sub test is end")


if __name__ == '__main__':
    suite = unittest.TestSuite
    suite.addTest(Test_add("test_add"))
    suite.addTest(Test_add("test_add1"))
    suite.addTest(Test_sub("test_sub"))
    suite.addTest(Test_sub("test_sub1"))

    runner = unittest.TextTestRunner()
    runner.run(suite)

四、用例公共部分hebing

前面每个测试类都有setUp()和tearDown()方法，而且两个方法内容都是一样的，用于打印开始与结束提示语句，是否可以合并在一起呢？

from calendar import *
import unittest


class Test_StartEnd(unittest.TestCase):
    def setUp(self):
        print('test start')

    def tearDown(self):
        print("test end")


class Test_add(Test_StartEnd):
    def test_add(self):
        j = Math(5, 5)
        self.assertEqual(j.add(), 10)


class Test_sub(Test_StartEnd):
    def test_sub(self):
        i = Math(8, 8)
        self.assertEqual(i.sub(), 0)


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

 五、用例执行顺序

观察如下测试脚本，思考测试用例执行顺序。

import unittest


class Test1(unittest.TestCase):
    def setUp(self):
        print("Test1 start")

    def test_c(self):
        print("test_c")

    def test_b(self):
        print("test_b")

    def tearDown(self):
        print("Test2 end")


class Test2(unittest.TestCase):
    def setUp(self):
        print("Test2 start")

    def test_d(self):
        print("test_d")

    def test_a(self):
        print("test_a")

    def tearDown(self):
        print("Test2 end!")


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

 执行顺序规则——测试类或测试方法的数字与字母顺序0～9，A-Z。

自定义用例执行顺序：按照unittest.TestSuite的addTest()方法添加的顺序执行。

import unittest


class Test1(unittest.TestCase):
    def setUp(self):
        print("Test1 start")

    def test_c(self):
        print("test_c")

    def test_b(self):
        print("test_b")

    def tearDown(self):
        print("Test2 end")


class Test2(unittest.TestCase):
    def setUp(self):
        print("Test2 start")

    def test_d(self):
        print("test_d")

    def test_a(self):
        print("test_a")

    def tearDown(self):
        print("Test2 end!")


if __name__ == '__main__':
    suite = unittest.TestSuite
    suite.addTest(Test2("test_d"))
    suite.addTest(Test1("test_c"))
    suite.addTest(Test2("test_a"))
    suite.addTest(Test1("test_b"))

    runner = unittest.TextTestRunner()
    runner.run(suite)

六、用例综合框架管理

前面测试用例与执行都是写在一个文件，当用例数量不断增加的时候，用例的执行与管理变得非常麻烦，因此需要对用例根据具体的功能模块来使用单独的模块来管理。
案例：Test_Project文件目录下包含4个python文件：
calculator.py   ——加减法运算方法的实现
start_end.py   ——SetUp与TearDown管理
test_add.py    ——加法测试用例
test_sub.py    ——减法测试用例
runtest.py      ——用例执行管理

calculator.py 

class Math(object):
    def __init__(self, a, b):
        self.a = int(a)
        self.b = int(b)

    def add(self):
        return self.a + self.b

    def sub(self):
        return self.a - self.b 

start_end.py 

import unittest


class Setup_tearDown(unittest.TestCase):
    def setUp(self):
        print("start test")

    def tearDown(self):
        print("end test")

test_add.py 

from calculator import *
from start_end import *


class Test_add(Setup_tearDown):
    def test_add(self):
        j = Math(5, 5)
        self.assertEqual(j.add(), 10)

    def test_add1(self):
        j = Math(8, 8)
        self.assertEqual(j.add(), 16)

 test_sub.py 

from calculator import *
from start_end import *


class Test_sub(Setup_tearDown):
    def test_sub(self):
        i = Math(5, 3)
        self.assertEqual(i.sub(), 2)

    def test_add1(self):
        i = Math(8, 8)
        self.assertEqual(i.sub(), 0)

runtest.py 

'''
使用discover可以一次调用多个脚本
test_dir被测试脚本的路径
pattern脚本名称匹配规则
'''

import unittest

test_dir = './'

discovery = unittest.defaultTestLoader.discover(test_dir, pattern="test*.py")

if __name__ == '__main__':
    runner = unittest.TextTestRunner()
    runner.run(discovery)

 七、跳过测试和预期失败

unittest.skip() 直接跳过测试
unittest.skipIf() 条件为真，跳过测试
unittest.skipUnless() 条件为假，跳过测试
unittest.expectedFailure 预期设置失败

import unittest


class Test1(unittest.TestCase):
    # 类运行之前运行次方法
    @classmethod
    def setUpClass(cls):
        print("class module start test>>>>>>>")

    # 类中的所有方法执行完成之后执行此方法
    @classmethod
    def tearDownClass(cls):
        print("class module end test<<<<<<<<")

    def setUp(self):
        print("Test1 start")

    @unittest.skipIf(4 > 3, "skip test_c")
    def test_c(self):
        print("test_c")

    @unittest.skipUnless(0 > 1, "skip test_b")
    def test_b(self):
        print("test_b")

    def tearDown(self):
        print("Test2 end")


# @unittest.skip("skip test2")
class Test2(unittest.TestCase):
    def setUp(self):
        print("Test2 start")

    def test_d(self):
        print("test_d")

    @unittest.expectedFailure
    def test_a(self):
        print("test_a")

    def tearDown(self):
        print("Test2 end!")


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

 八、web测试用例实战

'''
案例：百度搜索关键词"Selenium 自学网"并打开课程页面
'''

from selenium import webdriver
import unittest
from time import sleep


class TestBaidu(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.implicitly_wait(10)
        self.driver.get("http://www.baidu.com")

    def test_baidu(self):
        driver = self.driver
        driver.find_element_by_id("kw").clear()
        driver.find_element_by_id("kw").send_keys("Selenium 自学网")
        driver.find_element_by_id("su").click()
        sleep(3)

        title = driver.title
        self.assertEqual(title, "Selenium 自学网 百度搜索")

        driver.find_element_by_partial_link_text("Selenium 自动化").click()
        sleep(5)

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


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

import unittest

test_dir = './test_case'

discovery = unittest.defaultTestLoader.discover(test_dir, pattern="test*.py")

if __name__ == '__main__':
    runner = unittest.TextTestRunner()
    runner.run(discovery)

九、测试报告生成

自动化测试执行完成之后，需要生存测试报告来查看测试结果，使用HTMLTestRunner模块可以直接生成html格式的报告。

import unittest
from HTMLTestRunner import HTMLTestRunner
import time

# 定义测试用例路径
test_dir = './test_case'

discovery = unittest.defaultTestLoader.discover(test_dir, pattern="test*.py")

if __name__ == '__main__':
    # 存放报告的文件夹
    report_dir = './test_report'
    # 报告命名时间格式化
    now = time.strftime("%Y-%m-%d %H_%M_%S")
    # 报告文件完整路径
    report_name = report_dir + '/' + now + 'result.html'

    # 打开文件在报告文件写入测试结果
    with open(report_name, "wb") as f:
        runner = HTMLTestRunner(stream=f, title="Test Report", description="Test Case Result")
        runner.run(discovery)

十、测试报告优化

import unittest
from BSTestRunner import BSTestRunner
import time

# 定义测试用例路径
test_dir = './test_case'

discovery = unittest.defaultTestLoader.discover(test_dir, pattern="test*.py")

if __name__ == '__main__':
    # 存放报告的文件夹
    report_dir = './test_report'
    # 报告命名时间格式化
    now = time.strftime("%Y-%m-%d %H_%M_%S")
    # 报告文件完整路径
    report_name = report_dir + '/' + now + 'result.html'

    # 打开文件在报告文件写入测试结果
    with open(report_name, "wb") as f:
        runner = BSTestRunner(stream=f, title="Test Report", description="Test Case Result")
        runner.run(discovery)