| """Tests for distutils.cmd.""" |
|
|
| import os |
| from distutils import debug |
| from distutils.cmd import Command |
| from distutils.dist import Distribution |
| from distutils.errors import DistutilsOptionError |
|
|
| import pytest |
|
|
|
|
| class MyCmd(Command): |
| def initialize_options(self): |
| pass |
|
|
|
|
| @pytest.fixture |
| def cmd(request): |
| return MyCmd(Distribution()) |
|
|
|
|
| class TestCommand: |
| def test_ensure_string_list(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') |
|
|
| with pytest.raises(DistutilsOptionError): |
| cmd.ensure_string_list('not_string_list') |
|
|
| with pytest.raises(DistutilsOptionError): |
| cmd.ensure_string_list('not_string_list2') |
|
|
| cmd.option1 = 'ok,dok' |
| cmd.ensure_string_list('option1') |
| assert cmd.option1 == ['ok', 'dok'] |
|
|
| cmd.option2 = ['xxx', 'www'] |
| cmd.ensure_string_list('option2') |
|
|
| cmd.option3 = ['ok', 2] |
| with pytest.raises(DistutilsOptionError): |
| cmd.ensure_string_list('option3') |
|
|
| def test_make_file(self, cmd): |
| |
| with pytest.raises(TypeError): |
| cmd.make_file(infiles=True, outfile='', func='func', args=()) |
|
|
| |
| def _execute(func, args, exec_msg, level): |
| assert 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, cmd): |
| msgs = [] |
|
|
| def _announce(msg, level): |
| msgs.append(msg) |
|
|
| 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'] |
| assert msgs == wanted |
|
|
| def test_ensure_string(self, cmd): |
| cmd.option1 = 'ok' |
| cmd.ensure_string('option1') |
|
|
| cmd.option2 = None |
| cmd.ensure_string('option2', 'xxx') |
| assert hasattr(cmd, 'option2') |
|
|
| cmd.option3 = 1 |
| with pytest.raises(DistutilsOptionError): |
| cmd.ensure_string('option3') |
|
|
| def test_ensure_filename(self, cmd): |
| cmd.option1 = __file__ |
| cmd.ensure_filename('option1') |
| cmd.option2 = 'xxx' |
| with pytest.raises(DistutilsOptionError): |
| cmd.ensure_filename('option2') |
|
|
| def test_ensure_dirname(self, cmd): |
| cmd.option1 = os.path.dirname(__file__) or os.curdir |
| cmd.ensure_dirname('option1') |
| cmd.option2 = 'xxx' |
| with pytest.raises(DistutilsOptionError): |
| cmd.ensure_dirname('option2') |
|
|
| def test_debug_print(self, cmd, capsys, monkeypatch): |
| cmd.debug_print('xxx') |
| assert capsys.readouterr().out == '' |
| monkeypatch.setattr(debug, 'DEBUG', True) |
| cmd.debug_print('xxx') |
| assert capsys.readouterr().out == 'xxx\n' |
|
|
