def test_open(self):
# 1. Regular file
with zipio.open(os.path.join(TESTDATA, 'test.txt'), 'r') as fp:
data = fp.read()
self.assertEqual(data, 'This is test.txt\n')
if sys.version_info[0] == 3:
with zipio.open(os.path.join(TESTDATA, 'test.txt'), 'rb') as fp:
data = fp.read()
self.assertEqual(data, b'This is test.txt\n')
# 2. File inside zipfile
with zipio.open(os.path.join(TESTDATA, 'zipped.egg', 'test.txt'),
'r') as fp:
data = fp.read()
self.assertEqual(data, 'Zipped up test.txt\n')
if sys.version_info[0] == 3:
with zipio.open(os.path.join(TESTDATA, 'zipped.egg', 'test.txt'),
'rb') as fp:
data = fp.read()
self.assertEqual(data, b'Zipped up test.txt\n')
# 3. EXC: Directory inside zipfile
self.assertRaises(IOError, zipio.open,
os.path.join(TESTDATA, 'zipped.egg', 'subdir'))
self.assertRaises(IOError, zipio.open,
os.path.join(TESTDATA, 'zipped.egg', 'subdir2'))
self.assertRaises(
IOError, zipio.open,
os.path.join(TESTDATA, 'zipped.egg', 'subdir2/subdir'))
self.assertRaises(IOError, zipio.open,
os.path.join(TESTDATA, 'zipped.egg', 'subdir3'))
# TODO: Add subdir4/file.txt, without directory entry
self.assertRaises(IOError, zipio.open,
os.path.join(TESTDATA, 'zipped.egg', 'subdir4'))
# 4. EXC: No such file in zipfile
self.assertRaises(IOError, zipio.open,
os.path.join(TESTDATA, 'zipped.egg', 'no-such-file'))
self.assertRaises(
IOError, zipio.open,
os.path.join(TESTDATA, 'zipped.egg', 'subdir/no-such-file'))
# 5. EXC: No such regular file
self.assertRaises(IOError, zipio.open,
os.path.join(TESTDATA, 'no-such-file.txt'))
# 6. EXC: Open r/w
self.assertRaises(IOError, zipio.open,
os.path.join(TESTDATA, 'test.txt'), 'w')
self.assertRaises(IOError, zipio.open,
os.path.join(TESTDATA, 'test.txt'), 'a')
self.assertRaises(IOError, zipio.open,
os.path.join(TESTDATA, 'test.txt'), 'r+')
self.assertRaises(IOError, zipio.open,
os.path.join(TESTDATA, 'test.txt'), 'w+')
self.assertRaises(IOError, zipio.open,
os.path.join(TESTDATA, 'test.txt'), 'a+')
def test_open(self):
# 1. Regular file
fp = zipio.open(os.path.join(TESTDATA, 'test.txt'), 'r')
data = fp.read()
fp.close()
self.assertEqual(data, 'This is test.txt\n')
if sys.version_info[0] == 3:
fp = zipio.open(os.path.join(TESTDATA, 'test.txt'), 'rb')
data = fp.read()
fp.close()
self.assertEqual(data, b'This is test.txt\n')
# 2. File inside zipfile
fp = zipio.open(os.path.join(TESTDATA, 'zipped.egg', 'test.txt'), 'r')
data = fp.read()
fp.close()
self.assertEqual(data, 'Zipped up test.txt\n')
if sys.version_info[0] == 3:
fp = zipio.open(os.path.join(TESTDATA, 'zipped.egg', 'test.txt'), 'rb')
data = fp.read()
fp.close()
self.assertEqual(data, b'Zipped up test.txt\n')
# 3. EXC: Directory inside zipfile
self.assertRaises(IOError, zipio.open, os.path.join(TESTDATA, 'zipped.egg', 'subdir'))
self.assertRaises(IOError, zipio.open, os.path.join(TESTDATA, 'zipped.egg', 'subdir2'))
self.assertRaises(IOError, zipio.open, os.path.join(TESTDATA, 'zipped.egg', 'subdir2/subdir'))
self.assertRaises(IOError, zipio.open, os.path.join(TESTDATA, 'zipped.egg', 'subdir3'))
# TODO: Add subdir4/file.txt, without directory entry
self.assertRaises(IOError, zipio.open, os.path.join(TESTDATA, 'zipped.egg', 'subdir4'))
# 4. EXC: No such file in zipfile
self.assertRaises(IOError, zipio.open, os.path.join(TESTDATA, 'zipped.egg', 'no-such-file'))
self.assertRaises(IOError, zipio.open, os.path.join(TESTDATA, 'zipped.egg', 'subdir/no-such-file'))
# 5. EXC: No such regular file
self.assertRaises(IOError, zipio.open, os.path.join(TESTDATA, 'no-such-file.txt'))
# 6. EXC: Open r/w
self.assertRaises(IOError, zipio.open, os.path.join(TESTDATA, 'test.txt'), 'w')
self.assertRaises(IOError, zipio.open, os.path.join(TESTDATA, 'test.txt'), 'a')
self.assertRaises(IOError, zipio.open, os.path.join(TESTDATA, 'test.txt'), 'r+')
self.assertRaises(IOError, zipio.open, os.path.join(TESTDATA, 'test.txt'), 'w+')
self.assertRaises(IOError, zipio.open, os.path.join(TESTDATA, 'test.txt'), 'a+')
def test_contextlib(self):
# 1. Regular file
with zipio.open(os.path.join(TESTDATA, 'test.txt'), 'r') as fp:
data = fp.read()
try:
fp.read()
self.fail("file not closed")
except (ValueError, IOError):
pass
self.assertEqual(data, 'This is test.txt\n')
if sys.version_info[0] == 3:
with zipio.open(os.path.join(TESTDATA, 'test.txt'), 'rb') as fp:
data = fp.read()
try:
fp.read()
self.fail("file not closed")
except (ValueError, IOError):
pass
self.assertEqual(data, b'This is test.txt\n')
# 2. File inside zipfile
with zipio.open(os.path.join(TESTDATA, 'zipped.egg', 'test.txt'),
'r') as fp:
data = fp.read()
try:
fp.read()
self.fail("file not closed")
except (ValueError, IOError):
pass
self.assertEqual(data, 'Zipped up test.txt\n')
if sys.version_info[0] == 3:
with zipio.open(os.path.join(TESTDATA, 'zipped.egg', 'test.txt'),
'rb') as fp:
data = fp.read()
try:
fp.read()
self.fail("file not closed")
except (IOError, ValueError):
pass
self.assertEqual(data, b'Zipped up test.txt\n')
def test_contextlib(self):
# 1. Regular file
with zipio.open(os.path.join(TESTDATA, 'test.txt'), 'r') as fp:
data = fp.read()
try:
fp.read()
self.fail("file not closed")
except (ValueError, IOError):
pass
self.assertEqual(data, 'This is test.txt\n')
if sys.version_info[0] == 3:
with zipio.open(os.path.join(TESTDATA, 'test.txt'), 'rb') as fp:
data = fp.read()
try:
fp.read()
self.fail("file not closed")
except (ValueError, IOError):
pass
self.assertEqual(data, b'This is test.txt\n')
# 2. File inside zipfile
with zipio.open(os.path.join(TESTDATA, 'zipped.egg', 'test.txt'), 'r') as fp:
data = fp.read()
try:
fp.read()
self.fail("file not closed")
except (ValueError, IOError):
pass
self.assertEqual(data, 'Zipped up test.txt\n')
if sys.version_info[0] == 3:
with zipio.open(os.path.join(TESTDATA, 'zipped.egg', 'test.txt'), 'rb') as fp:
data = fp.read()
try:
fp.read()
self.fail("file not closed")
except (IOError, ValueError):
pass
self.assertEqual(data, b'Zipped up test.txt\n')