test_myzip.py

import unittest
import os
import filecmp
import shutil
import subprocess

# These tests rely on a working "unzip" command line utility
# I use the Info-ZIP one which is widely available

TEST_FILES = [
        'tests/resources/testfile1.bin',
        'tests/resources/testfile2.bin',
        'tests/resources/testfile3.bin',
        'tests/resources/testfile4.bin',
    ]

OUTPUT_DIR = 'tests/output'
OUTPUT_ZIP_NAME = os.path.join(OUTPUT_DIR, 'result.zip')

class TestMyzip(unittest.TestCase):
    def test_zip_utils(self):
        """
        Try to zip up each of the files
        and unzip them
        """

        for zip_exec in ['./myzip0', './myzip']:

            # make an output directory
            try:
                os.makedirs(OUTPUT_DIR)
            except FileExistsError:
                pass
            except:
                self.fail('could not make output directory')

            for filename in TEST_FILES:
                # make sure the test file even exists
                self.assertTrue(os.path.exists(filename))

                # zip up the file
                subprocess.run([zip_exec, OUTPUT_ZIP_NAME, filename])

                # check that zipped verison exists
                self.assertTrue(os.path.exists(OUTPUT_ZIP_NAME))

                # now try to unzip it with "known correct" tool
                subprocess.run(['unzip', '-q', '-o', OUTPUT_ZIP_NAME, '-d', OUTPUT_DIR],
                    stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)


                recovered_file_path = os.path.join(OUTPUT_DIR, filename)

                # make sure recovered file exists, and
                self.assertTrue(os.path.exists(recovered_file_path))

                # check that it is correct
                self.assertTrue(filecmp.cmp(filename, recovered_file_path))

            shutil.rmtree(OUTPUT_DIR)

if __name__ == '__main__':
    unittest.main()