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()

巴特西

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

最新文章

热门文章