def test__init__(self):
"""Test Ziphyr object init."""
z = module.Ziphyr()
self.assertEqual(z.password, None)
self.assertEqual(z.stream, None)
self.assertTrue(
issubclass(z.ZipInfo, RetroZipInfo),
msg="Ziphyr.ZipInfo is not a subclass of RetroZipInfo",
)
self.assertFalse(
issubclass(z.ZipInfo, module.PKCryptoZipInfo),
msg="Ziphyr.ZipInfo is a subclass of PKCryptoZipInfo",
)
_password = b'password'
z = module.Ziphyr(_password)
self.assertEqual(z.password, _password)
self.assertEqual(z.stream, None)
self.assertTrue(
issubclass(z.ZipInfo, module.PKCryptoZipInfo),
msg="Ziphyr.ZipInfo is not a subclass of RetroZipInfo",
)
def test_zip_unzip_passwordless(self):
"""
Test zipping with Ziphyr (passwordless) then unzipping with zipfile.
"""
with tempfile.TemporaryDirectory() as tmpdir:
test_fp = tmpdir + '/test.file'
test_zp = tmpdir + '/test.zip'
with open(test_fp, 'w') as f:
f.write("I thought what I'd do was, I'd pretend I was one "
"of those deaf-mutes. That way I wouldn't have to "
"have any goddam stupid useless conversations with "
"anybody.")
z = module.Ziphyr()
z.from_filepath(test_fp)
source = file_iterable(test_fp)
with open(test_zp, 'ab') as f:
for chunk in z.generator(source):
f.write(chunk)
with ZipFile(test_zp, 'r') as f:
test_rp = f.extract(test_fp[1:], tmpdir)
self.assertTrue(cmp(test_fp, test_rp))
def test_generator(self):
"""Test Ziphyr object's generator consumption."""
z = module.Ziphyr(b'password')
z.from_metadata("h2g2", 42)
zg = z.generator(list())
self.assertEqual(next(zg)[:4], b'PK\x03\x04')
def test_35_behavior(self, m_version_info):
""" """
m_version_info.minor = 7
z = module.Ziphyr(b'')
self.assertTrue(
issubclass(z.ZipFile, ZipFile),
msg="Ziphyr.ZipFile is not a subclass of ZipFile",
)
m_version_info.minor = 5
z = module.Ziphyr(b'')
self.assertTrue(
issubclass(z.ZipFile, RetroZipFile),
msg="Ziphyr.ZipFile is not a subclass of RetroZipFile",
)
def test_from_metadata(self):
"""Test Ziphyr.zinfo set-up from metadata."""
_filename = "Animal Farm"
_filesize = 1984
z = module.Ziphyr(b'password')
z.from_metadata(_filename, _filesize)
self.assertEqual(z.zinfo.orig_filename, _filename)
self.assertEqual(z.zinfo.filename, _filename)
self.assertEqual(z.zinfo.file_size, _filesize)
self.assertEqual(z.zinfo.external_attr, 25165824)
def test_primeless(self):
"""Test error when Ziphyr isn't primed."""
z = module.Ziphyr()
zg = z.generator(list())
with self.assertRaises(RuntimeError):
next(zg)