mirror of
https://github.com/python/cpython.git
synced 2025-08-03 00:23:06 +00:00
fad46e19b4
Running with regrtest does not show spurious output or unrestored sys.std* objects; sometimes running with make test is different, I’ll watch the buildbots. In addition, update the create module to use logging.
129 lines
4.3 KiB
Python
129 lines
4.3 KiB
Python
"""Tests for the packaging.uninstall module."""
|
|
import os
|
|
import logging
|
|
import packaging.util
|
|
|
|
from packaging.errors import PackagingError
|
|
from packaging.install import remove
|
|
from packaging.database import disable_cache, enable_cache
|
|
|
|
from packaging.tests import unittest, support
|
|
|
|
SETUP_CFG = """
|
|
[metadata]
|
|
name = %(name)s
|
|
version = %(version)s
|
|
|
|
[files]
|
|
packages =
|
|
%(pkg)s
|
|
%(pkg)s.sub
|
|
"""
|
|
|
|
|
|
class UninstallTestCase(support.TempdirManager,
|
|
support.LoggingCatcher,
|
|
support.EnvironRestorer,
|
|
unittest.TestCase):
|
|
|
|
restore_environ = ['PLAT']
|
|
|
|
def setUp(self):
|
|
super(UninstallTestCase, self).setUp()
|
|
self.addCleanup(enable_cache)
|
|
self.root_dir = self.mkdtemp()
|
|
self.cwd = os.getcwd()
|
|
disable_cache()
|
|
|
|
def tearDown(self):
|
|
packaging.util._path_created.clear()
|
|
super(UninstallTestCase, self).tearDown()
|
|
|
|
def get_path(self, dist, name):
|
|
# the dist argument must contain an install_dist command correctly
|
|
# initialized with a prefix option and finalized befored this method
|
|
# can be called successfully; practically, this means that you should
|
|
# call self.install_dist before self.get_path
|
|
cmd = dist.get_command_obj('install_dist')
|
|
return getattr(cmd, 'install_' + name)
|
|
|
|
def make_dist(self, name='Foo', **kw):
|
|
kw['name'] = name
|
|
pkg = name.lower()
|
|
if 'version' not in kw:
|
|
kw['version'] = '0.1'
|
|
project_dir, dist = self.create_dist(**kw)
|
|
kw['pkg'] = pkg
|
|
|
|
pkg_dir = os.path.join(project_dir, pkg)
|
|
os.makedirs(os.path.join(pkg_dir, 'sub'))
|
|
|
|
self.write_file((project_dir, 'setup.cfg'), SETUP_CFG % kw)
|
|
self.write_file((pkg_dir, '__init__.py'), '#')
|
|
self.write_file((pkg_dir, pkg + '_utils.py'), '#')
|
|
self.write_file((pkg_dir, 'sub', '__init__.py'), '#')
|
|
self.write_file((pkg_dir, 'sub', pkg + '_utils.py'), '#')
|
|
|
|
return project_dir
|
|
|
|
def install_dist(self, name='Foo', dirname=None, **kw):
|
|
if not dirname:
|
|
dirname = self.make_dist(name, **kw)
|
|
os.chdir(dirname)
|
|
|
|
dist = support.TestDistribution()
|
|
# for some unfathomable reason, the tests will fail horribly if the
|
|
# parse_config_files method is not called, even if it doesn't do
|
|
# anything useful; trying to build and use a command object manually
|
|
# also fails
|
|
dist.parse_config_files()
|
|
dist.finalize_options()
|
|
dist.run_command('install_dist',
|
|
{'prefix': ('command line', self.root_dir)})
|
|
|
|
site_packages = self.get_path(dist, 'purelib')
|
|
return dist, site_packages
|
|
|
|
def test_uninstall_unknown_distribution(self):
|
|
dist, site_packages = self.install_dist('Foospam')
|
|
self.assertRaises(PackagingError, remove, 'Foo',
|
|
paths=[site_packages])
|
|
|
|
def test_uninstall(self):
|
|
dist, site_packages = self.install_dist()
|
|
self.assertIsFile(site_packages, 'foo', '__init__.py')
|
|
self.assertIsFile(site_packages, 'foo', 'sub', '__init__.py')
|
|
self.assertIsFile(site_packages, 'Foo-0.1.dist-info', 'RECORD')
|
|
self.assertTrue(remove('Foo', paths=[site_packages]))
|
|
self.assertIsNotFile(site_packages, 'foo', 'sub', '__init__.py')
|
|
self.assertIsNotFile(site_packages, 'Foo-0.1.dist-info', 'RECORD')
|
|
|
|
def test_uninstall_error_handling(self):
|
|
# makes sure if there are OSErrors (like permission denied)
|
|
# remove() stops and displays a clean error
|
|
dist, site_packages = self.install_dist('Meh')
|
|
|
|
# breaking os.rename
|
|
old = os.rename
|
|
|
|
def _rename(source, target):
|
|
raise OSError(42, 'impossible operation')
|
|
|
|
os.rename = _rename
|
|
try:
|
|
self.assertFalse(remove('Meh', paths=[site_packages]))
|
|
finally:
|
|
os.rename = old
|
|
|
|
logs = [log for log in self.get_logs(logging.INFO)
|
|
if log.startswith('Error:')]
|
|
self.assertEqual(logs, ['Error: [Errno 42] impossible operation'])
|
|
|
|
self.assertTrue(remove('Meh', paths=[site_packages]))
|
|
|
|
|
|
def test_suite():
|
|
return unittest.makeSuite(UninstallTestCase)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(defaultTest='test_suite')
|