def test_decopmression_wrong_password(self):
compression.compress_recursively(
_CASE_PATH, _CASE_SRC, _CASE_DEST, password="******", encryption_method=compression.ENCRYPTION_AES_256
)
# Wrong password
self.assertRaises(
compression.CompressionWrongPassword,
compression.decompress_recursively,
_CASE_PATH,
_CASE_DEST,
_CASE_OUT,
_CASE_TMP,
password="******",
)
def test_copmression(self):
compression.compress_recursively(
_CASE_PATH, _CASE_SRC, _CASE_DEST, password=None, encryption_method=compression.ENCRYPTION_AES_256
)
src = os.path.join(_CASE_SRC, _CASE_PATH)
dest = os.path.join(_CASE_DEST, _CASE_PATH)
for path, dirs, files in os.walk(src):
rel_path = os.path.relpath(path, src)
dest_path = os.path.join(dest, rel_path)
for f in files:
self.assertTrue(
"Compressed file %s is not found for %s"
% (os.path.join(dest_path, compression.get_compressed_filename(f)), os.path.join(path, f)),
os.path.exists(os.path.join(dest_path, compression.get_compressed_filename(f))),
)
def test_decopmression_correct_password(self):
compression.compress_recursively(
_CASE_PATH, _CASE_SRC, _CASE_DEST, password="******", encryption_method=compression.ENCRYPTION_AES_256
)
compression.decompress_recursively(_CASE_PATH, _CASE_DEST, _CASE_OUT, _CASE_TMP, password="******")
src = os.path.join(_CASE_SRC, _CASE_PATH)
out = os.path.join(_CASE_OUT, _CASE_PATH)
for path, dirs, files in os.walk(src):
rel_path = os.path.relpath(path, src)
out_path = os.path.join(out, rel_path)
for f in files:
self.assertTrue(
"Decompressed file %s is not found for %s" % (os.path.join(out_path, f), os.path.join(path, f)),
os.path.exists(os.path.join(out_path, f)),
)
self.assertTrue(
"Decompressed file %s is not identical to %s" % (os.path.join(out_path, f), os.path.join(path, f)),
filecmp.cmp(os.path.join(path, f), os.path.join(out_path, f)),
)
