Fix CLI tests

This commit is contained in:
Micah Lee 2021-12-01 21:01:32 -08:00
parent 5322d4f037
commit 6c72e9e7a5
4 changed files with 46 additions and 32 deletions

View file

@ -541,7 +541,9 @@ class ZipWriter(object):
filename.
"""
def __init__(self, common, web, zip_filename=None, processed_size_callback=None):
def __init__(
self, common, web=None, zip_filename=None, processed_size_callback=None
):
self.common = common
self.web = web
self.cancel_compression = False
@ -555,7 +557,8 @@ class ZipWriter(object):
self.zip_filename = f"{self.zip_temp_dir.name}/onionshare_{self.common.random_string(4, 6)}.zip"
# Cleanup this temp dir
self.web.cleanup_tempdirs.append(self.zip_temp_dir)
if self.web:
self.web.cleanup_tempdirs.append(self.zip_temp_dir)
self.z = zipfile.ZipFile(self.zip_filename, "w", allowZip64=True)
self.processed_size_callback = processed_size_callback