def test_pre_parse_s3_debug(s3_mock, log_mock, _):
"""S3Payload - Pre Parse, Debug On"""
# Cache the logger level
log_level = LOGGER.getEffectiveLevel()
# Increase the logger level to debug
LOGGER.setLevel(logging.DEBUG)
records = ['_first_line_test_' * 10, '_second_line_test_' * 10]
s3_mock.side_effect = [((100, records[0]), (200, records[1]))]
raw_record = make_s3_raw_record('unit_bucket_name', 'unit_key_name')
s3_payload = load_stream_payload('s3', 'unit_key_name', raw_record)
S3Payload.s3_object_size = 350
_ = [_ for _ in s3_payload.pre_parse()]
calls = [
call(
'Processed %s S3 records out of an approximate total of %s '
'(average record size: %s bytes, total size: %s bytes)', 100, 350,
1, 350),
call(
'Processed %s S3 records out of an approximate total of %s '
'(average record size: %s bytes, total size: %s bytes)', 200, 350,
1, 350)
]
log_mock.assert_has_calls(calls)
# Reset the logger level and stop the patchers
LOGGER.setLevel(log_level)
def test_run_debug_log_alert(self, extract_mock, rules_mock, log_mock):
"""StreamAlert Class - Run, Debug Log Alert"""
extract_mock.return_value = ('kinesis', 'unit_test_default_stream')
rules_mock.return_value = ['success!!']
# Cache the logger level
log_level = LOGGER.getEffectiveLevel()
# Increase the logger level to debug
LOGGER.setLevel(logging.DEBUG)
self.__sa_handler.run(get_valid_event())
# Reset the logger level
LOGGER.setLevel(log_level)
log_mock.assert_called_with('Alerts:\n%s', '[\n "success!!"\n]')
def test_run_debug_log_alert(self, extract_mock, rules_mock, alerts_mock,
log_mock):
"""StreamAlert Class - Run, Debug Log Alert"""
extract_mock.return_value = ('kinesis', 'unit_test_default_stream')
rules_mock.return_value = ([Alert('rule_name', {},
{'output'})], ['normalized_records'])
alerts_mock.return_value = []
# Cache the logger level
log_level = LOGGER.getEffectiveLevel()
# Increase the logger level to debug
LOGGER.setLevel(logging.DEBUG)
self.__sa_handler.run(get_valid_event())
# Reset the logger level
LOGGER.setLevel(log_level)
log_mock.assert_called_with('Alerts:\n%s', ANY)
