| """Tests for packaging.command.install.""" |
| |
| import os |
| import imp |
| import sys |
| from sysconfig import (get_scheme_names, get_config_vars, |
| _SCHEMES, get_config_var, get_path) |
| |
| from packaging.command.build_ext import build_ext |
| from packaging.command.install_dist import install_dist |
| from packaging.compiler.extension import Extension |
| from packaging.dist import Distribution |
| from packaging.errors import PackagingOptionError |
| |
| from packaging.tests import unittest, support |
| |
| |
| _CONFIG_VARS = get_config_vars() |
| |
| |
| def _make_ext_name(modname): |
| if os.name == 'nt' and sys.executable.endswith('_d.exe'): |
| modname += '_d' |
| return modname + get_config_var('SO') |
| |
| |
| class InstallTestCase(support.TempdirManager, |
| support.LoggingCatcher, |
| unittest.TestCase): |
| |
| def test_home_installation_scheme(self): |
| # This ensure two things: |
| # - that --home generates the desired set of directory names |
| # - test --home is supported on all platforms |
| builddir = self.mkdtemp() |
| destination = os.path.join(builddir, "installation") |
| |
| dist = Distribution({"name": "foopkg"}) |
| dist.command_obj["build"] = support.DummyCommand( |
| build_base=builddir, |
| build_lib=os.path.join(builddir, "lib"), |
| ) |
| |
| old_posix_prefix = _SCHEMES.get('posix_prefix', 'platinclude') |
| old_posix_home = _SCHEMES.get('posix_home', 'platinclude') |
| |
| new_path = '{platbase}/include/python{py_version_short}' |
| _SCHEMES.set('posix_prefix', 'platinclude', new_path) |
| _SCHEMES.set('posix_home', 'platinclude', '{platbase}/include/python') |
| |
| try: |
| cmd = install_dist(dist) |
| cmd.home = destination |
| cmd.ensure_finalized() |
| finally: |
| _SCHEMES.set('posix_prefix', 'platinclude', old_posix_prefix) |
| _SCHEMES.set('posix_home', 'platinclude', old_posix_home) |
| |
| self.assertEqual(cmd.install_base, destination) |
| self.assertEqual(cmd.install_platbase, destination) |
| |
| def check_path(got, expected): |
| got = os.path.normpath(got) |
| expected = os.path.normpath(expected) |
| self.assertEqual(got, expected) |
| |
| libdir = os.path.join(destination, "lib", "python") |
| check_path(cmd.install_lib, libdir) |
| check_path(cmd.install_platlib, libdir) |
| check_path(cmd.install_purelib, libdir) |
| check_path(cmd.install_headers, |
| os.path.join(destination, "include", "python", "foopkg")) |
| check_path(cmd.install_scripts, os.path.join(destination, "bin")) |
| check_path(cmd.install_data, destination) |
| |
| def test_user_site(self): |
| # test install with --user |
| # preparing the environment for the test |
| self.old_user_base = get_config_var('userbase') |
| self.old_user_site = get_path('purelib', '%s_user' % os.name) |
| self.tmpdir = self.mkdtemp() |
| self.user_base = os.path.join(self.tmpdir, 'B') |
| self.user_site = os.path.join(self.tmpdir, 'S') |
| _CONFIG_VARS['userbase'] = self.user_base |
| scheme = '%s_user' % os.name |
| _SCHEMES.set(scheme, 'purelib', self.user_site) |
| |
| def _expanduser(path): |
| if path[0] == '~': |
| path = os.path.normpath(self.tmpdir) + path[1:] |
| return path |
| |
| self.old_expand = os.path.expanduser |
| os.path.expanduser = _expanduser |
| |
| def cleanup(): |
| _CONFIG_VARS['userbase'] = self.old_user_base |
| _SCHEMES.set(scheme, 'purelib', self.old_user_site) |
| os.path.expanduser = self.old_expand |
| |
| self.addCleanup(cleanup) |
| |
| schemes = get_scheme_names() |
| for key in ('nt_user', 'posix_user', 'os2_home'): |
| self.assertIn(key, schemes) |
| |
| dist = Distribution({'name': 'xx'}) |
| cmd = install_dist(dist) |
| |
| # making sure the user option is there |
| options = [name for name, short, lable in |
| cmd.user_options] |
| self.assertIn('user', options) |
| |
| # setting a value |
| cmd.user = True |
| |
| # user base and site shouldn't be created yet |
| self.assertFalse(os.path.exists(self.user_base)) |
| self.assertFalse(os.path.exists(self.user_site)) |
| |
| # let's run finalize |
| cmd.ensure_finalized() |
| |
| # now they should |
| self.assertTrue(os.path.exists(self.user_base)) |
| self.assertTrue(os.path.exists(self.user_site)) |
| |
| self.assertIn('userbase', cmd.config_vars) |
| self.assertIn('usersite', cmd.config_vars) |
| |
| def test_handle_extra_path(self): |
| dist = Distribution({'name': 'xx', 'extra_path': 'path,dirs'}) |
| cmd = install_dist(dist) |
| |
| # two elements |
| cmd.handle_extra_path() |
| self.assertEqual(cmd.extra_path, ['path', 'dirs']) |
| self.assertEqual(cmd.extra_dirs, 'dirs') |
| self.assertEqual(cmd.path_file, 'path') |
| |
| # one element |
| cmd.extra_path = ['path'] |
| cmd.handle_extra_path() |
| self.assertEqual(cmd.extra_path, ['path']) |
| self.assertEqual(cmd.extra_dirs, 'path') |
| self.assertEqual(cmd.path_file, 'path') |
| |
| # none |
| dist.extra_path = cmd.extra_path = None |
| cmd.handle_extra_path() |
| self.assertEqual(cmd.extra_path, None) |
| self.assertEqual(cmd.extra_dirs, '') |
| self.assertEqual(cmd.path_file, None) |
| |
| # three elements (no way !) |
| cmd.extra_path = 'path,dirs,again' |
| self.assertRaises(PackagingOptionError, cmd.handle_extra_path) |
| |
| def test_finalize_options(self): |
| dist = Distribution({'name': 'xx'}) |
| cmd = install_dist(dist) |
| |
| # must supply either prefix/exec-prefix/home or |
| # install-base/install-platbase -- not both |
| cmd.prefix = 'prefix' |
| cmd.install_base = 'base' |
| self.assertRaises(PackagingOptionError, cmd.finalize_options) |
| |
| # must supply either home or prefix/exec-prefix -- not both |
| cmd.install_base = None |
| cmd.home = 'home' |
| self.assertRaises(PackagingOptionError, cmd.finalize_options) |
| |
| # can't combine user with with prefix/exec_prefix/home or |
| # install_(plat)base |
| cmd.prefix = None |
| cmd.user = 'user' |
| self.assertRaises(PackagingOptionError, cmd.finalize_options) |
| |
| def test_old_record(self): |
| # test pre-PEP 376 --record option (outside dist-info dir) |
| install_dir = self.mkdtemp() |
| project_dir, dist = self.create_dist(py_modules=['hello'], |
| scripts=['sayhi']) |
| os.chdir(project_dir) |
| self.write_file('hello.py', "def main(): print('o hai')") |
| self.write_file('sayhi', 'from hello import main; main()') |
| |
| cmd = install_dist(dist) |
| dist.command_obj['install_dist'] = cmd |
| cmd.root = install_dir |
| cmd.record = os.path.join(project_dir, 'filelist') |
| cmd.ensure_finalized() |
| cmd.run() |
| |
| with open(cmd.record) as f: |
| content = f.read() |
| |
| found = [os.path.basename(line) for line in content.splitlines()] |
| expected = ['hello.py', 'hello.%s.pyc' % imp.get_tag(), 'sayhi', |
| 'METADATA', 'INSTALLER', 'REQUESTED', 'RECORD'] |
| self.assertEqual(sorted(found), sorted(expected)) |
| |
| # XXX test that fancy_getopt is okay with options named |
| # record and no-record but unrelated |
| |
| def test_old_record_extensions(self): |
| # test pre-PEP 376 --record option with ext modules |
| install_dir = self.mkdtemp() |
| project_dir, dist = self.create_dist(ext_modules=[ |
| Extension('xx', ['xxmodule.c'])]) |
| os.chdir(project_dir) |
| support.copy_xxmodule_c(project_dir) |
| |
| buildextcmd = build_ext(dist) |
| support.fixup_build_ext(buildextcmd) |
| buildextcmd.ensure_finalized() |
| |
| cmd = install_dist(dist) |
| dist.command_obj['install_dist'] = cmd |
| dist.command_obj['build_ext'] = buildextcmd |
| cmd.root = install_dir |
| cmd.record = os.path.join(project_dir, 'filelist') |
| cmd.ensure_finalized() |
| cmd.run() |
| |
| with open(cmd.record) as f: |
| content = f.read() |
| |
| found = [os.path.basename(line) for line in content.splitlines()] |
| expected = [_make_ext_name('xx'), |
| 'METADATA', 'INSTALLER', 'REQUESTED', 'RECORD'] |
| self.assertEqual(found, expected) |
| |
| |
| def test_suite(): |
| return unittest.makeSuite(InstallTestCase) |
| |
| if __name__ == "__main__": |
| unittest.main(defaultTest="test_suite") |