def testReadMultipleFiles(self):
self.setUpMultipleFile()
r = gcs_file_seg_reader._GCSFileSegReader(self.seg_prefix, 2)
result = ""
while True:
tmp = r.read(1)
if not tmp:
break
result += tmp
self.assertEqual("1234567890123456", result)
self.assertEqual(len(result), r.tell())
def _assertOutputEqual(self, seg_prefix, last_seg_index):
# Read back outputs.
reader = gcs_file_seg_reader._GCSFileSegReader(seg_prefix, last_seg_index)
result = ""
while True:
tmp = reader.read(n=100)
if not tmp:
break
result += tmp
# Verify output has no duplicates.
expected = ""
for i in range(30):
expected += "%s\n" % i
self.assertEqual(expected, result)
def ReadOneFileTest(self, read_size):
f = cloudstorage.open(self.seg_prefix + "0", "w",
options={self.writer_cls._VALID_LENGTH: "5"})
f.write("1234567")
f.close()
r = gcs_file_seg_reader._GCSFileSegReader(self.seg_prefix, 0)
result = ""
while True:
tmp = r.read(read_size)
if not tmp:
break
result += tmp
self.assertEqual("12345", result)
self.assertEqual(5, r.tell())
def _assertOutputEqual(self, seg_prefix, last_seg_index):
# Read back outputs.
reader = gcs_file_seg_reader._GCSFileSegReader(seg_prefix,
last_seg_index)
result = ""
while True:
tmp = reader.read(n=100)
if not tmp:
break
result += tmp
# Verify output has no duplicates.
expected = ""
for i in range(30):
expected += "%s\n" % i
self.assertEqual(expected, result)
