Modificación de variables globales en Python unittest framework

Question

Estoy trabajando en una serie de pruebas unitarias en Python, algunas de las cuales dependen del valor de una variable de configuración. Estas variables se almacenan en un archivo de configuración global de Python y se usan en otros módulos. Me gustaría escribir pruebas unitarias para diferentes valores de las variables de configuración, pero todavía no he encontrado la manera de hacerlo.

No tengo la posibilidad de volver a escribir las firmas de los métodos que estoy probando.

Esto es lo que me gustaría lograr:

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") 

Gracias.

Editar: Hizo que el código de ejemplo sea más explícito.

Answer 1

No hagas esto:

from my_module import my_function_with_global_var 

Pero esto:

import my_module 

y luego se puede inyectar en el MY_CONFIG_VARIABLEmy_module importados, sin cambiar el sistema que se está probando de esta manera:

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") 

Hice algo similar en my answer toHow can I simulate input to stdin for pyunit?.

Answer 2

El código importa MY_CONFIG_VARIABLE en el ámbito local e inmediatamente sobrescribe el nombre con un objeto diferente. Eso no cambiará el valor en el módulo config. Pruebe

import config 
config.MY_CONFIG_VARIABLE = False 

en su lugar.

Answer 3

Probablemente quiera burlarse de esas variables globales. La ventaja de esto es que los globales se reinician una vez que hayas terminado. Python se envía con un módulo de burla que le permite hacer esto.

unittest.mock.patch usarse como decorador:

class TestSomething(self.unittest): 

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

También puede utilizarlo como un gestor de contexto:

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