Tarek Ziadé | f072122 | 2009-02-28 10:16:43 +0000 | [diff] [blame] | 1 | """Tests for distutils.command.bdist_rpm.""" |
| 2 | |
| 3 | import unittest |
| 4 | import sys |
| 5 | import os |
| 6 | import tempfile |
| 7 | import shutil |
Éric Araujo | b344dd0 | 2011-02-02 21:38:37 +0000 | [diff] [blame^] | 8 | from test.support import run_unittest |
Tarek Ziadé | f072122 | 2009-02-28 10:16:43 +0000 | [diff] [blame] | 9 | |
| 10 | from distutils.core import Distribution |
| 11 | from distutils.command.bdist_rpm import bdist_rpm |
| 12 | from distutils.tests import support |
| 13 | from distutils.spawn import find_executable |
| 14 | from distutils import spawn |
| 15 | from distutils.errors import DistutilsExecError |
| 16 | |
| 17 | SETUP_PY = """\ |
| 18 | from distutils.core import setup |
| 19 | import foo |
| 20 | |
| 21 | setup(name='foo', version='0.1', py_modules=['foo'], |
| 22 | url='xxx', author='xxx', author_email='xxx') |
| 23 | |
| 24 | """ |
| 25 | |
| 26 | class BuildRpmTestCase(support.TempdirManager, |
| 27 | support.LoggingSilencer, |
| 28 | unittest.TestCase): |
| 29 | |
| 30 | def setUp(self): |
| 31 | super(BuildRpmTestCase, self).setUp() |
| 32 | self.old_location = os.getcwd() |
Tarek Ziadé | f456a15 | 2009-10-18 12:41:30 +0000 | [diff] [blame] | 33 | self.old_sys_argv = sys.argv, sys.argv[:] |
Tarek Ziadé | f072122 | 2009-02-28 10:16:43 +0000 | [diff] [blame] | 34 | |
| 35 | def tearDown(self): |
| 36 | os.chdir(self.old_location) |
Tarek Ziadé | f456a15 | 2009-10-18 12:41:30 +0000 | [diff] [blame] | 37 | sys.argv = self.old_sys_argv[0] |
| 38 | sys.argv[:] = self.old_sys_argv[1] |
Tarek Ziadé | f072122 | 2009-02-28 10:16:43 +0000 | [diff] [blame] | 39 | super(BuildRpmTestCase, self).tearDown() |
| 40 | |
| 41 | def test_quiet(self): |
| 42 | |
| 43 | # XXX I am unable yet to make this test work without |
| 44 | # spurious sdtout/stderr output under Mac OS X |
| 45 | if sys.platform != 'linux2': |
| 46 | return |
| 47 | |
| 48 | # this test will run only if the rpm commands are found |
| 49 | if (find_executable('rpm') is None or |
| 50 | find_executable('rpmbuild') is None): |
| 51 | return |
| 52 | |
| 53 | # let's create a package |
| 54 | tmp_dir = self.mkdtemp() |
| 55 | pkg_dir = os.path.join(tmp_dir, 'foo') |
| 56 | os.mkdir(pkg_dir) |
| 57 | self.write_file((pkg_dir, 'setup.py'), SETUP_PY) |
| 58 | self.write_file((pkg_dir, 'foo.py'), '#') |
| 59 | self.write_file((pkg_dir, 'MANIFEST.in'), 'include foo.py') |
| 60 | self.write_file((pkg_dir, 'README'), '') |
| 61 | |
| 62 | dist = Distribution({'name': 'foo', 'version': '0.1', |
| 63 | 'py_modules': ['foo'], |
| 64 | 'url': 'xxx', 'author': 'xxx', |
| 65 | 'author_email': 'xxx'}) |
| 66 | dist.script_name = 'setup.py' |
| 67 | os.chdir(pkg_dir) |
| 68 | |
| 69 | sys.argv = ['setup.py'] |
| 70 | cmd = bdist_rpm(dist) |
| 71 | cmd.fix_python = True |
| 72 | |
| 73 | # running in quiet mode |
| 74 | cmd.quiet = 1 |
| 75 | cmd.ensure_finalized() |
| 76 | cmd.run() |
| 77 | |
| 78 | dist_created = os.listdir(os.path.join(pkg_dir, 'dist')) |
Georg Brandl | ab91fde | 2009-08-13 08:51:18 +0000 | [diff] [blame] | 79 | self.assertTrue('foo-0.1-1.noarch.rpm' in dist_created) |
Tarek Ziadé | f072122 | 2009-02-28 10:16:43 +0000 | [diff] [blame] | 80 | |
| 81 | def test_no_optimize_flag(self): |
| 82 | |
| 83 | # XXX I am unable yet to make this test work without |
| 84 | # spurious sdtout/stderr output under Mac OS X |
| 85 | if sys.platform != 'linux2': |
| 86 | return |
| 87 | |
| 88 | # http://bugs.python.org/issue1533164 |
| 89 | # this test will run only if the rpm command is found |
| 90 | if (find_executable('rpm') is None or |
| 91 | find_executable('rpmbuild') is None): |
| 92 | return |
| 93 | |
| 94 | # let's create a package that brakes bdist_rpm |
| 95 | tmp_dir = self.mkdtemp() |
| 96 | pkg_dir = os.path.join(tmp_dir, 'foo') |
| 97 | os.mkdir(pkg_dir) |
| 98 | self.write_file((pkg_dir, 'setup.py'), SETUP_PY) |
| 99 | self.write_file((pkg_dir, 'foo.py'), '#') |
| 100 | self.write_file((pkg_dir, 'MANIFEST.in'), 'include foo.py') |
| 101 | self.write_file((pkg_dir, 'README'), '') |
| 102 | |
| 103 | dist = Distribution({'name': 'foo', 'version': '0.1', |
| 104 | 'py_modules': ['foo'], |
| 105 | 'url': 'xxx', 'author': 'xxx', |
| 106 | 'author_email': 'xxx'}) |
| 107 | dist.script_name = 'setup.py' |
| 108 | os.chdir(pkg_dir) |
| 109 | |
| 110 | sys.argv = ['setup.py'] |
| 111 | cmd = bdist_rpm(dist) |
| 112 | cmd.fix_python = True |
| 113 | |
Tarek Ziadé | f072122 | 2009-02-28 10:16:43 +0000 | [diff] [blame] | 114 | cmd.quiet = 1 |
| 115 | cmd.ensure_finalized() |
| 116 | cmd.run() |
| 117 | |
| 118 | dist_created = os.listdir(os.path.join(pkg_dir, 'dist')) |
Georg Brandl | ab91fde | 2009-08-13 08:51:18 +0000 | [diff] [blame] | 119 | self.assertTrue('foo-0.1-1.noarch.rpm' in dist_created) |
Tarek Ziadé | f072122 | 2009-02-28 10:16:43 +0000 | [diff] [blame] | 120 | os.remove(os.path.join(pkg_dir, 'dist', 'foo-0.1-1.noarch.rpm')) |
| 121 | |
Tarek Ziadé | f072122 | 2009-02-28 10:16:43 +0000 | [diff] [blame] | 122 | def test_suite(): |
| 123 | return unittest.makeSuite(BuildRpmTestCase) |
| 124 | |
| 125 | if __name__ == '__main__': |
Éric Araujo | b344dd0 | 2011-02-02 21:38:37 +0000 | [diff] [blame^] | 126 | run_unittest(test_suite()) |