让一个继承unittest.TestCase的类下的setUp和tearDown只执行一次

古城微笑少年丶 2022-02-03 08:43 138阅读 0赞

原文链接：[https://www.cnblogs.com/UncleYong/p/7076872.html][https_www.cnblogs.com_UncleYong_p_7076872.html]

【问题】：

在unittest单元测试框架中，执行继承了unittest.TestCase的类下每个test开头的方法（就是用例）时，都会执行setUp和tearDown（代码如下），很麻烦。

能否让setup和teardown只执行一次？

import unittest
     
    class TestSetupTeardown(unittest.TestCase):
        def setUp(self):
            print('连接数据库成功...')
        def tearDown(self):
            print('关闭数据库。')
     
        def test_a(self):
            print('test_a')
     
        def test_b(self):
            print('test_b')
     
    if __name__ == '__main__':
        unittest.main()

以上代码执行效果如下：

setUp连接数据库，tearDown关闭数据库，这样反复执行，无疑是会增加数据库服务器资源的损耗，且浪费时间

![20190509081801652.png][]

【解决方案】：

将setup()  和teardown() 换成setUpClass()和tearDownClass()

【注意事项】：

setUpClass():必须使用@classmethod 装饰器,  所有case运行之前只运行一次  
tearDownClass():必须使用@classmethod装饰器, 所有case运行完之后只运行一次

代码与执行效果如下：

import unittest
     
    class TestSetupTeardown(unittest.TestCase):
        @classmethod
        def setUpClass(cls):
            print('连接数据库成功...')
        @classmethod
        def tearDownClass(cls):
            print('关闭数据库。')
     
        def test_a(self):
            print('test_a')
     
        def test_b(self):
            print('test_b')
     
    if __name__ == '__main__':
        unittest.main()

![2019050908254050.png][]

[https_www.cnblogs.com_UncleYong_p_7076872.html]: https://www.cnblogs.com/UncleYong/p/7076872.html
[20190509081801652.png]: /images/20220203/4578ed2d65b4472a8f504252479212db.png
[2019050908254050.png]: /images/20220203/e9a45ecdb7ab48f4887196ce7fbcbac4.png