Tarek Ziadé | 6d327b0 | 2009-02-23 12:47:55 +0000 | [diff] [blame] | 1 | """Tests for distutils.command.bdist_dumb.""" |
| 2 | |
Tarek Ziadé | 6d327b0 | 2009-02-23 12:47:55 +0000 | [diff] [blame] | 3 | import os |
Éric Araujo | af2ffd7 | 2011-11-02 18:05:41 +0100 | [diff] [blame] | 4 | import imp |
| 5 | import sys |
| 6 | import zipfile |
| 7 | import unittest |
Éric Araujo | b344dd0 | 2011-02-02 21:38:37 +0000 | [diff] [blame] | 8 | from test.support import run_unittest |
Tarek Ziadé | 6d327b0 | 2009-02-23 12:47:55 +0000 | [diff] [blame] | 9 | |
| 10 | from distutils.core import Distribution |
| 11 | from distutils.command.bdist_dumb import bdist_dumb |
| 12 | from distutils.tests import support |
| 13 | |
| 14 | SETUP_PY = """\ |
| 15 | from distutils.core import setup |
| 16 | import foo |
| 17 | |
| 18 | setup(name='foo', version='0.1', py_modules=['foo'], |
| 19 | url='xxx', author='xxx', author_email='xxx') |
| 20 | |
| 21 | """ |
| 22 | |
Antoine Pitrou | 2c50a09 | 2011-03-15 21:02:59 +0100 | [diff] [blame] | 23 | try: |
| 24 | import zlib |
| 25 | ZLIB_SUPPORT = True |
| 26 | except ImportError: |
| 27 | ZLIB_SUPPORT = False |
| 28 | |
| 29 | |
Tarek Ziadé | 6d327b0 | 2009-02-23 12:47:55 +0000 | [diff] [blame] | 30 | class BuildDumbTestCase(support.TempdirManager, |
| 31 | support.LoggingSilencer, |
Tarek Ziadé | f456a15 | 2009-10-18 12:41:30 +0000 | [diff] [blame] | 32 | support.EnvironGuard, |
Tarek Ziadé | 6d327b0 | 2009-02-23 12:47:55 +0000 | [diff] [blame] | 33 | unittest.TestCase): |
| 34 | |
| 35 | def setUp(self): |
Hirokazu Yamamoto | 2703fd9 | 2009-05-29 09:14:04 +0000 | [diff] [blame] | 36 | super(BuildDumbTestCase, self).setUp() |
Tarek Ziadé | 6d327b0 | 2009-02-23 12:47:55 +0000 | [diff] [blame] | 37 | self.old_location = os.getcwd() |
Tarek Ziadé | f456a15 | 2009-10-18 12:41:30 +0000 | [diff] [blame] | 38 | self.old_sys_argv = sys.argv, sys.argv[:] |
Tarek Ziadé | 6d327b0 | 2009-02-23 12:47:55 +0000 | [diff] [blame] | 39 | |
| 40 | def tearDown(self): |
| 41 | os.chdir(self.old_location) |
Tarek Ziadé | f456a15 | 2009-10-18 12:41:30 +0000 | [diff] [blame] | 42 | sys.argv = self.old_sys_argv[0] |
| 43 | sys.argv[:] = self.old_sys_argv[1] |
Hirokazu Yamamoto | 2703fd9 | 2009-05-29 09:14:04 +0000 | [diff] [blame] | 44 | super(BuildDumbTestCase, self).tearDown() |
Tarek Ziadé | 6d327b0 | 2009-02-23 12:47:55 +0000 | [diff] [blame] | 45 | |
Antoine Pitrou | 2c50a09 | 2011-03-15 21:02:59 +0100 | [diff] [blame] | 46 | @unittest.skipUnless(ZLIB_SUPPORT, 'Need zlib support to run') |
Tarek Ziadé | 6d327b0 | 2009-02-23 12:47:55 +0000 | [diff] [blame] | 47 | def test_simple_built(self): |
| 48 | |
| 49 | # let's create a simple package |
| 50 | tmp_dir = self.mkdtemp() |
| 51 | pkg_dir = os.path.join(tmp_dir, 'foo') |
| 52 | os.mkdir(pkg_dir) |
| 53 | self.write_file((pkg_dir, 'setup.py'), SETUP_PY) |
| 54 | self.write_file((pkg_dir, 'foo.py'), '#') |
| 55 | self.write_file((pkg_dir, 'MANIFEST.in'), 'include foo.py') |
| 56 | self.write_file((pkg_dir, 'README'), '') |
| 57 | |
| 58 | dist = Distribution({'name': 'foo', 'version': '0.1', |
| 59 | 'py_modules': ['foo'], |
| 60 | 'url': 'xxx', 'author': 'xxx', |
| 61 | 'author_email': 'xxx'}) |
| 62 | dist.script_name = 'setup.py' |
| 63 | os.chdir(pkg_dir) |
| 64 | |
| 65 | sys.argv = ['setup.py'] |
| 66 | cmd = bdist_dumb(dist) |
| 67 | |
| 68 | # so the output is the same no matter |
| 69 | # what is the platform |
| 70 | cmd.format = 'zip' |
| 71 | |
| 72 | cmd.ensure_finalized() |
| 73 | cmd.run() |
| 74 | |
| 75 | # see what we have |
| 76 | dist_created = os.listdir(os.path.join(pkg_dir, 'dist')) |
Éric Araujo | af2ffd7 | 2011-11-02 18:05:41 +0100 | [diff] [blame] | 77 | base = "%s.%s.zip" % (dist.get_fullname(), cmd.plat_name) |
Tarek Ziadé | 6d327b0 | 2009-02-23 12:47:55 +0000 | [diff] [blame] | 78 | |
Éric Araujo | af2ffd7 | 2011-11-02 18:05:41 +0100 | [diff] [blame] | 79 | self.assertEqual(dist_created, [base]) |
Tarek Ziadé | 6d327b0 | 2009-02-23 12:47:55 +0000 | [diff] [blame] | 80 | |
| 81 | # now let's check what we have in the zip file |
Éric Araujo | af2ffd7 | 2011-11-02 18:05:41 +0100 | [diff] [blame] | 82 | fp = zipfile.ZipFile(os.path.join('dist', base)) |
| 83 | try: |
| 84 | contents = fp.namelist() |
| 85 | finally: |
| 86 | fp.close() |
| 87 | |
| 88 | contents = sorted(os.path.basename(fn) for fn in contents) |
Ezio Melotti | c28f6fa | 2013-03-16 19:48:51 +0200 | [diff] [blame] | 89 | wanted = ['foo-0.1-py%s.%s.egg-info' % sys.version_info[:2], 'foo.py'] |
| 90 | if not sys.dont_write_bytecode: |
| 91 | wanted.append('foo.%s.pyc' % imp.get_tag()) |
Éric Araujo | af2ffd7 | 2011-11-02 18:05:41 +0100 | [diff] [blame] | 92 | self.assertEqual(contents, sorted(wanted)) |
Tarek Ziadé | 6d327b0 | 2009-02-23 12:47:55 +0000 | [diff] [blame] | 93 | |
| 94 | def test_suite(): |
| 95 | return unittest.makeSuite(BuildDumbTestCase) |
| 96 | |
| 97 | if __name__ == '__main__': |
Éric Araujo | b344dd0 | 2011-02-02 21:38:37 +0000 | [diff] [blame] | 98 | run_unittest(test_suite()) |