Python zip_folder_content Examples

Example #1

File: test_utils_.py Project: Mritrocker96/addons-server

def test_zip_folder_content():
    extension_file = 'src/olympia/files/fixtures/files/extension.xpi'
    try:
        temp_folder = utils.extract_zip(extension_file)
        assert os.listdir(temp_folder) == [
            'install.rdf', 'chrome.manifest', 'chrome']
        temp_filename = amo.tests.get_temp_filename()
        utils.zip_folder_content(temp_folder, temp_filename)
        # Make sure the zipped files contain the same files.
        with zipfile.ZipFile(temp_filename, mode='r') as new:
            with zipfile.ZipFile(extension_file, mode='r') as orig:
                assert new.namelist() == orig.namelist()
    finally:
        if os.path.exists(temp_folder):
            amo.utils.rm_local_tmp_dir(temp_folder)
        if os.path.exists(temp_filename):
            os.unlink(temp_filename)

Example #2

File: test_utils_.py Project: darktrojan/addons-server

def test_zip_folder_content():
    extension_file = "src/olympia/files/fixtures/files/extension.xpi"
    temp_filename, temp_folder = None, None
    try:
        temp_folder = utils.extract_zip(extension_file)
        assert sorted(os.listdir(temp_folder)) == ["chrome", "chrome.manifest", "install.rdf"]
        temp_filename = amo.tests.get_temp_filename()
        utils.zip_folder_content(temp_folder, temp_filename)
        # Make sure the zipped files contain the same files.
        with zipfile.ZipFile(temp_filename, mode="r") as new:
            with zipfile.ZipFile(extension_file, mode="r") as orig:
                assert sorted(new.namelist()) == sorted(orig.namelist())
    finally:
        if temp_folder is not None and os.path.exists(temp_folder):
            amo.utils.rm_local_tmp_dir(temp_folder)
        if temp_filename is not None and os.path.exists(temp_filename):
            os.unlink(temp_filename)

Example #3

def test_zip_folder_content():
    extension_file = 'src/olympia/files/fixtures/files/extension.xpi'
    temp_filename, temp_folder = None, None
    try:
        temp_folder = utils.extract_zip(extension_file)
        assert sorted(os.listdir(temp_folder)) == [
            'chrome', 'chrome.manifest', 'install.rdf']
        temp_filename = amo.tests.get_temp_filename()
        utils.zip_folder_content(temp_folder, temp_filename)
        # Make sure the zipped files contain the same files.
        with zipfile.ZipFile(temp_filename, mode='r') as new:
            with zipfile.ZipFile(extension_file, mode='r') as orig:
                assert sorted(new.namelist()) == sorted(orig.namelist())
    finally:
        if temp_folder is not None and os.path.exists(temp_folder):
            amo.utils.rm_local_tmp_dir(temp_folder)
        if temp_filename is not None and os.path.exists(temp_filename):
            os.unlink(temp_filename)