def test_closed(self):
with self.get_raw_buffer() as data_file:
chunk = RawIOChunk(data_file, len(self.STRINGS[0]))
ok_(not chunk.closed)
ok_(chunk.closed)
with assert_raises(ValueError):
chunk.read()
def test_closed(self):
with self.get_raw_buffer() as data_file:
chunk = RawIOChunk(data_file, len(self.STRINGS[0]))
self.assertFalse(chunk.closed)
self.assertTrue(chunk.closed)
with self.assertRaises(ValueError):
chunk.read()
def test_full_file(self):
with self.get_raw_buffer() as data_file:
chunks = [RawIOChunk(data_file, len(self.STRINGS[0]))]
chunks.append(
RawIOChunk(data_file, len(self.STRINGS[1]), chunks[0].end))
chunks.append(
RawIOChunk(data_file, len(self.STRINGS[2]), chunks[1].end))
eq_(sum(chunk.size for chunk in chunks),
len(b''.join(self.STRINGS)))
def test_file_begin(self):
with self.get_raw_buffer() as data_file:
chunk = RawIOChunk(data_file, len(self.STRINGS[0]))
eq_(chunk.start, 0)
eq_(chunk.size, len(self.STRINGS[0]))
eq_(chunk.end, chunk.start + chunk.size)
eq_(chunk.tell(), chunk.start)
eq_(chunk.read(), self.STRINGS[0])
eq_(chunk.tell(), chunk.end)
def test_eof(self):
with self.get_raw_buffer() as data_file:
chunk = RawIOChunk(data_file,
len(self.STRINGS[2]) + 1,
len(self.STRINGS[0]) + len(self.STRINGS[1]))
with assert_raises(EOFError):
chunk.read()
chunk.seek(0)
eq_(chunk.read(chunk.size - 1), self.STRINGS[2])
with assert_raises(EOFError):
chunk.read(1)
def test_seek(self):
with self.get_raw_buffer() as data_file:
chunk = RawIOChunk(data_file, len(self.STRINGS[1]),
len(self.STRINGS[0]))
eq_(chunk.seek(1), 1)
eq_(chunk.tell(), 1)
eq_(chunk.seek(1, SEEK_CUR), 2)
eq_(chunk.tell(), 2)
eq_(chunk.seek(-1, SEEK_END), len(self.STRINGS[1]) - 1)
eq_(chunk.tell(), len(self.STRINGS[1]) - 1)
eq_(chunk.seek(0), 0)
eq_(chunk.read(), self.STRINGS[1])
eq_(chunk.seek(0), 0)
eq_(chunk.read(), self.STRINGS[1])
def test_file_middle(self):
with self.get_raw_buffer() as data_file:
chunk = RawIOChunk(data_file, len(self.STRINGS[1]),
len(self.STRINGS[0]))
eq_(chunk.read(), self.STRINGS[1])