def test_runscript_args(fakepopen, fakelog, fakeprocess):
proc = fakepopen.return_value
poll = proc.poll
poll.return_value = 0
read = proc.stdout.read
read.side_effect = iter('')
script.runscript('/foo/bar', 'fee', 'fi', 'fo')
popen = util.mock_call_with_name(
'',
['/foo/bar', 'fee', 'fi', 'fo'],
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT,
universal_newlines=True,
)
read = mock.call().stdout.read(1)
close = mock.call().stdout.close()
popen_calls = [
popen,
read,
close,
]
assert fakepopen.mock_calls == popen_calls
assert fakelog.mock_calls == []
assert fakeprocess.mock_calls == []
def _script(
scriptsrc,
srcdir,
linkdir,
multiprocess,
):
scriptdir = os.path.dirname(scriptsrc)
scripttail = os.path.relpath(scriptdir, srcdir)
scriptdst = os.path.join(linkdir, scripttail)
name = os.path.basename(scriptsrc)
name = os.path.join(scripttail, name)
if not os.path.exists(scriptdst):
os.makedirs(scriptdst)
log.debug(
'Running script {name}'.format(
name=name,
)
)
script.runscript(
scriptsrc,
srcdir,
linkdir,
scripttail,
name=name,
multiprocess=multiprocess,
)
def test_runscript_name(fakepopen, fakelog, fakeprocess):
proc = fakepopen.return_value
poll = proc.poll
poll.return_value = 0
read = proc.stdout.read
read.side_effect = iter('foo\n')
script.runscript('/foo/bar', name='foo-name')
popen = util.mock_call_with_name(
'',
['/foo/bar'],
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT,
universal_newlines=True,
)
read = mock.call().stdout.read(1)
reads = [read]*5
close = mock.call().stdout.close()
out_log = util.mock_call_with_name(
'info',
'foo',
extra={'source': 'SCRIPT', 'script': 'foo-name'},
)
popen_calls = [
popen,
]
popen_calls += reads
popen_calls.append(close)
log_calls = [
out_log,
]
assert fakepopen.mock_calls == popen_calls
assert fakelog.mock_calls == log_calls
assert fakeprocess.mock_calls == []
def test_runscript_args(fakepopen, fakelog, fakeprocess):
proc = fakepopen.return_value
poll = proc.poll
poll.return_value = 0
read = proc.stdout.read
read.side_effect = iter('')
script.runscript('/foo/bar', 'fee', 'fi', 'fo')
popen = util.mock_call_with_name(
'',
['/foo/bar', 'fee', 'fi', 'fo'],
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT,
universal_newlines=True,
)
read = mock.call().stdout.read(1)
close = mock.call().stdout.close()
popen_calls = [
popen,
read,
close,
]
assert fakepopen.mock_calls == popen_calls
assert fakelog.mock_calls == []
assert fakeprocess.mock_calls == []
def test_runscript_while_loop(fakepopen, fakelog, fakeprocess):
proc = fakepopen.return_value
poll = proc.poll
poll.side_effect = [None, 0, 0]
read = proc.stdout.read
def forever():
yield 'f'
yield '\n'
yield ''
yield ''
raise AssertionError('Looping forever')
read.side_effect = forever()
script.runscript('/foo/bar')
popen = util.mock_call_with_name(
'',
['/foo/bar'],
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT,
universal_newlines=True,
)
popen_calls = [
popen,
]
read = mock.call().stdout.read(1)
reads = [read]*3
popen_calls += reads
poll = mock.call().poll()
polls = [poll]*2
popen_calls += polls
popen_calls += [
mock.call().stdout.read(1),
mock.call().poll(),
]
close = mock.call().stdout.close()
popen_calls.append(close)
f_log = util.mock_call_with_name(
'info',
'f',
extra={'source': 'SCRIPT', 'script': 'bar'},
)
empty_log = util.mock_call_with_name(
'info',
'',
extra={'source': 'SCRIPT', 'script': 'bar'},
)
log_calls = [
f_log,
empty_log,
]
assert fakepopen.mock_calls == popen_calls
assert fakelog.mock_calls == log_calls
assert fakeprocess.mock_calls == []
def test_runscript_multiprocess(fakepopen, fakelog, fakeprocess):
script.runscript('/foo/bar', multiprocess=True)
assert fakepopen.mock_calls == []
assert fakelog.mock_calls == []
process = mock.call(
target=script._run,
args=(['/foo/bar'], 'bar'),
)
start = mock.call().start()
calls = [
process,
start,
]
assert fakeprocess.mock_calls == calls
def test_runscript_multiprocess(fakepopen, fakelog, fakeprocess):
script.runscript('/foo/bar', multiprocess=True)
assert fakepopen.mock_calls == []
assert fakelog.mock_calls == []
process = mock.call(
target=script._run,
args=(['/foo/bar'], 'bar'),
)
start = mock.call().start()
calls = [
process,
start,
]
assert fakeprocess.mock_calls == calls
def test_runscript_name(fakepopen, fakelog, fakeprocess):
proc = fakepopen.return_value
poll = proc.poll
poll.return_value = 0
read = proc.stdout.read
read.side_effect = iter('foo\n')
script.runscript('/foo/bar', name='foo-name')
popen = util.mock_call_with_name(
'',
['/foo/bar'],
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT,
universal_newlines=True,
)
read = mock.call().stdout.read(1)
reads = [read] * 5
close = mock.call().stdout.close()
out_log = util.mock_call_with_name(
'info',
'foo',
extra={
'source': 'SCRIPT',
'script': 'foo-name'
},
)
popen_calls = [
popen,
]
popen_calls += reads
popen_calls.append(close)
log_calls = [
out_log,
]
assert fakepopen.mock_calls == popen_calls
assert fakelog.mock_calls == log_calls
assert fakeprocess.mock_calls == []
def test_runscript_while_loop(fakepopen, fakelog, fakeprocess):
proc = fakepopen.return_value
poll = proc.poll
poll.side_effect = [None, 0, 0]
read = proc.stdout.read
def forever():
yield 'f'
yield '\n'
yield ''
yield ''
raise AssertionError('Looping forever')
read.side_effect = forever()
script.runscript('/foo/bar')
popen = util.mock_call_with_name(
'',
['/foo/bar'],
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT,
universal_newlines=True,
)
popen_calls = [
popen,
]
read = mock.call().stdout.read(1)
reads = [read] * 3
popen_calls += reads
poll = mock.call().poll()
polls = [poll] * 2
popen_calls += polls
popen_calls += [
mock.call().stdout.read(1),
mock.call().poll(),
]
close = mock.call().stdout.close()
popen_calls.append(close)
f_log = util.mock_call_with_name(
'info',
'f',
extra={
'source': 'SCRIPT',
'script': 'bar'
},
)
empty_log = util.mock_call_with_name(
'info',
'',
extra={
'source': 'SCRIPT',
'script': 'bar'
},
)
log_calls = [
f_log,
empty_log,
]
assert fakepopen.mock_calls == popen_calls
assert fakelog.mock_calls == log_calls
assert fakeprocess.mock_calls == []