def test_confirm_raises_own_exception(self):
confirm = mock.Mock(spec=click.confirm)
confirm.side_effect = click.Abort()
ui = utils.UI(self.out, self.err, confirm)
with pytest.raises(utils.AbortedError):
ui.confirm("Confirm?")
def test_confirm_returns_value(self):
confirm = mock.Mock(spec=click.confirm)
confirm.return_value = 'foo'
ui = utils.UI(self.out, self.err, confirm)
return_value = ui.confirm("Confirm?")
assert return_value == 'foo'
def setup(self):
self.out = StringIO()
self.err = StringIO()
self.ui = utils.UI(self.out, self.err)
