python unittest 使用记录

在使用python写自动化的时候，选择使用了unittest运行测试用例，在此记录使用unittest的常规用法

参考1：http://www.cnblogs.com/idbeta/archive/2015/12/11/5040187.html

参考2：http://www.2cto.com/kf/201409/330721.html

初始化环境

dome1：

每次执行用例时都会: setUp->test_case->tearDown ;也就是说可以使用setUp和tearDown做每个用例的环境初始化

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import unittest

class MyTestCase(unittest.TestCase):
# 预置环境
def setUp(self):
print '开始执行'

# 清理环境
def tearDown(self):
print '结束执行'

# 测试用例
def test_case1(self):
print '用例1'

# 测试用例
def test_case2(self):
print '用例2'

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

dome2：

每个class用一次，setUpClass->test_case1->test_case2->tearDownClass;也就是说使用setUpClass、tearDownClass可以做所有用例的前置初始化和结束

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import unittest

class MyTestCase(unittest.TestCase):
@classmethod
def setUpClass(cls):
print '开始执行'

@classmethod
def tearDownClass(cls):
print '结束执行'

# 测试用例
def test_case1(self):
print '用例1'

# 测试用例
def test_case2(self):
print '用例2'

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

dome3

setUp,tearDown,setUpClass,tearDownClass可以组合使用也可单独使用

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import unittest

class MyTestCase(unittest.TestCase):

# @classmethod
# def setUpClass(cls):
#     print '首先开始执行'

@classmethod
def tearDownClass(cls):
print '最后结束执行'

def setUp(self):
print '每次首先执行1'

# def tearDown(self):
#     print '每次结束执行1'

# 测试用例
def test_case1(self):
print '用例1'

# 测试用例
def test_case2(self):
print '用例2'

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

dome4：

整个文件使用一次，主要是针对一个文件中存在多个class

setUpModule()->class1->setUpClass->setUp->test_case->tearDown->tearDownClass->class2...->tearDownModule()

# !/usr/bin/env python
# -*- coding: utf-8 -*-
import unittest

def setUpModule():
print '1setUpModule'

def tearDownModule():
print '1tearDownModule'

class MyTestCase1(unittest.TestCase):
@classmethod
def setUpClass(cls):
print"1setUpClass"

@classmethod
def tearDownClass(cls):
print"1tearDownClass"

def setUp(self):
print '1setUp'

def tearDown(self):
print '1tearDown'

def test_case1(self):
print '1test_case1'

def test_case2(self):
print '1test_case2'

class MyTestCase2(unittest.TestCase):
@classmethod
def setUpClass(cls):
print"2setUpClass"

@classmethod
def tearDownClass(cls):
print"2tearDownClass"

def setUp(self):
print '2setUp'

def tearDown(self):
print '2tearDown'

def test_case1(self):
print '2test_case1'

def test_case2(self):
print '2test_case2'

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

unittest suite加载testcase的几种方法

dome1

def suite():
suite = unittest.TestSuite()
suite.addTest(WidgetTestCase('test_default_size'))
suite.addTest(WidgetTestCase('test_resize'))
return suite

dome2

def suite():
tests = ['test_default_size', 'test_resize']
return unittest.TestSuite(map(WidgetTestCase, tests))

dome3

多个测试套件构建成更大的测试套件

suite1 = module1.TheTestSuite()
suite2 = module2.TheTestSuite()
alltests = unittest.TestSuite([suite1, suite2])

dome4

默认的测试套件

suite = unittest.TestLoader().loadTestsFromTestCase(WidgetTestCase)

dome5

按给定的文件名称执行测试用例

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

构造测试用例集

def suite1():
# 构造测试用例集
suite = unittest.TestSuite()
# 加载测试用例
suite.addTest(test_class.TestCases('test_baidu_search'))
suite.addTest(test_class.TestCases('test_search2'))
suite.addTest(test_class.TestCases('test_get_title'))
return suite

def suites():
# 构造测试用例集
t_suite = unittest.TestSuite()
# 加载一个类文件下所有的测试用例
suite = t_suite(unittest.makeSuite(test_class.TestCases))
return suite

def suite2():
# 测试用例存放目录
test_dir = './'
# 加载test_dir路径下所有以test_开头.py结尾的文件中的测试用例
suite = unittest.TestLoader().discover(test_dir, pattern='test_*.py')
return suite