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...