python - 单元测试时重新加载当前模块

Question

运行简单的单元测试时，有时将测试保留在类中会更容易。但是，我不知道如何重新加载当前模块，因此无论何时需要，我都必须将测试移到一个单独的模块中。有解决办法吗？

module: foo
import unittest

class MyObject
...

class MockMyObject
...

class TestMock(unittest.TestCase):

    def setUp(self):
        MyObject = MockMyObject
        mocked = MyObject()

    def tearDown(self):
        reload(foo) # what goes here?

    def testFunction(self):
        mocked.do_mocked_function()

if __name__ == "__main__":
    unittest.main()

我发现处理这个问题的方法是在 tearDown 中 import sys 和 reload(sys.modules[__name__])方法，但我想知道是否有更好的方法。

Answer 1

您可以将原始类保存在一个变量中，然后在 tearDown 函数中恢复它。
这是一个例子:

class TestMock(unittest.TestCase):

    original = MyObject

    def setUp(self):
        global MyObject
        MyObject = MockMyObject

    def tearDown(self):
        global MyObject
        MyObject = TestMock.original

    def testFunction(self):
        MyObject().do_mocked_function()