def test5(self): """test FileWrapper with unseekable file object""" sio = StringIO('Simple test file') sio.seek = None f = FileWrapper(fobj=sio, use_mmap=False) self.assertFalse(f.is_seekable()) self.assertFalse(isinstance(f._fobj, mmap.mmap)) self.assertEqual(f.tell(), 0) self.assertEqual(f.read(3), 'Sim') self.assertEqual(f.tell(), 3) # seeking backward is not supposed to work self.assertRaises(IOError, f.seek, 0, os.SEEK_SET) self.assertRaises(IOError, f.seek, -2, os.SEEK_SET) self.assertRaises(IOError, f.seek, -2, os.SEEK_CUR) self.assertRaises(IOError, f.seek, 5, os.SEEK_END) # seeking forward (absolute) does work (skip 'pl') f.seek(5, os.SEEK_SET) self.assertEqual(f.read(3), 'e t') # seeking forward (relative) does work (skip 'est') f.seek(3, os.SEEK_CUR) # read the rest of the file self.assertEqual(f.read(), ' file') f.close() # sio should not be closed self.assertFalse(sio.closed)
def test7_1(self): """test FileWrapper's peek method (peek data and complete read)""" fname = self.fixture_file('filewrapper1.txt') with open(fname, 'r') as fobj: f = FileWrapper(fobj=fobj, use_mmap=False) self.assertFalse(isinstance(f._fobj, mmap.mmap)) self.assertEqual(f.peek(5), 'This ') self.assertEqual(f.tell(), 0) # read 3 bytes (arbitrarily chosen) self.assertEqual(f.read(3), 'Thi') self.assertEqual(f.tell(), 3) # read rest of file (2 bytes of peek data are still available) self.assertEqual(f.read(), 's is a simple\ntext file.\n')
def test1(self): """test FileWrapper with filename (no mmap)""" fname = self.fixture_file('filewrapper1.txt') f = FileWrapper(filename=fname, use_mmap=False) self.assertFalse(isinstance(f._fobj, mmap.mmap)) self.assertTrue(f.is_seekable()) self.assertEqual(f.tell(), 0) self.assertEqual(f.read(1), 'T') self.assertEqual(f.tell(), 1) # seek to position 0 again f.seek(0) self.assertEqual(f.read(5), 'This ') self.assertEqual(f.tell(), 5) # read rest of the file self.assertEqual(f.read(), 'is a simple\ntext file.\n') f.close()
def test7(self): """test FileWrapper's peek method""" fname = self.fixture_file('filewrapper1.txt') with open(fname, 'r') as fobj: f = FileWrapper(fobj=fobj, use_mmap=False) self.assertFalse(isinstance(f._fobj, mmap.mmap)) self.assertEqual(f.peek(3), 'Thi') self.assertEqual(f.tell(), 0) # seeking is not allowed unless all peek'ed bytes are read self.assertRaises(IOError, f.seek, 8, os.SEEK_SET) self.assertEqual(f.read(2), 'Th') self.assertEqual(f.tell(), 2) # seek is still not possible (1 peek byte left) self.assertRaises(IOError, f.seek, 8, os.SEEK_SET) self.assertEqual(f.read(2), 'is') self.assertEqual(f.tell(), 4) # seek is possible f.seek(8, os.SEEK_SET) self.assertEqual(f.read(), 'a simple\ntext file.\n')
def test2(self): """test FileWrapper with file object (no mmap)""" fname = self.fixture_file('filewrapper1.txt') with open(fname, 'r') as fobj: f = FileWrapper(fobj=fobj, use_mmap=False) self.assertFalse(isinstance(f._fobj, mmap.mmap)) self.assertTrue(f.is_seekable()) self.assertEqual(f.tell(), 0) self.assertEqual(f.read(1), 'T') self.assertEqual(f.tell(), 1) # seek to position 0 again f.seek(-1, os.SEEK_CUR) self.assertEqual(f.tell(), 0) self.assertEqual(f.read(5), 'This ') self.assertEqual(f.tell(), 5) # read rest of the file self.assertEqual(f.read(), 'is a simple\ntext file.\n') f.close() # the fobj should not be closed in this case self.assertFalse(fobj.closed)