def test_read_downloaded_object_fallback(self): """S3Payload - Read Downloaded Object, Fallback""" value = 'non-json-value\n'.encode() with tempfile.SpooledTemporaryFile(max_size=10*1024) as reader: reader.write(value) reader.seek(0) read_lines = list(S3Payload._read_downloaded_object(reader)) assert_equal(read_lines, [(1, value)])
def test_read_downloaded_object(self): """S3Payload - Read Downloaded Object""" record = {'key': 'value'} with tempfile.SpooledTemporaryFile(max_size=10*1024) as reader: reader.write(json.dumps(record, indent=2).encode()) reader.seek(0) read_lines = list(S3Payload._read_downloaded_object(reader)) assert_equal(read_lines, [(1, record)])