def test_rescue_generic_exception__no_args(self, echo_mock):
dummy_ebcli_app = MagicMock()
dummy_ebcli_app.setup = MagicMock(
side_effect=TestEbRun.MyDummyGenericException(''))
ebrun.run_app(dummy_ebcli_app)
echo_mock.assert_called_with(
io.bold(io.color('red', 'ERROR: MyDummyGenericException')))
dummy_ebcli_app.close.assert_called_once_with(code=4)
def test_rescue_generic_exception__debug_mode(self, traceback_mock,
echo_mock):
dummy_ebcli_app = MagicMock()
dummy_ebcli_app.setup = MagicMock(
side_effect=TestEbRun.MyDummyGenericException(
'My Exception Message'))
with patch.object(sys, 'argv', ['--debug']):
ebrun.run_app(dummy_ebcli_app)
echo_mock.side_effect = [
traceback_mock, 'INFO: My Exception Message'
]
dummy_ebcli_app.close.assert_called_once_with(code=4)
def test_rescue_EBCLIException__without_verbose_or_debug_flag(
self, echo_mock):
dummy_ebcli_app = MagicMock()
dummy_ebcli_app.setup = MagicMock(
side_effect=TestEbRun.MyDummyEBCLIException(
'My Exception Message'))
ebrun.run_app(dummy_ebcli_app)
echo_mock.assert_called_with(
io.bold(
io.color(
'red', 'ERROR: {}'.format(
'MyDummyEBCLIException - My Exception Message'))))
dummy_ebcli_app.close.assert_called_once_with(code=4)
def test_rescue_EBCLIException__with_verbose_flag(self, traceback_mock,
echo_mock):
dummy_ebcli_app = MagicMock()
dummy_ebcli_app.setup = MagicMock(
side_effect=TestEbRun.MyDummyEBCLIException(
'My Exception Message'))
with patch.object(sys, 'argv', ['--verbose']):
sys.argv.append('--verbose')
ebrun.run_app(dummy_ebcli_app)
echo_mock.side_effect = [
traceback_mock, 'INFO: My Exception Message'
]
dummy_ebcli_app.close.assert_called_once_with(code=4)
def test_rescue_AttributeError(self, echo_mock):
dummy_ebcli_app = MagicMock()
dummy_ebcli_app.setup = MagicMock(side_effect=AttributeError(
'This is my error', 'This is my error as well'))
ebrun.run_app(dummy_ebcli_app)
echo_mock.assert_called_with(
io.bold(
io.color(
'red',
"ERROR: {error_type} - ('{argument_1}', '{argument_2}')".
format(error_type='AttributeError',
linesep=os.linesep,
argument_1='This is my error',
argument_2='This is my error as well'))))
dummy_ebcli_app.close.assert_called_once_with(code=4)
def main():
app = EBP()
ebrun.run_app(app)
def main():
app = EBP()
ebrun.run_app(app)
