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)