mirror of
https://github.com/python/cpython.git
synced 2025-11-04 03:44:55 +00:00
f3f283a1cb
- Use os.makedirs (I had forgotten about it!) - Let TempdirManager.write_file call os.path.join for us - Remove custom command added by test_dist - Use a skip instead of hiding a method with an underscore - Address pyflakes warnings
134 lines
4.5 KiB
Python
134 lines
4.5 KiB
Python
"""Tests for the packaging.uninstall module."""
|
|
import os
|
|
import sys
|
|
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(setattr, sys, 'stdout', sys.stdout)
|
|
self.addCleanup(setattr, sys, 'stderr', sys.stderr)
|
|
self.addCleanup(os.chdir, os.getcwd())
|
|
self.addCleanup(enable_cache)
|
|
self.root_dir = self.mkdtemp()
|
|
self.cwd = os.getcwd()
|
|
disable_cache()
|
|
|
|
def tearDown(self):
|
|
os.chdir(self.cwd)
|
|
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')
|