def test_read_less_than_page_size(self):
with open(self.file, 'w') as file:
file.write('F' * 512)
file.write('B' * 1024)
raw = RawDirect(self.file)
# Ask to only read 10 bytes
self.assertRaises(OSError, raw.read, 10)
# Read 512 bytes
self.assertEquals(raw.read(512), 'F' * 512)
# Read 1024
self.assertEquals(raw.read(1024), 'B' * 1024)
raw.close()
def test_read_past_eof(self):
# Create a file that is less than the block size
fd, file = tempfile.mkstemp(dir='/tmp')
os.write(fd, 'A' * 511)
os.close(fd)
raw = RawDirect(file)
# If requested read is past the EOF
# return the bytes we actually read
self.assertEquals(raw.read(512), ('A' * 511))
# Empty string means we are at EOF
self.assertEquals(raw.read(512), '')
raw.close()
def test_write_less_than_page_size(self):
raw = RawDirect(self.file)
# Write only 10 bytes
self.assertRaises(OSError, raw.write, ('G' * 10))
# Write 512
self.assertEquals(raw.write('A' * 512), 512)
# Write 1024
self.assertEquals(raw.write('B' * 1024), 1024)
# Ensure the data is there
raw.seek(0)
self.assertEquals(raw.read(512), 'A' * 512)
self.assertEquals(raw.read(1024), 'B' * 1024)
raw.close()
def test_read(self):
with open(self.file, 'w') as file:
file.write('B' * 512)
raw = RawDirect(self.file)
self.assertEquals(raw.read(512), 'B' * 512)
raw.close()
def test_read_greater_than_page_size(self):
with open(self.file, 'w') as file:
file.write('J' * 8192)
raw = RawDirect(self.file)
# Ask to read 8 more bytes then the block size
self.assertRaises(OSError, raw.read, 520)
# Read 8192 ( double the typical page size )
self.assertEquals(raw.read(8192), 'J' * 8192)
raw.close()
def test_tell(self):
raw = RawDirect(self.file)
raw.write('A' * 512)
self.assertEquals(raw.tell(), 512)
raw.write('B' * 512)
self.assertEquals(raw.tell(), 1024)
raw.seek(0)
self.assertEquals(raw.read(512), ('A' * 512))
self.assertEquals(raw.read(512), ('B' * 512))
self.assertEquals(raw.tell(), 1024)
