Commit | Line | Data |
---|---|---|
920dae64 AT |
1 | #! /usr/bin/env python |
2 | """Test script for popen2.py | |
3 | Christian Tismer | |
4 | """ | |
5 | ||
6 | import os | |
7 | import sys | |
8 | from test.test_support import TestSkipped | |
9 | ||
10 | # popen2 contains its own testing routine | |
11 | # which is especially useful to see if open files | |
12 | # like stdin can be read successfully by a forked | |
13 | # subprocess. | |
14 | ||
15 | def main(): | |
16 | print "Test popen2 module:" | |
17 | if (sys.platform[:4] == 'beos' or sys.platform[:6] == 'atheos') \ | |
18 | and __name__ != '__main__': | |
19 | # Locks get messed up or something. Generally we're supposed | |
20 | # to avoid mixing "posix" fork & exec with native threads, and | |
21 | # they may be right about that after all. | |
22 | raise TestSkipped, "popen2() doesn't work during import on " + sys.platform | |
23 | try: | |
24 | from os import popen | |
25 | except ImportError: | |
26 | # if we don't have os.popen, check that | |
27 | # we have os.fork. if not, skip the test | |
28 | # (by raising an ImportError) | |
29 | from os import fork | |
30 | import popen2 | |
31 | popen2._test() | |
32 | ||
33 | ||
34 | def _test(): | |
35 | # same test as popen2._test(), but using the os.popen*() API | |
36 | print "Testing os module:" | |
37 | import popen2 | |
38 | cmd = "cat" | |
39 | teststr = "ab cd\n" | |
40 | if os.name == "nt": | |
41 | cmd = "more" | |
42 | # "more" doesn't act the same way across Windows flavors, | |
43 | # sometimes adding an extra newline at the start or the | |
44 | # end. So we strip whitespace off both ends for comparison. | |
45 | expected = teststr.strip() | |
46 | print "testing popen2..." | |
47 | w, r = os.popen2(cmd) | |
48 | w.write(teststr) | |
49 | w.close() | |
50 | got = r.read() | |
51 | if got.strip() != expected: | |
52 | raise ValueError("wrote %r read %r" % (teststr, got)) | |
53 | print "testing popen3..." | |
54 | try: | |
55 | w, r, e = os.popen3([cmd]) | |
56 | except: | |
57 | w, r, e = os.popen3(cmd) | |
58 | w.write(teststr) | |
59 | w.close() | |
60 | got = r.read() | |
61 | if got.strip() != expected: | |
62 | raise ValueError("wrote %r read %r" % (teststr, got)) | |
63 | got = e.read() | |
64 | if got: | |
65 | raise ValueError("unexpected %r on stderr" % (got,)) | |
66 | for inst in popen2._active[:]: | |
67 | inst.wait() | |
68 | if popen2._active: | |
69 | raise ValueError("_active not empty") | |
70 | print "All OK" | |
71 | ||
72 | main() | |
73 | _test() |