mirror of https://github.com/pypa/pip
67 lines
2.6 KiB
Python
67 lines
2.6 KiB
Python
import zipfile
|
|
import textwrap
|
|
from os.path import abspath, exists, join
|
|
from pip.download import path_to_url2
|
|
from tests.lib import tests_data
|
|
from tests.lib.path import Path
|
|
from tests.lib.local_repos import local_checkout
|
|
|
|
|
|
def test_create_bundle(script):
|
|
"""
|
|
Test making a bundle. We'll grab one package from the filesystem
|
|
(the FSPkg dummy package), one from vcs (initools) and one from an
|
|
index (pip itself).
|
|
|
|
"""
|
|
fspkg = path_to_url2(Path(tests_data)/'packages'/'FSPkg')
|
|
script.pip('install', '-e', fspkg)
|
|
pkg_lines = textwrap.dedent('''\
|
|
-e %s
|
|
-e %s#egg=initools-dev
|
|
pip''' % (fspkg, local_checkout('svn+http://svn.colorstudy.com/INITools/trunk')))
|
|
script.scratch_path.join("bundle-req.txt").write(pkg_lines)
|
|
# Create a bundle in env.scratch_path/ test.pybundle
|
|
result = script.pip('bundle', '-r', script.scratch_path/ 'bundle-req.txt', script.scratch_path/ 'test.pybundle')
|
|
bundle = result.files_after.get(join('scratch', 'test.pybundle'), None)
|
|
assert bundle is not None
|
|
|
|
files = zipfile.ZipFile(bundle.full).namelist()
|
|
assert 'src/FSPkg/' in files
|
|
assert 'src/initools/' in files
|
|
assert 'build/pip/' in files
|
|
|
|
|
|
def test_cleanup_after_create_bundle(script):
|
|
"""
|
|
Test clean up after making a bundle. Make sure (build|src)-bundle/ dirs are removed but not src/.
|
|
|
|
"""
|
|
# Install an editable to create a src/ dir.
|
|
args = ['install']
|
|
args.extend(['-e',
|
|
'%s#egg=pip-test-package' %
|
|
local_checkout('git+http://github.com/pypa/pip-test-package.git')])
|
|
script.pip(*args)
|
|
build = script.venv_path/"build"
|
|
src = script.venv_path/"src"
|
|
assert not exists(build), "build/ dir still exists: %s" % build
|
|
assert exists(src), "expected src/ dir doesn't exist: %s" % src
|
|
|
|
# Make the bundle.
|
|
fspkg = 'file://%s/FSPkg' %join(tests_data, 'packages')
|
|
pkg_lines = textwrap.dedent('''\
|
|
-e %s
|
|
-e %s#egg=initools-dev
|
|
pip''' % (fspkg, local_checkout('svn+http://svn.colorstudy.com/INITools/trunk')))
|
|
script.scratch_path.join("bundle-req.txt").write(pkg_lines)
|
|
script.pip('bundle', '-r', 'bundle-req.txt', 'test.pybundle')
|
|
build_bundle = script.scratch_path/"build-bundle"
|
|
src_bundle = script.scratch_path/"src-bundle"
|
|
assert not exists(build_bundle), "build-bundle/ dir still exists: %s" % build_bundle
|
|
assert not exists(src_bundle), "src-bundle/ dir still exists: %s" % src_bundle
|
|
script.assert_no_temp()
|
|
|
|
# Make sure previously created src/ from editable still exists
|
|
assert exists(src), "expected src dir doesn't exist: %s" % src
|