unittest系统如何在一个测试类多个测试用例执行中只初始化和清理一次？

Posted 2021-08-29 leiziv5

之前分享了一系列的文章，分别从原理，运行，断言，执行，测试套件，如何跳过用例来讲解unittest，那么我们继续分享

我们首先看下下面的代码

import  unittest
class TestDemo(unittest.TestCase):
    
    def setUp(self):
        print("start")
    def tearDown(self):
        print("end")
    def testone(self):
        self.assertEqual(1,1)
        self.assertFalse(False)

    def testtwo(self):
        self.assertEqual(1, 1)
        self.assertFalse(False)
if __name__=="__main__":
    unittest.main()

我们执行下，看下结果

我们发现在初始化的时候呢，我们每次都会初始化，但是在实际的测试中呢，我们可能会是有些参数或者动作只需要做一次即可，那么我们只能在用例中初始化一次。那么unittest里面有没有这样的方法 呢？答案是肯定呢，

 @classmethod
    def  setUpClass(cls):
        pass

    @classmethod
    def tearDownClass(cls):
        pass

我们只需要在初始化一次的时候调用setUpClass和tearDownClass 即可。我们将这些代码带入到我们的测试用例中，看下效果会如何。

import  unittest
class TestDemo(unittest.TestCase):

    @classmethod
    def  setUpClass(cls):
        print("one start")

    @classmethod
    def tearDownClass(cls):
        print("one end")
    def setUp(self):
        print("start")
    def tearDown(self):
        print("end")
    def testone(self):
        self.assertEqual(1,1)
        self.assertFalse(False)

    def testtwo(self):
        self.assertEqual(1, 1)
        self.assertFalse(False)
if __name__=="__main__":
    unittest.main()

我们执行下看下最后的效果

这样我们就实现了初始化一次，清理一次的需求。

欢迎关注我的个人公众号