127 lines
3.7 KiB
Python
127 lines
3.7 KiB
Python
|
"""Tests for distutils.cmd."""
|
||
|
|
import unittest
|
||
|
|
import os
|
||
|
|
from test.support import captured_stdout, run_unittest
|
||
|
|
|
||
|
|
from distutils.cmd import Command
|
||
|
|
from distutils.dist import Distribution
|
||
|
|
from distutils.errors import DistutilsOptionError
|
||
|
|
from distutils import debug
|
||
|
|
|
||
|
|
class MyCmd(Command):
|
||
|
|
def initialize_options(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
class CommandTestCase(unittest.TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
dist = Distribution()
|
||
|
|
self.cmd = MyCmd(dist)
|
||
|
|
|
||
|
|
def test_ensure_string_list(self):
|
||
|
|
|
||
|
|
cmd = self.cmd
|
||
|
|
cmd.not_string_list = ['one', 2, 'three']
|
||
|
|
cmd.yes_string_list = ['one', 'two', 'three']
|
||
|
|
cmd.not_string_list2 = object()
|
||
|
|
cmd.yes_string_list2 = 'ok'
|
||
|
|
cmd.ensure_string_list('yes_string_list')
|
||
|
|
cmd.ensure_string_list('yes_string_list2')
|
||
|
|
|
||
|
|
self.assertRaises(DistutilsOptionError,
|
||
|
|
cmd.ensure_string_list, 'not_string_list')
|
||
|
|
|
||
|
|
self.assertRaises(DistutilsOptionError,
|
||
|
|
cmd.ensure_string_list, 'not_string_list2')
|
||
|
|
|
||
|
|
cmd.option1 = 'ok,dok'
|
||
|
|
cmd.ensure_string_list('option1')
|
||
|
|
self.assertEqual(cmd.option1, ['ok', 'dok'])
|
||
|
|
|
||
|
|
cmd.option2 = ['xxx', 'www']
|
||
|
|
cmd.ensure_string_list('option2')
|
||
|
|
|
||
|
|
cmd.option3 = ['ok', 2]
|
||
|
|
self.assertRaises(DistutilsOptionError, cmd.ensure_string_list,
|
||
|
|
'option3')
|
||
|
|
|
||
|
|
|
||
|
|
def test_make_file(self):
|
||
|
|
|
||
|
|
cmd = self.cmd
|
||
|
|
|
||
|
|
# making sure it raises when infiles is not a string or a list/tuple
|
||
|
|
self.assertRaises(TypeError, cmd.make_file,
|
||
|
|
infiles=1, outfile='', func='func', args=())
|
||
|
|
|
||
|
|
# making sure execute gets called properly
|
||
|
|
def _execute(func, args, exec_msg, level):
|
||
|
|
self.assertEqual(exec_msg, 'generating out from in')
|
||
|
|
cmd.force = True
|
||
|
|
cmd.execute = _execute
|
||
|
|
cmd.make_file(infiles='in', outfile='out', func='func', args=())
|
||
|
|
|
||
|
|
def test_dump_options(self):
|
||
|
|
|
||
|
|
msgs = []
|
||
|
|
def _announce(msg, level):
|
||
|
|
msgs.append(msg)
|
||
|
|
cmd = self.cmd
|
||
|
|
cmd.announce = _announce
|
||
|
|
cmd.option1 = 1
|
||
|
|
cmd.option2 = 1
|
||
|
|
cmd.user_options = [('option1', '', ''), ('option2', '', '')]
|
||
|
|
cmd.dump_options()
|
||
|
|
|
||
|
|
wanted = ["command options for 'MyCmd':", ' option1 = 1',
|
||
|
|
' option2 = 1']
|
||
|
|
self.assertEqual(msgs, wanted)
|
||
|
|
|
||
|
|
def test_ensure_string(self):
|
||
|
|
cmd = self.cmd
|
||
|
|
cmd.option1 = 'ok'
|
||
|
|
cmd.ensure_string('option1')
|
||
|
|
|
||
|
|
cmd.option2 = None
|
||
|
|
cmd.ensure_string('option2', 'xxx')
|
||
|
|
self.assertTrue(hasattr(cmd, 'option2'))
|
||
|
|
|
||
|
|
cmd.option3 = 1
|
||
|
|
self.assertRaises(DistutilsOptionError, cmd.ensure_string, 'option3')
|
||
|
|
|
||
|
|
def test_ensure_filename(self):
|
||
|
|
cmd = self.cmd
|
||
|
|
cmd.option1 = __file__
|
||
|
|
cmd.ensure_filename('option1')
|
||
|
|
cmd.option2 = 'xxx'
|
||
|
|
self.assertRaises(DistutilsOptionError, cmd.ensure_filename, 'option2')
|
||
|
|
|
||
|
|
def test_ensure_dirname(self):
|
||
|
|
cmd = self.cmd
|
||
|
|
cmd.option1 = os.path.dirname(__file__) or os.curdir
|
||
|
|
cmd.ensure_dirname('option1')
|
||
|
|
cmd.option2 = 'xxx'
|
||
|
|
self.assertRaises(DistutilsOptionError, cmd.ensure_dirname, 'option2')
|
||
|
|
|
||
|
|
def test_debug_print(self):
|
||
|
|
cmd = self.cmd
|
||
|
|
with captured_stdout() as stdout:
|
||
|
|
cmd.debug_print('xxx')
|
||
|
|
stdout.seek(0)
|
||
|
|
self.assertEqual(stdout.read(), '')
|
||
|
|
|
||
|
|
debug.DEBUG = True
|
||
|
|
try:
|
||
|
|
with captured_stdout() as stdout:
|
||
|
|
cmd.debug_print('xxx')
|
||
|
|
stdout.seek(0)
|
||
|
|
self.assertEqual(stdout.read(), 'xxx\n')
|
||
|
|
finally:
|
||
|
|
debug.DEBUG = False
|
||
|
|
|
||
|
|
def test_suite():
|
||
|
|
return unittest.makeSuite(CommandTestCase)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
run_unittest(test_suite())
|