Spying on Python functions
I was writing test code for a Python project, and needed to spy on the functions of one of the main object’s, called “server”, object instances called “rest_client”. By simply replacing the “rest_client” instance with a mock that passes the actual call to the wrapped object, I’m able to execute the production code of “rest_client” while later on making assertions.
|
1 2 3 4 5 6 |
server = Server() # The constructor runs "rest_client = RestClient() server.rest_client = mock.Mock(wraps=server.rest_client) server.do_something() server.rest_client.some_method.assert_called_once_with(my_parameter="some value") |