python - 多个测试模块的设置和拆卸方法

Question

我有很多测试，分散在不同的测试模块和类中。我需要在运行测试之前/之后启动/拆除 docker 容器，但 setUpClass() 和 setUpModule() 似乎都不合适。如何为多个模块指定设置和拆卸方法？

Answer 1

为所有测试创建一个基类，在其中放置常用的 setUpClass 和 tearDownClass 方法:

class MyTestCase(unittest.TestCase):
    def setUpClass(self):
        ...set up docker stuff...

    def tearDownClass(self):
        ...tear down docker stuff...

现在所有的测试类都继承自该基类:

from mymodule.mytestcase import MyTestCase

class TestMyNiftyFeature(MyTestCase):
    def test_feature(self):
        ...test things here...

这将在每个类之前和之后运行设置/拆卸方法，这通常是您想要的，因为您希望测试是独立的，这样您就可以在一个文件中运行测试而无需在另一个文件中运行测试。