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