python - nosetest 的设置方法。 (测试类)

Question

我在进行的一些测试中模拟了一个数据库。我如何为整个类创建一个设置方法，以便它在每次运行类中的单个测试时运行？

我正在尝试做的事情的例子。

from mocks import MockDB

class DBTests(unittest.TestCase):

    def setup(self):
        self.mock_db = MockDB()

    def test_one(self):
        #deal with self.mock_db

    def test_two(self):
        #deal with self.mock_db, as if nothing from test_one has happened

我假设拆解方法也是可能的，但我找不到可以执行此类操作的文档。

Answer 1

如果您使用的是 Python 单元测试框架，那么这就是您想要的:

class Test(unittest.TestCase):


    def setUp(self):
        self.mock_db = MockDB()

    def tearDown(self):
        pass  # clean up 

    def test_1(self):
        pass  # test stuff

Documentation