在 Python unittest 框架中修改全局变量

Question

我正在使用 Python 进行一系列单元测试，其中一些取决于配置变量的值。这些变量存储在全局 Python 配置文件中，并在其他模块中使用。我想为配置变量的不同值编写单元测试，但还没有找到方法。

我无法重写我正在测试的方法的签名。

这就是我想要实现的目标：

from my_module import my_function_with_global_var

class TestSomething(self.unittest):

    def test_first_case(self):
         from config import MY_CONFIG_VARIABLE
         MY_CONFIG_VARIABLE = True
         self.assertEqual(my_function_with_global_var(), "First result")

    def test_second_case(self):
         from config import MY_CONFIG_VARIABLE
         MY_CONFIG_VARIABLE = False
         self.assertEqual(my_function_with_global_var(), "Second result")

谢谢。

编辑：使示例代码更加明确。

Answer 1

您可能希望改为模拟这些全局变量。这样做的好处是，一旦你完成，全局变量就会被重置。 Python 附带了一个模拟模块，可让您执行此操作。

unittest.mock.patch 用作装饰器：

class TestSomething(self.unittest):

    @patch('config.MY_CONFIG_VARIABLE', True)
    def test_first_case(self):
         self.assertEqual(my_function_with_global_var(), "First result")

您也可以将其用作上下文管理器：

    def test_first_case(self):
        with patch('config.MY_CONFIG_VARIABLE', True):
            self.assertEqual(my_function_with_global_var(), "First result")

Answer 2

使用unittest.mock.patch，如@Flimm's answer，如果您可以使用的话。

---

原答案

不要这样做：

from my_module import my_function_with_global_var

但是这个：

import my_module

然后您可以将MY_CONFIG_VARIABLE 注入到导入的my_module 中，而无需像这样更改被测系统：

class TestSomething(unittest.TestCase): # Fixed that for you!

    def test_first_case(self):
         my_module.MY_CONFIG_VARIABLE = True
         self.assertEqual(my_module.my_function_with_global_var(), "First result")

    def test_second_case(self):
         my_module.MY_CONFIG_VARIABLE = False
         self.assertEqual(my_module.my_function_with_global_var(), "Second result")

我在my answer toHow can I simulate input to stdin for pyunit?做了类似的事情。

Answer 3

您将MY_CONFIG_VARIABLE 代码导入本地范围，然后立即用不同的对象覆盖名称。这不会改变 config 模块中的值。试试

import config
config.MY_CONFIG_VARIABLE = False

改为。