def get_button(self, t_mock, **kwargs): """Proxy for calling install_button.""" template_mock = Mock() t_mock.return_value = template_mock install_button(self.context, self.addon, **kwargs) # Extract button from the kwargs from the first call. return template_mock.render.call_args[0][0]['button']
def render(self, **kwargs): return PyQuery(install_button(self.context, self.addon, **kwargs))
def get_button(self, render_mock, **kwargs): """Proxy for calling install_button.""" install_button(self.context, self.addon, **kwargs) # Extract button from the kwargs from the first call. return render_mock.call_args[0][1]['button']