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