Fix tests to point to new location of ZipWriter class

This commit is contained in:
Micah Lee 2018-09-20 23:45:13 -07:00
parent 71ea9bf29e
commit 357985fd12
No known key found for this signature in database
GPG Key ID: 403C2657CD994F73

View File

@ -64,7 +64,7 @@ def temp_file_1024_delete():
# pytest > 2.9 only needs @pytest.fixture
@pytest.yield_fixture(scope='session')
def custom_zw():
zw = web.ZipWriter(
zw = web.share_mode.ZipWriter(
common.Common(),
zip_filename=common.Common.random_string(4, 6),
processed_size_callback=lambda _: 'custom_callback'
@ -77,7 +77,7 @@ def custom_zw():
# pytest > 2.9 only needs @pytest.fixture
@pytest.yield_fixture(scope='session')
def default_zw():
zw = web.ZipWriter(common.Common())
zw = web.share_mode.ZipWriter(common.Common())
yield zw
zw.close()
tmp_dir = os.path.dirname(zw.zip_filename)