def test_processrunner_correct_stdout_count(self, autostart, manualstart):
testLen = 10000
command = [
sys.executable, self.sampleCommandPath, "--lines",
"{}".format(testLen), "--block", "1", "--sleep", "0"
]
with ProcessRunner(command, autostart=autostart) as proc:
# Verify start works when placed before collectLines
if manualstart:
proc.start()
# Store process output
output = proc.collectLines()
result = proc.wait().poll()
length = len(output)
# Verify the count of lines is identical
message = 'Count of output lines unexpected: Expected ' \
'{} got {}. Top 20:\n{}'.format(testLen,
length,
"\n".join(output[:20]))
self.assertEqual(length, testLen, message)
# Verify the test script exit code was 0
self.assertEqual(result, 0, 'Test script return code not zero')
def test_processrunner_collectLines_raise_NotStarted(self):
"""Ensure NotStarted is raised when using collectLines before
start()"""
command = ["echo", "bonjour"]
with self.assertRaises(NotStarted):
with ProcessRunner(command, autostart=False) as proc:
output = proc.collectLines()
def test_processrunner_second_start_raises_exception_once(
self, autostart, start_a, start_b):
command = ["echo", "bonjour"]
with ProcessRunner(command, autostart=autostart) as proc:
# Duplicate proc.starts()
with self.assertRaises(AlreadyStarted):
if start_a:
proc.start()
if start_b:
proc.start()
def test_processrunner_check_mixed_ascii_emoji_content(self):
textIn = "a😂" * 10
command = ["echo", textIn]
with ProcessRunner(command) as proc:
textOut = proc.collectLines()[0]
self.assertEqual(
textIn, textOut,
"Returned unicode text is not equivalent: In {}, Out {}".format(
textIn, textOut))
def test_processrunner_correct_stdout_count(self):
testLen = 10000
command = [self.sampleCommandPath, "--lines", str(testLen), "--block", "1", "--sleep", "0"]
proc = ProcessRunner(command)
output = proc.collectLines()
result = proc.wait().poll()
proc.terminate()
proc.shutdown()
length = len(output)
self.assertEqual(length, testLen,
'Count of output lines unexpected: Expected '+str(testLen)+' got ' + str(length))
self.assertEqual(result, 0,
'Test script return code not zero')
def test_processrunner_onek_check_content(self):
textIn = "a" * 2**10 # "a" repeated 1,024 times
with NamedTemporaryFile() as tempFile:
tempFile.write(textIn.encode("utf-8"))
tempFile.flush()
command = ["cat", tempFile.name]
with ProcessRunner(command) as proc:
textOut = proc.collectLines()[0]
self.assertEqual(
len(textIn), len(textOut),
"Returned text not the same length: in {}, out {}".format(
len(textIn), len(textOut)))
def run():
proc = ProcessRunner(command)
# Key aspect
# When using the threading library, and WriteOut writes to a pipe, the return code
# doesn't always come back as expected
# Seems to be fixed when using multiprocessing in place of threading
with open("/dev/null", 'a') as devnull:
proc.mapLines(WriteOut(pipe=devnull,
outputPrefix="validation-stdout> "),
procPipeName="stdout")
proc.wait()
result = proc.poll()
if result != 1:
print ""
print "Result output isn't 1!: '" + str(result) + "'"
print "Waiting another moment..."
time.sleep(1)
print "Next Poll(): " + str(proc.poll())
proc.terminate()
return result
def test_processrunner_check_onem_emoji_content(self):
"""Checks the integrity of the inter-process locking mechanisms with unicode text"""
textIn = "😂" * 2**22 # a * 2**20 is "a" repeated 1,048,576 times, 22 is 4,194,304
with NamedTemporaryFile() as tempFile:
tempFile.write(textIn.encode("utf-8"))
tempFile.flush()
command = ["cat", tempFile.name]
with ProcessRunner(command) as proc:
textOut = proc.collectLines()[
0] # < Sometimes fails with index out of range
self.assertEqual(
len(textIn), len(textOut),
"Returned text not the same length: in {}, out {}".format(
len(textIn), len(textOut)))
def test_processrunner_leak_check(self):
limit = 100
command = ["echo", "bonjour"]
openFilesCommand = ["lsof", "-p", str(os.getpid())]
openFilesStart = runCommand(openFilesCommand, returnAllContent=True)
for i in range(limit):
self.spinner.spin()
with ProcessRunner(command) as proc:
discard = proc.collectLines()
openFilesEnd = runCommand(openFilesCommand, returnAllContent=True)
self.assertEqual(
len(openFilesStart), len(openFilesEnd),
"Open file count changed: Start {} to End {}".format(
len(openFilesStart), len(openFilesEnd)))
def test_processrunner_second_start_raises_exception_repeated(
self, autostart):
testLen = 5
command = ["echo", "bonjour"]
with ProcessRunner(command, autostart=autostart) as proc:
# Trigger the process to start if it wasn't done by autostart
if not autostart:
proc.start()
# Run start N times
# Duplicate proc.starts()
for i in range(0, testLen):
try:
proc.start()
except AlreadyStarted:
pass
# Trigger one more that is the test
with self.assertRaises(AlreadyStarted):
proc.start()
def run():
proc = ProcessRunner(command)
# Key aspect
# When using the threading library, and writeOut writes to a pipe, the return code
# doesn't always come back as expected
# Isn't fixed even after the switch to multiprocessing
with open("/dev/null", 'a') as devnull:
proc.mapLines(writeOut(pipe=devnull, outputPrefix="test-output-script.py-stdout> "), procPipeName="stdout")
proc.mapLines(writeOut(pipe=sys.stderr, outputPrefix="test-output-script.py-stderr> "), procPipeName="stderr")
proc.wait()
result = proc.poll()
if result != 1:
print("")
print("Result output isn't 1!: '" + str(result) + "'")
print("Waiting another moment...")
time.sleep(.1)
result = proc.poll()
print("Next Poll(): " + str(result))
# proc.terminate()
proc.shutdown()
if result != 1:
errorText = "Result output isn't 1!: '{}'".format(result)
raise ValueError(errorText)
return result
def test_processrunner_correct_commandnotfound_raised(self):
with self.assertRaises(CommandNotFound):
ProcessRunner([self.sampleCommandPath + "badpath"])