Testing en Python: Hacer mock de una variable global

Estaba intentando probar algunas variables globales con Python, especialmente para un script el cual contiene variables globales. Y después de intentarlo y fallar, pienso que puedo mostrarte la versión más simple de hacerlo.

🏃 La solución rápida

1
# archivo_demo.py
2
ENVS = { 'valor_real': True, 'otro_valor_real': False }
3

4
# testando_tu_archivo_demo.py
5
mocker.patch("archivo_demo.ENVS", new={'MOCKED_NONE': None, 'MOCKED_VACIO': ''})

Después en tu archivo_demo la variable ENVS estará como mocked.

✏️ La solución detallada

Requerimientos

Para facilidad del trabajo. Recomiendo la siguiente librería: https://pypi.org/project/pytest-mock/

¿Qué estaba intentando hacer?

Primero, supongamos que tengo un archivo llamado archivo_demo.py con el siguiente contenido (y donde la variable global está localizada).

1
# archivo_demo.py
2
import os
3

4
ENVS = { 'valor_real': os.environ('VALOR_REAL'), 'otro_valor_real': os.environ('OTRO_VALOR_REAL') }
5

6
def metodo_simple():
7
		print(ENVS)
8
		return 0

Si tratas de ejecutar metodo_simple va a mostrar el diccionario con valor_real y otro_valor_real allí.

Ahora tenemos el siguiente archivo de test.

1
# testeando_mis_envs.py
2
from archivo_demo import metodo_simple
3

4
def testear_si_metodo_simple_funciona():
5
		assert metodo_simple() == 0

Ya sé que el test se ve simple, pero se asegura al menos de retornar un valor cero después de mostrar los valores ENVs.

👾 Sobreescribiendo y simulando (hacer un mock de) las variables

Pero como puedes ver, estamos llamando las variables de entorno de la computadora que usemos, y si tú escribes tus tests, esperando tenerlas en el servidor CI/CD, entonces los tests comenzarán a fallar. ¿Cómo podemos evitar eso y decirle a nuestro servidor de testing (o pipeline de testing) que queremos valores fijos?

Si instalaste la librería mencionada arriba, en la sección de requerimientos, es muy fácil comenzar a mockear valores. Así que solamente necesitamos agregar una sola línea.

1
# testeando_mis_envs.py
2
from archivo_demo import metodo_simple
3

4
@mocker.patch("archivo_demo.ENVS", new={'VALOR_REAL_MOCKED': None, 'OTRO_VALOR_REAL_MOCKED': ''})
5
def testear_si_metodo_simple_funciona():
6
		assert metodo_simple() == 0

Y tus tests no ejecutarán la variable ENVS y, por lo tanto, tampoco llamarán a os.environ.

¿Correcciones? ¿Recomendaciones? Click aquí para editar esta página en Github

Mis posts no son generados por la IA, sin embargo, podrían estar corregidos por ella. El primer borrador siempre es de mi creación

En otros lenguajes

Autor

Escrito por Helmer Davila

Testing en Python: Hacer mock de una variable global

🏃 La solución rápida

✏️ La solución detallada

Requerimientos

¿Qué estaba intentando hacer?

👾 Sobreescribiendo y simulando (hacer un mock de) las variables

Tags

En otros lenguajes

Faire un mock pour une variable globale dans les tests de Python

Mocking global variables in Python tests

Fazer mocks de variáveis globais em testes Python

Autor