def test_invalid_cmd():
"""
``Command`` accepts only a list of strings.
"""
with pytest.raises(gluetool.GlueError,
match=r'^Only list of strings is accepted$'):
Command('/bin/ls').run()
with pytest.raises(
gluetool.GlueError,
match=
r"^Only list of strings is accepted, \[<type 'str'>, <type 'int'>\] found$"
):
Command(['/bin/ls', 13]).run()
def test_std_streams_mix(popen, log):
"""
Stdout and stderr are not mixed together.
"""
popen.return_value.communicate.return_value = ('This goes to stdout\n',
'This goes to stderr\n')
command = ['/bin/foo']
output = Command(command).run()
assert output.exit_code == 0
assert output.stdout == 'This goes to stdout\n'
assert output.stderr == 'This goes to stderr\n'
_assert_logging(
log,
6,
command,
stdout=
'stdout:\n---v---v---v---v---v---\nThis goes to stdout\n\n---^---^---^---^---^---',
stderr=
'stderr:\n---v---v---v---v---v---\nThis goes to stderr\n\n---^---^---^---^---^---'
)
def test_forwarding(popen, log, actual_comm, stdout, stderr):
popen.return_value.communicate.return_value = actual_comm
command = ['/bin/foo']
output = Command(command).run(stdout=stdout[0], stderr=stderr[0])
assert output.exit_code == 0
assert output.stdout == stdout[2]
assert output.stderr == stderr[2]
popen.assert_called_once_with(command, stdout=stdout[0], stderr=stderr[0])
# pylint: disable=line-too-long
expected_stdout = ' command produced no output' if stdout[
2] is None else '---v---v---v---v---v---\n{}\n---^---^---^---^---^---'.format(
stdout[2]) # Ignore PEP8Bear
expected_stderr = ' command produced no output' if stderr[
2] is None else '---v---v---v---v---v---\n{}\n---^---^---^---^---^---'.format(
stderr[2]) # Ignore PEP8Bear
_assert_logging(log,
6,
command,
stdout='stdout:\n{}'.format(expected_stdout),
stderr='stderr:\n{}'.format(expected_stderr))
def test_sanity(popen, log):
"""
Basic usage - run a command, and log its output.
"""
popen.return_value.communicate.return_value = ('root listing', '')
command = [u'/bin/ls', u'/']
output = Command(command).run()
assert output.exit_code == 0
assert output.stdout == 'root listing'
assert output.stderr == ''
popen.assert_called_once_with(command,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
_assert_logging(log, 8, command)
assert log.records[
4].message == 'stdout: (See "verbose" log for the actual message)'
assert log.records[
5].message == 'stdout:\n---v---v---v---v---v---\nroot listing\n---^---^---^---^---^---'
assert log.records[
6].message == 'stderr: (See "verbose" log for the actual message)'
assert log.records[
7].message == 'stderr:\n---v---v---v---v---v---\n\n---^---^---^---^---^---'
def test_std_streams_mix(popen, log):
"""
Stdout and stderr are not mixed together.
"""
popen.return_value.communicate.return_value = ('This goes to stdout\n',
'This goes to stderr\n')
command = [u'/bin/foo']
output = Command(command).run()
assert output.exit_code == 0
assert output.stdout == 'This goes to stdout\n'
assert output.stderr == 'This goes to stderr\n'
_assert_logging(log, 8, command)
assert log.records[
4].message == 'stdout: (See "verbose" log for the actual message)'
assert log.records[
5].message == 'stdout:\n---v---v---v---v---v---\nThis goes to stdout\n\n---^---^---^---^---^---'
assert log.records[
6].message == 'stderr: (See "verbose" log for the actual message)'
assert log.records[
7].message == 'stderr:\n---v---v---v---v---v---\nThis goes to stderr\n\n---^---^---^---^---^---'
def test_invalid_cmd():
"""
``Command`` accepts only a list of strings.
"""
with pytest.raises(gluetool.GlueError,
match=r'^Only list of strings is accepted$'):
Command('/bin/ls').run()
# These yield different string in Python 3 and Python 2 - "class ..." vs "type ..."
type_str = str(type(''))
type_int = str(type(1))
with pytest.raises(
gluetool.GlueError,
match=
r"^Only list of strings is accepted, \[\('/bin/ls', {}\), \(13, {}\)\] found$"
.format(type_str, type_int)):
Command(['/bin/ls', 13]).run()
def test_forwarding(popen, log, actual_comm, stdout, stderr):
stdout_arg, stdout_output = stdout
stderr_arg, stderr_output = stderr
popen.return_value.communicate.return_value = actual_comm
command = [u'/bin/foo']
output = Command(command).run(stdout=stdout_arg, stderr=stderr_arg)
assert output.exit_code == 0
assert output.stdout == stdout_output
assert output.stderr == stderr_output
popen.assert_called_once_with(command,
stdout=stdout_arg,
stderr=stderr_arg)
_assert_logging(log, 7, command)
index = 4
if stdout_output is None:
assert log.records[
index].message == 'stdout:\n command produced no output'
index += 1
else:
assert log.records[
index].message == 'stdout: (See "verbose" log for the actual message)'
assert log.records[
index +
1].message == 'stdout:\n---v---v---v---v---v---\n{}\n---^---^---^---^---^---'.format(
stdout_output)
index += 2
if stderr_output is None:
assert log.records[
index].message == 'stderr:\n command produced no output'
else:
assert log.records[
index].message == 'stderr: (See "verbose" log for the actual message)'
assert log.records[
index +
1].message == 'stderr:\n---v---v---v---v---v---\n{}\n---^---^---^---^---^---'.format(
stderr_output)
def test_invalid_stdout(popen, log):
def throw(*args, **kwargs):
# pylint: disable=unused-argument
raise AttributeError("'tuple' object has no attribute 'fileno'")
popen.side_effect = throw
command = ['/bin/foo']
with pytest.raises(AttributeError,
match=r"^'tuple' object has no attribute 'fileno'$"):
Command(command).run(stdout=(13, 17))
_assert_logging(log, 3, command)
def test_oserror(popen, log, actual_errno, expected_exc, expected_message):
def throw(*args, **kwargs):
# pylint: disable=unused-argument
exc = OSError(expected_message)
exc.errno = actual_errno
raise exc
popen.side_effect = throw
command = ['/bin/foo']
with pytest.raises(expected_exc, match=r'^{}$'.format(expected_message)):
Command(command).run()
_assert_logging(log, 3, command)
def test_exit_code_error(popen, log):
"""
Command exited with non-zero exit code.
"""
popen.return_value.poll.return_value = 1
command = ['/bin/foo']
with pytest.raises(gluetool.GlueCommandError, match=r"^Command '\['/bin/foo'\]' failed with exit code 1$") \
as excinfo:
Command(command).run()
_assert_logging(log, 6, command)
assert log.records[4].message == 'stdout:\n command produced no output'
assert log.records[5].message == 'stderr:\n command produced no output'
assert excinfo.value.output.exit_code == 1
assert excinfo.value.output.stdout is None
assert excinfo.value.output.stderr is None
def test_sanity(popen, log):
"""
Basic usage - run a command, and log its output.
"""
popen.return_value.communicate.return_value = ('root listing', '')
command = ['/bin/ls', '/']
output = Command(command).run()
assert output.exit_code == 0
assert output.stdout == 'root listing'
assert output.stderr == ''
popen.assert_called_once_with(command,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
_assert_logging(
log,
6,
command,
stdout=
'stdout:\n---v---v---v---v---v---\nroot listing\n---^---^---^---^---^---',
stderr='stderr:\n---v---v---v---v---v---\n\n---^---^---^---^---^---')
