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)