def test_extract_zipfile(self, check_call):
test_data = os.path.join(os.path.dirname(__file__), 'data')
test_zip = os.path.join(test_data, 'test.zip')
extract_location = os.path.join(self.tempdir, 'extracted')
# Since we're not really creating a virtual environment, we mock the
# location of site-packages so we can confirm the files that should
# be copied there.
site_packages = os.path.join(extract_location, 'site-packages')
os.makedirs(site_packages)
with mock.patch(
'deployable_zipfile.extract.locate_virtualenv_site_packages',
return_value=site_packages):
extract_zipfile(test_zip, extract_location)
# Verify that all files are extracted properly, that the appropriate
# files are copied to site-packages, and that all unnecessary files are
# cleaned up from the extract location.
six.assertCountEqual(self, [
os.path.join(root, name)
for root, __, files in os.walk(extract_location) for name in files
], [
os.path.join(extract_location, f) for f in [
'deployable_zip/foo.txt',
'deployable_zip/subdir/bar.txt',
'site-packages/deployable_zip.pth',
'site-packages/loadenv.pth',
'site-packages/loadenv.py',
]
])
def test_extract_zipfile(self, check_call):
test_data = os.path.join(os.path.dirname(__file__), 'data')
test_zip = os.path.join(test_data, 'deployable_with_wheels.zip')
extract_location = os.path.join(self.tempdir, 'extracted')
extract_zipfile(test_zip, extract_location)
extract_location = os.path.join(extract_location, 'current')
# Verify that all files are extracted properly.
six.assertCountEqual(self, [
os.path.join(root, name)
for root, __, files in os.walk(extract_location) for name in files
], [
os.path.join(extract_location, f) for f in [
'bootstrap_wheels/pip-19.2.2-py2.py3-none-any.whl',
'bootstrap_wheels/setuptools-41.1.0-py2.py3-none-any.whl',
'deployable_zip/foo.txt',
'deployable_zip/subdir/bar.txt',
'wheels/test-0.0.1-py2.py3-none-any.whl',
]
])
# Verify that the virtual environment was created and that the setup
# script was called appropriately.
check_call.assert_has_calls([
# First call should create the virtual environment.
mock.call([
sys.executable,
'-m',
'virtualenv',
'--never-download',
'--no-wheel',
'--extra-search-dir=%s' %
os.path.join(extract_location, 'bootstrap_wheels'),
os.path.join(extract_location, 'venv'),
]),
# Second call should invoke the setup script using the Python
# binary in the virtual environment.
mock.call([
os.path.join(extract_location, 'venv', 'bin', 'python'),
os.path.join(extract_location, 'setup.py'),
])
])