python - unittest中的setUp与setUpClass

其实我也没想到，有一天我会记录python的问题。。

事情是这样的，最近在支持python SDK，遇到了一件非常神奇的事

test class中使用到了一个简单的Callable，定义在setUpClass中，如下

class TestClass(unittest.TestCase):
    @classmethod
    def setUpClass(cls) -> None:
        cls.func = get_a_func()
    
    def test(self):
        self.func(param1=xxx)


if __name__ == '__main__':
    unittest.main(testRunner=xmlrunner.XMLTestRunner())

这样使用，非常开心（并不）的获得了一个异常，是这样的TypeError: func() got multiple values for argument 'param1' 

原因是，使用setUpClass时，cls.func实际上将func定义成了自己的成员，所以在使用self.func(param1=xxx)时实际上把self也作为参数传走了，像这样func(self, param1=xxx)，所以会出现multiple values的报错。

正确的方法是使用setUp替换setUpClass，如下

class TestClass(unittest.TestCase):
    def setUp(self) -> None:
        self.func = get_a_func()
    
    def test(self):
        self.func(param1=xxx)


if __name__ == '__main__':
    unittest.main(testRunner=xmlrunner.XMLTestRunner())

这样就搞定啦~ （虽然记起来很简单，但确实困扰了我一俩小时Orz...