def test_read_buffer_at_offset(self):
"""Tests the read_buffer_at_offset function."""
if not unittest.source:
return
modi_handle = pymodi.handle()
modi_handle.open(unittest.source)
file_size = modi_handle.get_size()
# Test normal read.
data = modi_handle.read_buffer_at_offset(4096, 0)
self.assertIsNotNone(data)
self.assertEqual(len(data), min(file_size, 4096))
# Test read beyond file size.
if file_size > 16:
data = modi_handle.read_buffer_at_offset(4096, file_size - 16)
self.assertIsNotNone(data)
self.assertEqual(len(data), 16)
with self.assertRaises(ValueError):
modi_handle.read_buffer_at_offset(-1, 0)
with self.assertRaises(ValueError):
modi_handle.read_buffer_at_offset(4096, -1)
modi_handle.close()
# Test the read without open.
with self.assertRaises(IOError):
modi_handle.read_buffer_at_offset(4096, 0)
def test_seek_offset(self):
"""Tests the seek_offset function."""
test_source = unittest.source
if not test_source:
raise unittest.SkipTest("missing source")
modi_handle = pymodi.handle()
modi_handle.open(test_source)
modi_handle.open_band_data_files()
media_size = modi_handle.get_media_size()
modi_handle.seek_offset(16, os.SEEK_SET)
offset = modi_handle.get_offset()
self.assertEqual(offset, 16)
modi_handle.seek_offset(16, os.SEEK_CUR)
offset = modi_handle.get_offset()
self.assertEqual(offset, 32)
modi_handle.seek_offset(-16, os.SEEK_CUR)
offset = modi_handle.get_offset()
self.assertEqual(offset, 16)
if media_size > 16:
modi_handle.seek_offset(-16, os.SEEK_END)
offset = modi_handle.get_offset()
self.assertEqual(offset, media_size - 16)
modi_handle.seek_offset(16, os.SEEK_END)
offset = modi_handle.get_offset()
self.assertEqual(offset, media_size + 16)
# TODO: change IOError into ValueError
with self.assertRaises(IOError):
modi_handle.seek_offset(-1, os.SEEK_SET)
# TODO: change IOError into ValueError
with self.assertRaises(IOError):
modi_handle.seek_offset(-32 - media_size, os.SEEK_CUR)
# TODO: change IOError into ValueError
with self.assertRaises(IOError):
modi_handle.seek_offset(-32 - media_size, os.SEEK_END)
# TODO: change IOError into ValueError
with self.assertRaises(IOError):
modi_handle.seek_offset(0, -1)
modi_handle.close()
# Test the seek without open.
with self.assertRaises(IOError):
modi_handle.seek_offset(16, os.SEEK_SET)
def test_open_file_object(self):
"""Tests the open_file_object function."""
test_source = unittest.source
if not test_source:
raise unittest.SkipTest("missing source")
if not os.path.isfile(test_source):
raise unittest.SkipTest("source not a regular file")
modi_handle = pymodi.handle()
with open(test_source, "rb") as file_object:
modi_handle.open_file_object(file_object)
with self.assertRaises(IOError):
modi_handle.open_file_object(file_object)
modi_handle.close()
with self.assertRaises(TypeError):
modi_handle.open_file_object(None)
with self.assertRaises(ValueError):
modi_handle.open_file_object(file_object, mode="w")
def test_open_close(self):
"""Tests the open and close functions."""
test_source = unittest.source
if not test_source:
return
modi_handle = pymodi.handle()
# Test open and close.
modi_handle.open(test_source)
modi_handle.close()
# Test open and close a second time to validate clean up on close.
modi_handle.open(test_source)
modi_handle.close()
if os.path.isfile(test_source):
with open(test_source, "rb") as file_object:
# Test open_file_object and close.
modi_handle.open_file_object(file_object)
modi_handle.close()
# Test open_file_object and close a second time to validate clean up on close.
modi_handle.open_file_object(file_object)
modi_handle.close()
# Test open_file_object and close and dereferencing file_object.
modi_handle.open_file_object(file_object)
del file_object
modi_handle.close()
def test_open_close(self):
"""Tests the open and close functions."""
if not unittest.source:
return
modi_handle = pymodi.handle()
# Test open and close.
modi_handle.open(unittest.source)
modi_handle.close()
# Test open and close a second time to validate clean up on close.
modi_handle.open(unittest.source)
modi_handle.close()
file_object = open(unittest.source, "rb")
# Test open_file_object and close.
modi_handle.open_file_object(file_object)
modi_handle.close()
# Test open_file_object and close a second time to validate clean up on close.
modi_handle.open_file_object(file_object)
modi_handle.close()
# Test open_file_object and close and dereferencing file_object.
modi_handle.open_file_object(file_object)
del file_object
modi_handle.close()
def test_close(self):
"""Tests the close function."""
if not unittest.source:
return
modi_handle = pymodi.handle()
with self.assertRaises(IOError):
modi_handle.close()
def test_seek_offset(self):
"""Tests the seek_offset function."""
if not unittest.source:
return
modi_handle = pymodi.handle()
modi_handle.open(unittest.source)
file_size = modi_handle.get_size()
modi_handle.seek_offset(16, os.SEEK_SET)
offset = modi_handle.get_offset()
self.assertEqual(offset, 16)
modi_handle.seek_offset(16, os.SEEK_CUR)
offset = modi_handle.get_offset()
self.assertEqual(offset, 32)
modi_handle.seek_offset(-16, os.SEEK_CUR)
offset = modi_handle.get_offset()
self.assertEqual(offset, 16)
modi_handle.seek_offset(-16, os.SEEK_END)
offset = modi_handle.get_offset()
self.assertEqual(offset, file_size - 16)
modi_handle.seek_offset(16, os.SEEK_END)
offset = modi_handle.get_offset()
self.assertEqual(offset, file_size + 16)
# TODO: change IOError into ValueError
with self.assertRaises(IOError):
modi_handle.seek_offset(-1, os.SEEK_SET)
# TODO: change IOError into ValueError
with self.assertRaises(IOError):
modi_handle.seek_offset(-32 - file_size, os.SEEK_CUR)
# TODO: change IOError into ValueError
with self.assertRaises(IOError):
modi_handle.seek_offset(-32 - file_size, os.SEEK_END)
# TODO: change IOError into ValueError
with self.assertRaises(IOError):
modi_handle.seek_offset(0, -1)
modi_handle.close()
# Test the seek without open.
with self.assertRaises(IOError):
modi_handle.seek_offset(16, os.SEEK_SET)
def test_close(self):
"""Tests the close function."""
test_source = unittest.source
if not test_source:
raise unittest.SkipTest("missing source")
modi_handle = pymodi.handle()
with self.assertRaises(IOError):
modi_handle.close()
def test_get_offset(self):
"""Tests the get_offset function."""
test_source = unittest.source
if not test_source:
raise unittest.SkipTest("missing source")
modi_handle = pymodi.handle()
modi_handle.open(test_source)
offset = modi_handle.get_offset()
self.assertIsNotNone(offset)
modi_handle.close()
def test_get_media_size(self):
"""Tests the get_media_size function and media_size property."""
test_source = unittest.source
if not test_source:
raise unittest.SkipTest("missing source")
modi_handle = pymodi.handle()
modi_handle.open(test_source)
media_size = modi_handle.get_media_size()
self.assertIsNotNone(media_size)
self.assertIsNotNone(modi_handle.media_size)
modi_handle.close()
def test_open(self):
"""Tests the open function."""
if not unittest.source:
return
modi_handle = pymodi.handle()
modi_handle.open(unittest.source)
with self.assertRaises(IOError):
modi_handle.open(unittest.source)
modi_handle.close()
with self.assertRaises(TypeError):
modi_handle.open(None)
with self.assertRaises(ValueError):
modi_handle.open(unittest.source, mode="w")
def test_read_buffer_file_object(self):
"""Tests the read_buffer function on a file-like object."""
if not unittest.source:
return
file_object = open(unittest.source, "rb")
modi_handle = pymodi.handle()
modi_handle.open_file_object(file_object)
file_size = modi_handle.get_size()
# Test normal read.
data = modi_handle.read_buffer(size=4096)
self.assertIsNotNone(data)
self.assertEqual(len(data), min(file_size, 4096))
modi_handle.close()
def test_open(self):
"""Tests the open function."""
test_source = unittest.source
if not test_source:
raise unittest.SkipTest("missing source")
modi_handle = pymodi.handle()
modi_handle.open(test_source)
with self.assertRaises(IOError):
modi_handle.open(test_source)
modi_handle.close()
with self.assertRaises(TypeError):
modi_handle.open(None)
with self.assertRaises(ValueError):
modi_handle.open(test_source, mode="w")
def test_open_file_object(self):
"""Tests the open_file_object function."""
if not unittest.source:
return
file_object = open(unittest.source, "rb")
modi_handle = pymodi.handle()
modi_handle.open_file_object(file_object)
with self.assertRaises(IOError):
modi_handle.open_file_object(file_object)
modi_handle.close()
# TODO: change IOError into TypeError
with self.assertRaises(IOError):
modi_handle.open_file_object(None)
with self.assertRaises(ValueError):
modi_handle.open_file_object(file_object, mode="w")
def test_read_buffer_file_object(self):
"""Tests the read_buffer function on a file-like object."""
test_source = unittest.source
if not test_source:
raise unittest.SkipTest("missing source")
if not os.path.isfile(test_source):
raise unittest.SkipTest("source not a regular file")
modi_handle = pymodi.handle()
with open(test_source, "rb") as file_object:
modi_handle.open_file_object(file_object)
modi_handle.open_band_data_files()
media_size = modi_handle.get_media_size()
# Test normal read.
data = modi_handle.read_buffer(size=4096)
self.assertIsNotNone(data)
self.assertEqual(len(data), min(media_size, 4096))
modi_handle.close()
def test_read_buffer_at_offset(self):
"""Tests the read_buffer_at_offset function."""
test_source = unittest.source
if not test_source:
raise unittest.SkipTest("missing source")
modi_handle = pymodi.handle()
modi_handle.open(test_source)
modi_handle.open_band_data_files()
media_size = modi_handle.get_media_size()
# Test normal read.
data = modi_handle.read_buffer_at_offset(4096, 0)
self.assertIsNotNone(data)
self.assertEqual(len(data), min(media_size, 4096))
if media_size > 8:
# Read buffer on media_size boundary.
data = modi_handle.read_buffer_at_offset(4096, media_size - 8)
self.assertIsNotNone(data)
self.assertEqual(len(data), 8)
# Read buffer beyond media_size boundary.
data = modi_handle.read_buffer_at_offset(4096, media_size + 8)
self.assertIsNotNone(data)
self.assertEqual(len(data), 0)
# Stress test read buffer.
for _ in range(1024):
random_number = random.random()
media_offset = int(random_number * media_size)
read_size = int(random_number * 4096)
data = modi_handle.read_buffer_at_offset(read_size, media_offset)
self.assertIsNotNone(data)
remaining_media_size = media_size - media_offset
data_size = len(data)
if read_size > remaining_media_size:
read_size = remaining_media_size
self.assertEqual(data_size, read_size)
remaining_media_size -= data_size
if not remaining_media_size:
modi_handle.seek_offset(0, os.SEEK_SET)
with self.assertRaises(ValueError):
modi_handle.read_buffer_at_offset(-1, 0)
with self.assertRaises(ValueError):
modi_handle.read_buffer_at_offset(4096, -1)
modi_handle.close()
# Test the read without open.
with self.assertRaises(IOError):
modi_handle.read_buffer_at_offset(4096, 0)
def test_read_buffer(self):
"""Tests the read_buffer function."""
test_source = unittest.source
if not test_source:
raise unittest.SkipTest("missing source")
modi_handle = pymodi.handle()
modi_handle.open(test_source)
modi_handle.open_band_data_files()
media_size = modi_handle.get_media_size()
if media_size < 4096:
# Test read without maximum size.
modi_handle.seek_offset(0, os.SEEK_SET)
data = modi_handle.read_buffer()
self.assertIsNotNone(data)
self.assertEqual(len(data), media_size)
# Test read with maximum size.
modi_handle.seek_offset(0, os.SEEK_SET)
data = modi_handle.read_buffer(size=4096)
self.assertIsNotNone(data)
self.assertEqual(len(data), min(media_size, 4096))
if media_size > 8:
modi_handle.seek_offset(-8, os.SEEK_END)
# Read buffer on media_size boundary.
data = modi_handle.read_buffer(size=4096)
self.assertIsNotNone(data)
self.assertEqual(len(data), 8)
# Read buffer beyond media_size boundary.
data = modi_handle.read_buffer(size=4096)
self.assertIsNotNone(data)
self.assertEqual(len(data), 0)
# Stress test read buffer.
modi_handle.seek_offset(0, os.SEEK_SET)
remaining_media_size = media_size
for _ in range(1024):
read_size = int(random.random() * 4096)
data = modi_handle.read_buffer(size=read_size)
self.assertIsNotNone(data)
data_size = len(data)
if read_size > remaining_media_size:
read_size = remaining_media_size
self.assertEqual(data_size, read_size)
remaining_media_size -= data_size
if not remaining_media_size:
modi_handle.seek_offset(0, os.SEEK_SET)
remaining_media_size = media_size
with self.assertRaises(ValueError):
modi_handle.read_buffer(size=-1)
modi_handle.close()
# Test the read without open.
with self.assertRaises(IOError):
modi_handle.read_buffer(size=4096)
def test_signal_abort(self):
"""Tests the signal_abort function."""
modi_handle = pymodi.handle()
modi_handle.signal_abort()
