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)