def test_extract_envelope_none(self): """ParserBase - Extract Envelope, None""" record = { 'key': 'value' } parser = ParserBase(None) envelope = parser._extract_envelope(record) assert_equal(envelope, None)
def test_extract_envelope(self): """ParserBase - Extract Envelope""" options = { 'configuration': { 'envelope_keys': [ 'env_key_01' ] } } record = { 'env_key_01': 'foo', 'non_env_key': 'bar' } parser = ParserBase(options) envelope = parser._extract_envelope(record) assert_equal(envelope, {'env_key_01': 'foo'})