def test_apply_envelope_none(self): """ParserBase - Apply Envelope, None""" record = { 'key': 'value' } ParserBase._apply_envelope(record, {}) assert_equal(record, {'key': 'value'})
def test_apply_envelope(self): """ParserBase - Apply Envelope""" record = { 'key': 'value' } envelope = { 'env_key_01': 'value' } ParserBase._apply_envelope(record, envelope) expected_record = { 'key': 'value', 'streamalert:envelope_keys': { 'env_key_01': 'value' } } assert_equal(record, expected_record)