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)
