Cómo hacer pytest para mostrar que un método ha sido llamado con argumentos de tipos específicos?

Answer

1. 

   davy.ai

   August 4, 2023 at 6:58 pm

   Para probar que self.client.post ha sido llamado pasando data=query.__dict__ en lugar de data=query, se puede utilizar el objeto MagicMock del módulo unittest.mock.

   Primero, puedes crear un método de prueba que configure el objeto simulado (mock object) y la instancia de myClass:

   import unittest.mock as mock
   
   def test_myMethod():
       # Crear una simulación de myClient
       client_mock = mock.MagicMock()
   
       # Crear una instancia de myClass usando la simulación de myClient
       my_class_instance = myClass()
       my_class_instance.client = client_mock
   
       # Llamar a myMethod
       my_class_instance.myMethod()
   

   Luego, puedes usar el método MagicMock.assert_called_with() para confirmar que self.client.post ha sido llamado con data=query.__dict__. Para hacer esto, modificarías el método de prueba para añadir la confirmación:

   def test_myMethod():
       # Crear una simulación de myClient
       client_mock = mock.MagicMock()
   
       # Crear una instancia de myClass usando la simulación de myClient
       my_class_instance = myClass()
       my_class_instance.client = client_mock
   
       # Llamar a myMethod
       my_class_instance.myMethod()
   
       # Confirmar que self.client.post fue llamado con data=query.__dict__
       query = Query()
       client_mock.post.assert_called_with(data=query.__dict__)
   

   Este método de prueba verificaría que self.client.post fue llamado con data=query.__dict__ en lugar de data=query.

Comments are closed.