1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
# coding: utf-8
"""
unittest subprocess helper
~~~~~~~~~~~~~~~~~~~~~~~~~~
:copyleft: 2015 by python-creole team, see AUTHORS for more details.
:license: GNU GPL v3 or above, see LICENSE for more details.
"""
from __future__ import print_function, unicode_literals
import json
import os
import subprocess
import sys
class SubprocessMixin(object):
# call .../env/bin/python will not add the .../env/bin/ to the PATH
SEARCH_PATH=[os.path.dirname(sys.executable)] + os.environ.get("PATH", "").split(os.pathsep)
def find_executable(self, program):
self.assertNotIn(os.sep, program)
for path in self.SEARCH_PATH:
filepath = os.path.join(path, program)
if os.path.isfile(filepath):
if not os.access(filepath, os.X_OK):
sys.stderr.write("File %r is not executable?!?\n" % filepath)
else:
return filepath
self.fail("Program %s not found in:\n\t* %s" % (json.dumps(program), "\n\t* ".join(self.SEARCH_PATH)))
def subprocess(self, popen_args, verbose=True):
assert isinstance(popen_args, (tuple, list))
if verbose:
print("Call:", popen_args)
try:
process = subprocess.Popen(popen_args,
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT,
universal_newlines=True,
)
except Exception as err:
self.fail("Error subprocess call with %r: %s" % (popen_args, err))
stdout, stderr = process.communicate()
retcode = process.poll()
if verbose:
print("return code: %r" % retcode)
print("stdout: %r" % stdout)
print("stderr: %r" % stderr)
stdout = stdout.strip()
return popen_args, retcode, stdout
def assertSubprocess(self, popen_args, retcode, stdout, verbose=True):
popen_args2, retcode2, stdout2 = self.subprocess(popen_args, verbose)
stdout = stdout.strip()
try:
self.assertEqual(stdout, stdout2, "stdout wrong:")
self.assertEqual(retcode, retcode2, "return code wrong:")
except AssertionError as err:
msg = (
"Error: %s"
"call via subprocess: %s\n"
"return code........: %r\n"
" ---------- [stdout] ---------- \n"
"%s\n"
"-------------------------------"
) % (
err,
repr(popen_args2), retcode2,
stdout2,
)
self.fail(msg)
|