78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
"""Tests for distutils.command.bdist."""
|
|
|
|
from packaging import util
|
|
from packaging.command.bdist import bdist, show_formats
|
|
|
|
from packaging.tests import unittest, support, captured_stdout
|
|
|
|
|
|
class BuildTestCase(support.TempdirManager,
|
|
support.LoggingCatcher,
|
|
unittest.TestCase):
|
|
|
|
def _mock_get_platform(self):
|
|
self._get_platform_called = True
|
|
return self._get_platform()
|
|
|
|
def setUp(self):
|
|
super(BuildTestCase, self).setUp()
|
|
|
|
# mock util.get_platform
|
|
self._get_platform_called = False
|
|
self._get_platform = util.get_platform
|
|
util.get_platform = self._mock_get_platform
|
|
|
|
def tearDown(self):
|
|
super(BuildTestCase, self).tearDown()
|
|
util.get_platform = self._get_platform
|
|
|
|
def test_formats(self):
|
|
|
|
# let's create a command and make sure
|
|
# we can fix the format
|
|
pkg_pth, dist = self.create_dist()
|
|
cmd = bdist(dist)
|
|
cmd.formats = ['msi']
|
|
cmd.ensure_finalized()
|
|
self.assertEqual(cmd.formats, ['msi'])
|
|
|
|
# what format bdist offers ?
|
|
# XXX an explicit list in bdist is
|
|
# not the best way to bdist_* commands
|
|
# we should add a registry
|
|
formats = sorted(('zip', 'gztar', 'bztar', 'ztar',
|
|
'tar', 'wininst', 'msi'))
|
|
found = sorted(cmd.format_command)
|
|
self.assertEqual(found, formats)
|
|
|
|
def test_skip_build(self):
|
|
pkg_pth, dist = self.create_dist()
|
|
cmd = bdist(dist)
|
|
cmd.skip_build = False
|
|
cmd.formats = ['ztar']
|
|
cmd.ensure_finalized()
|
|
self.assertFalse(self._get_platform_called)
|
|
|
|
pkg_pth, dist = self.create_dist()
|
|
cmd = bdist(dist)
|
|
cmd.skip_build = True
|
|
cmd.formats = ['ztar']
|
|
cmd.ensure_finalized()
|
|
self.assertTrue(self._get_platform_called)
|
|
|
|
def test_show_formats(self):
|
|
__, stdout = captured_stdout(show_formats)
|
|
|
|
# the output should be a header line + one line per format
|
|
num_formats = len(bdist.format_commands)
|
|
output = [line for line in stdout.split('\n')
|
|
if line.strip().startswith('--formats=')]
|
|
self.assertEqual(len(output), num_formats)
|
|
|
|
|
|
def test_suite():
|
|
return unittest.makeSuite(BuildTestCase)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(defaultTest='test_suite')
|