Tarek Ziade | 1231a4e | 2011-05-19 13:07:25 +0200 | [diff] [blame^] | 1 | """Tests for the uninstall command.""" |
| 2 | import os |
| 3 | import sys |
| 4 | |
| 5 | from packaging.database import disable_cache, enable_cache |
| 6 | from packaging.run import main |
| 7 | from packaging.errors import PackagingError |
| 8 | from packaging.install import remove |
| 9 | from packaging.command.install_dist import install_dist |
| 10 | |
| 11 | from packaging.tests import unittest, support |
| 12 | |
| 13 | SETUP_CFG = """ |
| 14 | [metadata] |
| 15 | name = %(name)s |
| 16 | version = %(version)s |
| 17 | |
| 18 | [files] |
| 19 | packages = |
| 20 | %(pkg)s |
| 21 | %(pkg)s.sub |
| 22 | """ |
| 23 | |
| 24 | |
| 25 | class UninstallTestCase(support.TempdirManager, |
| 26 | support.LoggingCatcher, |
| 27 | support.EnvironRestorer, |
| 28 | unittest.TestCase): |
| 29 | |
| 30 | restore_environ = ['PLAT'] |
| 31 | |
| 32 | def setUp(self): |
| 33 | super(UninstallTestCase, self).setUp() |
| 34 | self.addCleanup(setattr, sys, 'stdout', sys.stdout) |
| 35 | self.addCleanup(setattr, sys, 'stderr', sys.stderr) |
| 36 | self.addCleanup(os.chdir, os.getcwd()) |
| 37 | self.addCleanup(enable_cache) |
| 38 | self.root_dir = self.mkdtemp() |
| 39 | disable_cache() |
| 40 | |
| 41 | def run_setup(self, *args): |
| 42 | # run setup with args |
| 43 | args = ['run'] + list(args) |
| 44 | dist = main(args) |
| 45 | return dist |
| 46 | |
| 47 | def get_path(self, dist, name): |
| 48 | cmd = install_dist(dist) |
| 49 | cmd.prefix = self.root_dir |
| 50 | cmd.finalize_options() |
| 51 | return getattr(cmd, 'install_' + name) |
| 52 | |
| 53 | def make_dist(self, name='Foo', **kw): |
| 54 | kw['name'] = name |
| 55 | pkg = name.lower() |
| 56 | if 'version' not in kw: |
| 57 | kw['version'] = '0.1' |
| 58 | project_dir, dist = self.create_dist(**kw) |
| 59 | kw['pkg'] = pkg |
| 60 | |
| 61 | pkg_dir = os.path.join(project_dir, pkg) |
| 62 | os.mkdir(pkg_dir) |
| 63 | os.mkdir(os.path.join(pkg_dir, 'sub')) |
| 64 | |
| 65 | self.write_file((project_dir, 'setup.cfg'), SETUP_CFG % kw) |
| 66 | self.write_file((pkg_dir, '__init__.py'), '#') |
| 67 | self.write_file((pkg_dir, pkg + '_utils.py'), '#') |
| 68 | self.write_file((pkg_dir, 'sub', '__init__.py'), '#') |
| 69 | self.write_file((pkg_dir, 'sub', pkg + '_utils.py'), '#') |
| 70 | |
| 71 | return project_dir |
| 72 | |
| 73 | def install_dist(self, name='Foo', dirname=None, **kw): |
| 74 | if not dirname: |
| 75 | dirname = self.make_dist(name, **kw) |
| 76 | os.chdir(dirname) |
| 77 | dist = self.run_setup('install_dist', '--prefix=' + self.root_dir) |
| 78 | install_lib = self.get_path(dist, 'purelib') |
| 79 | return dist, install_lib |
| 80 | |
| 81 | def test_uninstall_unknow_distribution(self): |
| 82 | self.assertRaises(PackagingError, remove, 'Foo', |
| 83 | paths=[self.root_dir]) |
| 84 | |
| 85 | def test_uninstall(self): |
| 86 | dist, install_lib = self.install_dist() |
| 87 | self.assertIsFile(install_lib, 'foo', '__init__.py') |
| 88 | self.assertIsFile(install_lib, 'foo', 'sub', '__init__.py') |
| 89 | self.assertIsFile(install_lib, 'Foo-0.1.dist-info', 'RECORD') |
| 90 | remove('Foo', paths=[install_lib]) |
| 91 | self.assertIsNotFile(install_lib, 'foo', 'sub', '__init__.py') |
| 92 | self.assertIsNotFile(install_lib, 'Foo-0.1.dist-info', 'RECORD') |
| 93 | |
| 94 | |
| 95 | def test_suite(): |
| 96 | return unittest.makeSuite(UninstallTestCase) |
| 97 | |
| 98 | if __name__ == '__main__': |
| 99 | unittest.main(defaultTest='test_suite') |