Suggest to check `calledWith` to ensure the function is called with exactly the message object.