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)])
