Tarek Ziadé | 11b34d0 | 2009-02-23 12:41:29 +0000 | [diff] [blame] | 1 | """Tests for distutils.command.bdist_dumb.""" |
| 2 | |
Tarek Ziadé | 11b34d0 | 2009-02-23 12:41:29 +0000 | [diff] [blame] | 3 | import os |
Éric Araujo | ae50bab | 2012-02-26 01:53:53 +0100 | [diff] [blame] | 4 | import sys |
| 5 | import zipfile |
| 6 | import unittest |
| 7 | from test.test_support import run_unittest |
Tarek Ziadé | 11b34d0 | 2009-02-23 12:41:29 +0000 | [diff] [blame] | 8 | |
Ezio Melotti | 1036a7f | 2009-09-12 14:43:43 +0000 | [diff] [blame] | 9 | # zlib is not used here, but if it's not available |
| 10 | # test_simple_built will fail |
| 11 | try: |
| 12 | import zlib |
| 13 | except ImportError: |
| 14 | zlib = None |
| 15 | |
Tarek Ziadé | 11b34d0 | 2009-02-23 12:41:29 +0000 | [diff] [blame] | 16 | from distutils.core import Distribution |
| 17 | from distutils.command.bdist_dumb import bdist_dumb |
| 18 | from distutils.tests import support |
| 19 | |
| 20 | SETUP_PY = """\ |
| 21 | from distutils.core import setup |
| 22 | import foo |
| 23 | |
| 24 | setup(name='foo', version='0.1', py_modules=['foo'], |
| 25 | url='xxx', author='xxx', author_email='xxx') |
| 26 | |
| 27 | """ |
| 28 | |
| 29 | class BuildDumbTestCase(support.TempdirManager, |
| 30 | support.LoggingSilencer, |
Tarek Ziadé | 2b06d42 | 2009-10-18 09:28:26 +0000 | [diff] [blame] | 31 | support.EnvironGuard, |
Tarek Ziadé | 11b34d0 | 2009-02-23 12:41:29 +0000 | [diff] [blame] | 32 | unittest.TestCase): |
| 33 | |
| 34 | def setUp(self): |
Tarek Ziadé | 3e3eace | 2009-05-29 08:08:07 +0000 | [diff] [blame] | 35 | super(BuildDumbTestCase, self).setUp() |
Tarek Ziadé | 11b34d0 | 2009-02-23 12:41:29 +0000 | [diff] [blame] | 36 | self.old_location = os.getcwd() |
Tarek Ziadé | 2b06d42 | 2009-10-18 09:28:26 +0000 | [diff] [blame] | 37 | self.old_sys_argv = sys.argv, sys.argv[:] |
Tarek Ziadé | 11b34d0 | 2009-02-23 12:41:29 +0000 | [diff] [blame] | 38 | |
| 39 | def tearDown(self): |
| 40 | os.chdir(self.old_location) |
Tarek Ziadé | 2b06d42 | 2009-10-18 09:28:26 +0000 | [diff] [blame] | 41 | sys.argv = self.old_sys_argv[0] |
| 42 | sys.argv[:] = self.old_sys_argv[1] |
Tarek Ziadé | 3e3eace | 2009-05-29 08:08:07 +0000 | [diff] [blame] | 43 | super(BuildDumbTestCase, self).tearDown() |
Tarek Ziadé | 11b34d0 | 2009-02-23 12:41:29 +0000 | [diff] [blame] | 44 | |
Ezio Melotti | 1036a7f | 2009-09-12 14:43:43 +0000 | [diff] [blame] | 45 | @unittest.skipUnless(zlib, "requires zlib") |
Tarek Ziadé | 11b34d0 | 2009-02-23 12:41:29 +0000 | [diff] [blame] | 46 | def test_simple_built(self): |
| 47 | |
| 48 | # let's create a simple package |
| 49 | tmp_dir = self.mkdtemp() |
| 50 | pkg_dir = os.path.join(tmp_dir, 'foo') |
| 51 | os.mkdir(pkg_dir) |
| 52 | self.write_file((pkg_dir, 'setup.py'), SETUP_PY) |
| 53 | self.write_file((pkg_dir, 'foo.py'), '#') |
| 54 | self.write_file((pkg_dir, 'MANIFEST.in'), 'include foo.py') |
| 55 | self.write_file((pkg_dir, 'README'), '') |
| 56 | |
| 57 | dist = Distribution({'name': 'foo', 'version': '0.1', |
| 58 | 'py_modules': ['foo'], |
| 59 | 'url': 'xxx', 'author': 'xxx', |
| 60 | 'author_email': 'xxx'}) |
| 61 | dist.script_name = 'setup.py' |
| 62 | os.chdir(pkg_dir) |
| 63 | |
| 64 | sys.argv = ['setup.py'] |
| 65 | cmd = bdist_dumb(dist) |
| 66 | |
| 67 | # so the output is the same no matter |
| 68 | # what is the platform |
| 69 | cmd.format = 'zip' |
| 70 | |
| 71 | cmd.ensure_finalized() |
| 72 | cmd.run() |
| 73 | |
| 74 | # see what we have |
| 75 | dist_created = os.listdir(os.path.join(pkg_dir, 'dist')) |
Éric Araujo | ae50bab | 2012-02-26 01:53:53 +0100 | [diff] [blame] | 76 | base = "%s.%s.zip" % (dist.get_fullname(), cmd.plat_name) |
Tarek Ziadé | 11b34d0 | 2009-02-23 12:41:29 +0000 | [diff] [blame] | 77 | if os.name == 'os2': |
| 78 | base = base.replace(':', '-') |
| 79 | |
Éric Araujo | ae50bab | 2012-02-26 01:53:53 +0100 | [diff] [blame] | 80 | self.assertEqual(dist_created, [base]) |
Tarek Ziadé | 11b34d0 | 2009-02-23 12:41:29 +0000 | [diff] [blame] | 81 | |
| 82 | # now let's check what we have in the zip file |
Éric Araujo | ae50bab | 2012-02-26 01:53:53 +0100 | [diff] [blame] | 83 | fp = zipfile.ZipFile(os.path.join('dist', base)) |
| 84 | try: |
| 85 | contents = fp.namelist() |
| 86 | finally: |
| 87 | fp.close() |
| 88 | |
| 89 | contents = sorted(os.path.basename(fn) for fn in contents) |
Ezio Melotti | e015b6b | 2013-03-16 20:04:44 +0200 | [diff] [blame] | 90 | wanted = ['foo-0.1-py%s.%s.egg-info' % sys.version_info[:2], 'foo.py'] |
| 91 | if not sys.dont_write_bytecode: |
| 92 | wanted.append('foo.pyc') |
Éric Araujo | ae50bab | 2012-02-26 01:53:53 +0100 | [diff] [blame] | 93 | self.assertEqual(contents, sorted(wanted)) |
Tarek Ziadé | 11b34d0 | 2009-02-23 12:41:29 +0000 | [diff] [blame] | 94 | |
Tarek Ziadé | d0ca455 | 2009-07-02 12:47:54 +0000 | [diff] [blame] | 95 | def test_finalize_options(self): |
| 96 | pkg_dir, dist = self.create_dist() |
| 97 | os.chdir(pkg_dir) |
| 98 | cmd = bdist_dumb(dist) |
Ezio Melotti | 2623a37 | 2010-11-21 13:34:58 +0000 | [diff] [blame] | 99 | self.assertEqual(cmd.bdist_dir, None) |
Tarek Ziadé | d0ca455 | 2009-07-02 12:47:54 +0000 | [diff] [blame] | 100 | cmd.finalize_options() |
| 101 | |
| 102 | # bdist_dir is initialized to bdist_base/dumb if not set |
| 103 | base = cmd.get_finalized_command('bdist').bdist_base |
Ezio Melotti | 2623a37 | 2010-11-21 13:34:58 +0000 | [diff] [blame] | 104 | self.assertEqual(cmd.bdist_dir, os.path.join(base, 'dumb')) |
Tarek Ziadé | d0ca455 | 2009-07-02 12:47:54 +0000 | [diff] [blame] | 105 | |
| 106 | # the format is set to a default value depending on the os.name |
| 107 | default = cmd.default_format[os.name] |
Ezio Melotti | 2623a37 | 2010-11-21 13:34:58 +0000 | [diff] [blame] | 108 | self.assertEqual(cmd.format, default) |
Tarek Ziadé | d0ca455 | 2009-07-02 12:47:54 +0000 | [diff] [blame] | 109 | |
Tarek Ziadé | 11b34d0 | 2009-02-23 12:41:29 +0000 | [diff] [blame] | 110 | def test_suite(): |
| 111 | return unittest.makeSuite(BuildDumbTestCase) |
| 112 | |
| 113 | if __name__ == '__main__': |
Georg Brandl | 28aeb0d | 2010-02-06 23:53:52 +0000 | [diff] [blame] | 114 | run_unittest(test_suite()) |