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)