From 1703ed6ebbe2552990fb847d5fd5dde1d0382a37 Mon Sep 17 00:00:00 2001 From: Denis 'GNUtoo' Carikli Date: Mon, 6 Dec 2021 17:45:47 +0100 Subject: zip archives: keep individual files compression type While hardcoding the compression to zipfile.ZIP_DEFLATED works for most use cases of mat, being able to produce cleaned up uncompressed zip files is useful for content that cannot be compressed more. In addition it also enables to use mat2 for reproducible builds of Android bootanimation files file that don't support compression. Signed-off-by: Denis 'GNUtoo' Carikli --- tests/test_libmat2.py | 43 +++++++++++++++++++++++++++++++++++++------ 1 file changed, 37 insertions(+), 6 deletions(-) (limited to 'tests') diff --git a/tests/test_libmat2.py b/tests/test_libmat2.py index af92db3..f8e62de 100644 --- a/tests/test_libmat2.py +++ b/tests/test_libmat2.py @@ -175,14 +175,30 @@ class TestGetMeta(unittest.TestCase): def test_zip(self): with zipfile.ZipFile('./tests/data/dirty.zip', 'w') as zout: - zout.write('./tests/data/dirty.flac') - zout.write('./tests/data/dirty.docx') - zout.write('./tests/data/dirty.jpg') + zout.write('./tests/data/dirty.flac', + compress_type = zipfile.ZIP_STORED) + zout.write('./tests/data/dirty.docx', + compress_type = zipfile.ZIP_DEFLATED) + zout.write('./tests/data/dirty.jpg', + compress_type = zipfile.ZIP_BZIP2) + zout.write('./tests/data/dirty.txt', + compress_type = zipfile.ZIP_LZMA) p, mimetype = parser_factory.get_parser('./tests/data/dirty.zip') self.assertEqual(mimetype, 'application/zip') meta = p.get_meta() self.assertEqual(meta['tests/data/dirty.flac']['comments'], 'Thank you for using MAT !') self.assertEqual(meta['tests/data/dirty.docx']['word/media/image1.png']['Comment'], 'This is a comment, be careful!') + + with zipfile.ZipFile('./tests/data/dirty.zip') as zipin: + members = { + 'tests/data/dirty.flac' : zipfile.ZIP_STORED, + 'tests/data/dirty.docx': zipfile.ZIP_DEFLATED, + 'tests/data/dirty.jpg' : zipfile.ZIP_BZIP2, + 'tests/data/dirty.txt' : zipfile.ZIP_LZMA, + } + for k, v in members.items(): + self.assertEqual(zipin.getinfo(k).compress_type, v) + os.remove('./tests/data/dirty.zip') def test_wmv(self): @@ -595,9 +611,14 @@ class TestCleaning(unittest.TestCase): class TestCleaningArchives(unittest.TestCase): def test_zip(self): with zipfile.ZipFile('./tests/data/dirty.zip', 'w') as zout: - zout.write('./tests/data/dirty.flac') - zout.write('./tests/data/dirty.docx') - zout.write('./tests/data/dirty.jpg') + zout.write('./tests/data/dirty.flac', + compress_type = zipfile.ZIP_STORED) + zout.write('./tests/data/dirty.docx', + compress_type = zipfile.ZIP_DEFLATED) + zout.write('./tests/data/dirty.jpg', + compress_type = zipfile.ZIP_BZIP2) + zout.write('./tests/data/dirty.txt', + compress_type = zipfile.ZIP_LZMA) p = archive.ZipParser('./tests/data/dirty.zip') meta = p.get_meta() self.assertEqual(meta['tests/data/dirty.docx']['word/media/image1.png']['Comment'], 'This is a comment, be careful!') @@ -609,6 +630,16 @@ class TestCleaningArchives(unittest.TestCase): self.assertEqual(p.get_meta(), {}) self.assertTrue(p.remove_all()) + with zipfile.ZipFile('./tests/data/dirty.zip') as zipin: + members = { + 'tests/data/dirty.flac' : zipfile.ZIP_STORED, + 'tests/data/dirty.docx': zipfile.ZIP_DEFLATED, + 'tests/data/dirty.jpg' : zipfile.ZIP_BZIP2, + 'tests/data/dirty.txt' : zipfile.ZIP_LZMA, + } + for k, v in members.items(): + self.assertEqual(zipin.getinfo(k).compress_type, v) + os.remove('./tests/data/dirty.zip') os.remove('./tests/data/dirty.cleaned.zip') os.remove('./tests/data/dirty.cleaned.cleaned.zip') -- cgit v1.3