Fix CLI tests

This commit is contained in:
Micah Lee 2021-12-01 21:01:32 -08:00
parent c8ba508d26
commit b3d53ca2f2
No known key found for this signature in database
GPG key ID: 403C2657CD994F73
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