79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
|
#! /usr/bin/env python
|
||
|
|
"""Test script for popen2.py
|
||
|
|
Christian Tismer
|
||
|
|
"""
|
||
|
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
from test.test_support import TestSkipped, is_jython, reap_children
|
||
|
|
|
||
|
|
# popen2 contains its own testing routine
|
||
|
|
# which is especially useful to see if open files
|
||
|
|
# like stdin can be read successfully by a forked
|
||
|
|
# subprocess.
|
||
|
|
|
||
|
|
def main():
|
||
|
|
print "Test popen2 module:"
|
||
|
|
if (sys.platform[:4] == 'beos' or sys.platform[:6] == 'atheos') \
|
||
|
|
and __name__ != '__main__':
|
||
|
|
# Locks get messed up or something. Generally we're supposed
|
||
|
|
# to avoid mixing "posix" fork & exec with native threads, and
|
||
|
|
# they may be right about that after all.
|
||
|
|
raise TestSkipped, "popen2() doesn't work during import on " + sys.platform
|
||
|
|
try:
|
||
|
|
from os import popen
|
||
|
|
except ImportError:
|
||
|
|
# if we don't have os.popen, check that
|
||
|
|
# we have os.fork. if not, skip the test
|
||
|
|
# (by raising an ImportError)
|
||
|
|
from os import fork
|
||
|
|
import popen2
|
||
|
|
popen2._test()
|
||
|
|
|
||
|
|
|
||
|
|
def _test():
|
||
|
|
# same test as popen2._test(), but using the os.popen*() API
|
||
|
|
print "Testing os module:"
|
||
|
|
import popen2
|
||
|
|
# When the test runs, there shouldn't be any open pipes
|
||
|
|
popen2._cleanup()
|
||
|
|
assert not popen2._active, "Active pipes when test starts " + repr([c.cmd for c in popen2._active])
|
||
|
|
cmd = "cat"
|
||
|
|
teststr = "ab cd\n"
|
||
|
|
if os.name == "nt" or (is_jython and os._name == 'nt'):
|
||
|
|
cmd = "more"
|
||
|
|
# "more" doesn't act the same way across Windows flavors,
|
||
|
|
# sometimes adding an extra newline at the start or the
|
||
|
|
# end. So we strip whitespace off both ends for comparison.
|
||
|
|
expected = teststr.strip()
|
||
|
|
print "testing popen2..."
|
||
|
|
w, r = os.popen2(cmd)
|
||
|
|
w.write(teststr)
|
||
|
|
w.close()
|
||
|
|
got = r.read()
|
||
|
|
if got.strip() != expected:
|
||
|
|
raise ValueError("wrote %r read %r" % (teststr, got))
|
||
|
|
print "testing popen3..."
|
||
|
|
try:
|
||
|
|
w, r, e = os.popen3([cmd])
|
||
|
|
except:
|
||
|
|
w, r, e = os.popen3(cmd)
|
||
|
|
w.write(teststr)
|
||
|
|
w.close()
|
||
|
|
got = r.read()
|
||
|
|
if got.strip() != expected:
|
||
|
|
raise ValueError("wrote %r read %r" % (teststr, got))
|
||
|
|
got = e.read()
|
||
|
|
if got:
|
||
|
|
raise ValueError("unexpected %r on stderr" % (got,))
|
||
|
|
for inst in popen2._active[:]:
|
||
|
|
inst.wait()
|
||
|
|
popen2._cleanup()
|
||
|
|
if popen2._active:
|
||
|
|
raise ValueError("_active not empty")
|
||
|
|
print "All OK"
|
||
|
|
|
||
|
|
main()
|
||
|
|
_test()
|
||
|
|
reap_children()
|