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'})