def test_log_result_no_exceptions(): from shadow.proxy.web import ProxyFlask from shadow.proxy.web import AbstractResultsLogger svc = mock.Mock() rs_log = mock.Mock(spec=AbstractResultsLogger) rs_log.log_result.side_effect = Exception('Test exception') app = ProxyFlask(svc, ['localhost'], ['localhost'], result_loggers=[rs_log]) app.log_result('hello') assert (rs_log.log_result.called) eq_(rs_log.log_result.call_count, 1) rs_log.log_result.assert_called_with('hello')
def test_log_result(): from shadow.proxy.web import ProxyFlask from shadow.proxy.web import AbstractResultsLogger svc = mock.Mock() class MockResultLogger(AbstractResultsLogger): pass rs_log = MockResultLogger() rs_log.log_result = mock.Mock() app = ProxyFlask(svc, ['localhost'], ['localhost'], result_loggers=[rs_log, rs_log]) app.log_result('hello') assert (rs_log.log_result.called) eq_(rs_log.log_result.call_count, 2) eq_(rs_log.log_result.call_args_list, [call('hello'), call('hello')])
def test_log_result_no_exceptions(): from shadow.proxy.web import ProxyFlask from shadow.proxy.web import AbstractResultsLogger svc = mock.Mock() rs_log = mock.Mock(spec=AbstractResultsLogger) rs_log.log_result.side_effect = Exception('Test exception') app = ProxyFlask( svc, ['localhost'], ['localhost'], result_loggers=[rs_log] ) app.log_result('hello') assert(rs_log.log_result.called) eq_(rs_log.log_result.call_count, 1) rs_log.log_result.assert_called_with('hello')
def test_log_result(): from shadow.proxy.web import ProxyFlask from shadow.proxy.web import AbstractResultsLogger svc = mock.Mock() class MockResultLogger(AbstractResultsLogger): pass rs_log = MockResultLogger() rs_log.log_result = mock.Mock() app = ProxyFlask( svc, ['localhost'], ['localhost'], result_loggers=[rs_log, rs_log] ) app.log_result('hello') assert(rs_log.log_result.called) eq_(rs_log.log_result.call_count, 2) eq_(rs_log.log_result.call_args_list, [call('hello'), call('hello')])