在Python單元測試框架中修改全局變量

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

不要這樣做：

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 2

您將代碼導入MY_CONFIG_VARIABLE到本地作用域中，然後立即用不同的對象覆蓋該名稱。這不會改變config模塊中的值。嘗試

import config 
config.MY_CONFIG_VARIABLE = False 

改爲。

Answer 3

您可能想要模擬這些全局變量。這樣做的好處是，一旦你完成了全局變量重置。 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")