Tarek Ziade | 1231a4e | 2011-05-19 13:07:25 +0200 | [diff] [blame] | 1 | """Tests for the uninstall command.""" |
| 2 | import os |
| 3 | import sys |
Tarek Ziade | 2da7565 | 2011-05-30 12:25:21 +0200 | [diff] [blame] | 4 | from io import StringIO |
Tarek Ziade | f47fa58 | 2011-05-30 23:26:51 +0200 | [diff] [blame] | 5 | import stat |
Éric Araujo | c06f46f | 2011-06-17 15:43:18 +0200 | [diff] [blame] | 6 | import packaging.util |
Tarek Ziade | 1231a4e | 2011-05-19 13:07:25 +0200 | [diff] [blame] | 7 | |
| 8 | from packaging.database import disable_cache, enable_cache |
| 9 | from packaging.run import main |
| 10 | from packaging.errors import PackagingError |
| 11 | from packaging.install import remove |
| 12 | from packaging.command.install_dist import install_dist |
| 13 | |
| 14 | from packaging.tests import unittest, support |
| 15 | |
| 16 | SETUP_CFG = """ |
| 17 | [metadata] |
| 18 | name = %(name)s |
| 19 | version = %(version)s |
| 20 | |
| 21 | [files] |
| 22 | packages = |
| 23 | %(pkg)s |
| 24 | %(pkg)s.sub |
| 25 | """ |
| 26 | |
| 27 | |
| 28 | class UninstallTestCase(support.TempdirManager, |
| 29 | support.LoggingCatcher, |
| 30 | support.EnvironRestorer, |
| 31 | unittest.TestCase): |
| 32 | |
| 33 | restore_environ = ['PLAT'] |
| 34 | |
| 35 | def setUp(self): |
| 36 | super(UninstallTestCase, self).setUp() |
| 37 | self.addCleanup(setattr, sys, 'stdout', sys.stdout) |
| 38 | self.addCleanup(setattr, sys, 'stderr', sys.stderr) |
| 39 | self.addCleanup(os.chdir, os.getcwd()) |
| 40 | self.addCleanup(enable_cache) |
| 41 | self.root_dir = self.mkdtemp() |
Tarek Ziade | e1177d9 | 2011-05-23 19:07:56 +0200 | [diff] [blame] | 42 | self.cwd = os.getcwd() |
Tarek Ziade | 1231a4e | 2011-05-19 13:07:25 +0200 | [diff] [blame] | 43 | disable_cache() |
| 44 | |
Tarek Ziade | e1177d9 | 2011-05-23 19:07:56 +0200 | [diff] [blame] | 45 | def tearDown(self): |
| 46 | os.chdir(self.cwd) |
Éric Araujo | c06f46f | 2011-06-17 15:43:18 +0200 | [diff] [blame] | 47 | packaging.util._path_created.clear() |
Tarek Ziade | e1177d9 | 2011-05-23 19:07:56 +0200 | [diff] [blame] | 48 | super(UninstallTestCase, self).tearDown() |
| 49 | |
Tarek Ziade | 1231a4e | 2011-05-19 13:07:25 +0200 | [diff] [blame] | 50 | def run_setup(self, *args): |
| 51 | # run setup with args |
| 52 | args = ['run'] + list(args) |
| 53 | dist = main(args) |
| 54 | return dist |
| 55 | |
| 56 | def get_path(self, dist, name): |
| 57 | cmd = install_dist(dist) |
| 58 | cmd.prefix = self.root_dir |
| 59 | cmd.finalize_options() |
| 60 | return getattr(cmd, 'install_' + name) |
| 61 | |
| 62 | def make_dist(self, name='Foo', **kw): |
| 63 | kw['name'] = name |
| 64 | pkg = name.lower() |
| 65 | if 'version' not in kw: |
| 66 | kw['version'] = '0.1' |
| 67 | project_dir, dist = self.create_dist(**kw) |
| 68 | kw['pkg'] = pkg |
| 69 | |
| 70 | pkg_dir = os.path.join(project_dir, pkg) |
| 71 | os.mkdir(pkg_dir) |
| 72 | os.mkdir(os.path.join(pkg_dir, 'sub')) |
| 73 | |
| 74 | self.write_file((project_dir, 'setup.cfg'), SETUP_CFG % kw) |
| 75 | self.write_file((pkg_dir, '__init__.py'), '#') |
| 76 | self.write_file((pkg_dir, pkg + '_utils.py'), '#') |
| 77 | self.write_file((pkg_dir, 'sub', '__init__.py'), '#') |
| 78 | self.write_file((pkg_dir, 'sub', pkg + '_utils.py'), '#') |
| 79 | |
| 80 | return project_dir |
| 81 | |
| 82 | def install_dist(self, name='Foo', dirname=None, **kw): |
| 83 | if not dirname: |
| 84 | dirname = self.make_dist(name, **kw) |
| 85 | os.chdir(dirname) |
Tarek Ziade | 1cc9bbd | 2011-05-30 19:16:49 +0200 | [diff] [blame] | 86 | old_out = sys.stderr |
Tarek Ziade | 2da7565 | 2011-05-30 12:25:21 +0200 | [diff] [blame] | 87 | sys.stderr = StringIO() |
Tarek Ziade | f47fa58 | 2011-05-30 23:26:51 +0200 | [diff] [blame] | 88 | dist = self.run_setup('install_dist', '--prefix=' + self.root_dir) |
Tarek Ziade | 1231a4e | 2011-05-19 13:07:25 +0200 | [diff] [blame] | 89 | install_lib = self.get_path(dist, 'purelib') |
| 90 | return dist, install_lib |
| 91 | |
| 92 | def test_uninstall_unknow_distribution(self): |
| 93 | self.assertRaises(PackagingError, remove, 'Foo', |
| 94 | paths=[self.root_dir]) |
| 95 | |
Tarek Ziade | 1231a4e | 2011-05-19 13:07:25 +0200 | [diff] [blame] | 96 | def test_uninstall(self): |
| 97 | dist, install_lib = self.install_dist() |
| 98 | self.assertIsFile(install_lib, 'foo', '__init__.py') |
| 99 | self.assertIsFile(install_lib, 'foo', 'sub', '__init__.py') |
| 100 | self.assertIsFile(install_lib, 'Foo-0.1.dist-info', 'RECORD') |
Tarek Ziade | f47fa58 | 2011-05-30 23:26:51 +0200 | [diff] [blame] | 101 | self.assertTrue(remove('Foo', paths=[install_lib])) |
Tarek Ziade | 1231a4e | 2011-05-19 13:07:25 +0200 | [diff] [blame] | 102 | self.assertIsNotFile(install_lib, 'foo', 'sub', '__init__.py') |
| 103 | self.assertIsNotFile(install_lib, 'Foo-0.1.dist-info', 'RECORD') |
| 104 | |
Tarek Ziade | f47fa58 | 2011-05-30 23:26:51 +0200 | [diff] [blame] | 105 | def test_remove_issue(self): |
| 106 | # makes sure if there are OSErrors (like permission denied) |
| 107 | # remove() stops and display a clean error |
| 108 | dist, install_lib = self.install_dist('Meh') |
| 109 | |
| 110 | # breaking os.rename |
| 111 | old = os.rename |
| 112 | |
| 113 | def _rename(source, target): |
Éric Araujo | 2ef747c | 2011-06-04 22:33:16 +0200 | [diff] [blame] | 114 | raise OSError |
Tarek Ziade | f47fa58 | 2011-05-30 23:26:51 +0200 | [diff] [blame] | 115 | |
| 116 | os.rename = _rename |
| 117 | try: |
| 118 | self.assertFalse(remove('Meh', paths=[install_lib])) |
| 119 | finally: |
| 120 | os.rename = old |
| 121 | |
| 122 | self.assertTrue(remove('Meh', paths=[install_lib])) |
| 123 | |
Tarek Ziade | 1231a4e | 2011-05-19 13:07:25 +0200 | [diff] [blame] | 124 | |
| 125 | def test_suite(): |
| 126 | return unittest.makeSuite(UninstallTestCase) |
| 127 | |
| 128 | if __name__ == '__main__': |
| 129 | unittest.main(defaultTest='test_suite') |