Python单元测试框架-unittest

参考
https://docs.python.org/2/library/unittest.html#organizing-tests

import unittest

class TestStringMethods(unittest.TestCase):

  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'])
      # check that s.split fails when the separator is not a string
      with self.assertRaises(TypeError):
          s.split(2)

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

首先是一个基本的例子，
这个类中有三个方法，以test开头，这种命名会作为测试用例。会通知runner

assertEqual() 检查相等
assertTrue() 检查条件为真
assertFalse() 检查条件为假
assertRaises() 检查会抛出某种异常

这里例子虽然简单，其中特性可以满足我们每天进行的测试了。

做下面的命令行测试时候，请注意Python的版本 ，2.7.10的不报错，可是2.6.6就会报错。

[root@-01-VM-back-test- ut]# /usr/local/bin/python -m unittest test.TestStringMethod.test_upper
test_isupper (test.TestStringMethod) ... ok
test_split (test.TestStringMethod) ... ok
test_upper (test.TestStringMethod) ... ok

----------------------------------------------------------------------
Ran 3 tests in 0.000s

OK
.
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK

[root@back-test- ut]# python -m unittest discover
...
----------------------------------------------------------------------
Ran 3 tests in 0.000s

OK

测试用例case是基本的测试块， 而测试用例 又是TestCase的实例，

最基本的 TestCase 类必须重写runTest()方法

import unittest

class DefaultWidgetSizeTestCase(unittest.TestCase):
    def runTest(self):
        widget = Widget('The widget')
        self.assertEqual(widget.size(), (50, 50), 'incorrect default size')

在上个例子里，每次写测试用例，都需要 自己 先构造一个widget很麻烦， unittest提供了 setUp()方法，来帮助我们自动地在每次运行测试用例前，做一下准备工作，像下面这样：

import unittest

class SimpleWidgetTestCase(unittest.TestCase):
    #准备工作
    def setUp(self):
        self.widget = Widget('The widget')

    #清理工作    
    def tearDown(self):
        self.widget.dispose()
        self.widget = None

#01
class DefaultWidgetSizeTestCase(SimpleWidgetTestCase):
    def runTest(self):
        self.assertEqual(self.widget.size(), (50,50),
                         'incorrect default size')
#02
class WidgetResizeTestCase(SimpleWidgetTestCase):
    def runTest(self):
        self.widget.resize(100,150)
        self.assertEqual(self.widget.size(), (100,150),
                         'wrong size after resize')

在写01 ，02 两个测试用例时候，我们继承了SimpleWidgetTestCase 这个类，在SimpleWidgetTestCase这个类中有setUp()方法，来帮我们自动的每次做好准备的工作。

类似的，有一个tearDown() 方法，来做每次测试后的清理工作。

这样的测试环境，在unittest中叫测试运行的fixture， 类似上下文。

但是如果这样的测试用例很多 ，SimpleWidgetTestCase 这个类就要被重载许多次，这很累，所以下面这个机制，可以省一点时间。就是把这些都写在一个类里面。

import unittest

class WidgetTestCase(unittest.TestCase):
    def setUp(self):
        self.widget = Widget('The widget')

    def tearDown(self):
        self.widget.dispose()
        self.widget = None

    def test_default_size(self):
        self.assertEqual(self.widget.size(), (50,50),
                         'incorrect default size')

    def test_resize(self):
        self.widget.resize(100,150)
        self.assertEqual(self.widget.size(), (100,150),
                         'wrong size after resize')

这里我们没有提供runTest() 的方法，没关系，实例会运行每个test_*()的方法。所以只要名字写的规范就可以了。

跳过测试用例,使用装饰器@unittest.skip()，如下：

class MyTestCase(unittest.TestCase):

    @unittest.skip("demonstrating skipping")
    def test_nothing(self):
        self.fail("shouldn't happen")

    @unittest.skipIf(mylib.__version__ < (1, 3),
                     "not supported in this library version")
    def test_format(self):
        # Tests that work for only a certain version of the library.
        pass

    @unittest.skipUnless(sys.platform.startswith("win"), "requires Windows")
    def test_windows_support(self):
        # windows specific testing code
        pass

也可以跳过整个类。