Lib/dos-8x3/test_pop.py

   1 #! /usr/bin/env python
   2 """Test script for popen2.py
   3    Christian Tismer
   4 """
   5
   6 import os
   7
   8 # popen2 contains its own testing routine
   9 # which is especially useful to see if open files
  10 # like stdin can be read successfully by a forked
  11 # subprocess.
  12
  13 def main():
  14     print "Test popen2 module:"
  15     try:
  16         from os import popen
  17     except ImportError:
  18         # if we don't have os.popen, check that
  19         # we have os.fork.  if not, skip the test
  20         # (by raising an ImportError)
  21         from os import fork
  22     import popen2
  23     popen2._test()
  24
  25
  26 def _test():
  27     # same test as popen2._test(), but using the os.popen*() API
  28     print "Testing os module:"
  29     import popen2
  30     cmd  = "cat"
  31     teststr = "ab cd\n"
  32     if os.name == "nt":
  33         cmd = "more"
  34     # "more" doesn't act the same way across Windows flavors,
  35     # sometimes adding an extra newline at the start or the
  36     # end.  So we strip whitespace off both ends for comparison.
  37     expected = teststr.strip()
  38     print "testing popen2..."
  39     w, r = os.popen2(cmd)
  40     w.write(teststr)
  41     w.close()
  42     got = r.read()
  43     if got.strip() != expected:
  44         raise ValueError("wrote %s read %s" % (`teststr`, `got`))
  45     print "testing popen3..."
  46     try:
  47         w, r, e = os.popen3([cmd])
  48     except:
  49         w, r, e = os.popen3(cmd)
  50     w.write(teststr)
  51     w.close()
  52     got = r.read()
  53     if got.strip() != expected:
  54         raise ValueError("wrote %s read %s" % (`teststr`, `got`))
  55     got = e.read()
  56     if got:
  57         raise ValueError("unexected %s on stderr" % `got`)
  58     for inst in popen2._active[:]:
  59         inst.wait()
  60     if popen2._active:
  61         raise ValueError("_active not empty")
  62     print "All OK"
  63
  64 main()
  65 _test()