python利用unittest测试框架组织测试用例的5种方法

利用unittest测试框架可以编写测试用例，执行方式分两大类：利用main方法和利用testsuite，其中利用测试套件来组织测试用例可以有4种写法。

在此之前，先了解几个概念

• TestCase：所有测试用例的基本类，给一个测试方法的名字，就会返回一个测试用例实例；
• TestSuite：组织测试用例的实例，支持测试用例的添加和删除，最终将传递给  testRunner进行测试执行；
• TextTestRunner：进行测试用例执行的实例，其中Text的意思是以文本形式显示测试结果。测试的结果会保存到TextTestResult实例中，包括运行了多少测试用例，成功了多少，失败了多少等信息；
• TestLoader：用来加载TestCase到TestSuite中的，其中有几个  loadTestsFrom__()方法，就是从各个地方寻找TestCase，创建它们的实例，然后add到TestSuite中，再返回一个TestSuite实例；
• defaultTestLoader.discover：将discover方法筛选出来的用例，循环添加到测试套件中,打印出的用例信息会递增

方法一：通过unittest.main()来执行测试用例的方式

import unittest

class TestStringMethods(unittest.TestCase):

    def setUp(self):
        unittest.TestCase.setUp(self)

    def tearDown(self):
        unittest.TestCase.tearDown(self)

    def test_upper(self):
        self.assertEqual('foo'.upper(), 'FOO')

    def test_isupper(self):
        self.assertTrue('FOO'.isupper())
        self.assertFalse('Foo'.isupper())

    def test_split(self):
        s = 'hello world'
        self.assertEqual(s.split(), ['hello','world'])
        with self.assertRaises(TypeError):
            s.split(2)

if __name__ == "__main__":
    #import sys;sys.argv = ['', 'Test.testName']
     unittest.main()

unittest.main()会去查找所有以test开头的方法，并把它们当做一个个测试用例

方法二：用测试套件

import unittest

class TestStringMethods(unittest.TestCase):

    def setUp(self):
        unittest.TestCase.setUp(self)

    def tearDown(self):
        unittest.TestCase.tearDown(self)

    def test_upper(self):
        self.assertEqual('foo'.upper(), 'FOO')

    def test_isupper(self):
        self.assertTrue('FOO'.isupper())
        self.assertFalse('Foo'.isupper())

    def test_split(self):
        s = 'hello world'
        self.assertEqual(s.split(), ['hello','world'])
        with self.assertRaises(TypeError):
            s.split(2)

def suite():

    suite = unittest.TestSuite()
    suite.addTest(TestStringMethods('test_upper'))
    suite.addTest(TestStringMethods('test_isupper'))
    unittest.TextTestRunner(verbosity=2).run(suite)

if __name__ == "__main__":

    suite()
    #import sys;sys.argv = ['', 'Test.testName']
#     unittest.main()

方法三：用TestLoader

import unittest
class TestCase1(unittest.TestCase):
    #def setUp(self):
    #def tearDown(self):
    def testCase1(self):
        print 'aaa'
    def testCase2(self):
        print 'bbb'

class TestCase2(unittest.TestCase):
    #def setUp(self):
    #def tearDown(self):
    def testCase1(self):
        print 'aaa1'
    def testCase2(self):
        print 'bbb1'

if __name__ == "__main__":
    #此用法可以同时测试多个类
    suite1 = unittest.TestLoader().loadTestsFromTestCase(TestCase1)
    suite2 = unittest.TestLoader().loadTestsFromTestCase(TestCase2)
    suite = unittest.TestSuite([suite1, suite2])
    unittest.TextTestRunner(verbosity=2).run(suite)

方法四：测试套件的一种变性，和方法二差不多

import unittest

class TestStringMethods(unittest.TestCase):

    def setUp(self):
        unittest.TestCase.setUp(self)

    def tearDown(self):
        unittest.TestCase.tearDown(self)

    def test_upper(self):
        self.assertEqual('foo'.upper(), 'FOO')

    def test_isupper(self):
        self.assertTrue('FOO'.isupper())
        self.assertFalse('Foo'.isupper())

    def test_split(self):
        s = 'hello world'
        self.assertEqual(s.split(), ['hello','world'])
        with self.assertRaises(TypeError):
            s.split(2)

def suite():

     tests = ['test_upper','test_isupper']
     suite = unittest.TestSuite(map(TestStringMethods,tests))
     unittest.TextTestRunner(verbosity=2).run(suite)

if __name__ == "__main__":

    suite()
    #import sys;sys.argv = ['', 'Test.testName']
#     unittest.main()

方法五：用discover

import unittest

class TestStringMethods(unittest.TestCase):

    def setUp(self):
        unittest.TestCase.setUp(self)

    def tearDown(self):
        unittest.TestCase.tearDown(self)

    def test_upper(self):
        self.assertEqual('foo'.upper(), 'FOO')

    def test_isupper(self):
        self.assertTrue('FOO'.isupper())
        self.assertFalse('Foo'.isupper())

    def test_split(self):
        s = 'hello world'
        self.assertEqual(s.split(), ['hello','world'])
        with self.assertRaises(TypeError):
            s.split(2)

def suite():

     suite=unittest.TestSuite()
     discover=unittest.defaultTestLoader.discover(casepath, pattern='test_*.py', top_level_dir=None)
     #将discover方法筛选出来的用例，循环添加到测试套件中,打印出的用例信息会递增
     for test_case in discover:
             suite.addTests(test_case)
     print(suite)
     return suite

if __name__ == "__main__":

    all_test_cases = suite()
    unittest.TextTestRunner().run(all_test_cases)
    #import sys;sys.argv = ['', 'Test.testName']
#     unittest.main()