鼻子测试的设置方法。（测试班）

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

文档