| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1 | import unittest | 
| Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 2 | from test import support | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 3 | import subprocess | 
 | 4 | import sys | 
 | 5 | import signal | 
| Gregory P. Smith | 112bb3a | 2011-03-15 14:55:17 -0400 | [diff] [blame] | 6 | import io | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 7 | import os | 
| Gregory P. Smith | a59c59f | 2010-03-01 00:17:40 +0000 | [diff] [blame] | 8 | import errno | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 9 | import tempfile | 
 | 10 | import time | 
| Tim Peters | 3761e8d | 2004-10-13 04:07:12 +0000 | [diff] [blame] | 11 | import re | 
| Ezio Melotti | 184bdfb | 2010-02-18 09:37:05 +0000 | [diff] [blame] | 12 | import sysconfig | 
| Gregory P. Smith | d23047b | 2010-12-04 09:10:44 +0000 | [diff] [blame] | 13 | import warnings | 
| Gregory P. Smith | 51ee270 | 2010-12-13 07:59:39 +0000 | [diff] [blame] | 14 | import select | 
| Gregory P. Smith | 81ce685 | 2011-03-15 02:04:11 -0400 | [diff] [blame] | 15 | import shutil | 
| Benjamin Peterson | b870aa1 | 2011-12-10 12:44:25 -0500 | [diff] [blame] | 16 | import gc | 
| Benjamin Peterson | 964561b | 2011-12-10 12:31:42 -0500 | [diff] [blame] | 17 |  | 
 | 18 | try: | 
 | 19 |     import resource | 
 | 20 | except ImportError: | 
 | 21 |     resource = None | 
 | 22 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 23 | mswindows = (sys.platform == "win32") | 
 | 24 |  | 
 | 25 | # | 
 | 26 | # Depends on the following external programs: Python | 
 | 27 | # | 
 | 28 |  | 
 | 29 | if mswindows: | 
| Tim Peters | 3b01a70 | 2004-10-12 22:19:32 +0000 | [diff] [blame] | 30 |     SETBINARY = ('import msvcrt; msvcrt.setmode(sys.stdout.fileno(), ' | 
 | 31 |                                                 'os.O_BINARY);') | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 32 | else: | 
 | 33 |     SETBINARY = '' | 
 | 34 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 35 |  | 
 | 36 | try: | 
 | 37 |     mkstemp = tempfile.mkstemp | 
 | 38 | except AttributeError: | 
 | 39 |     # tempfile.mkstemp is not available | 
 | 40 |     def mkstemp(): | 
 | 41 |         """Replacement for mkstemp, calling mktemp.""" | 
 | 42 |         fname = tempfile.mktemp() | 
 | 43 |         return os.open(fname, os.O_RDWR|os.O_CREAT), fname | 
 | 44 |  | 
| Tim Peters | 3761e8d | 2004-10-13 04:07:12 +0000 | [diff] [blame] | 45 |  | 
| Florent Xicluna | c049d87 | 2010-03-27 22:47:23 +0000 | [diff] [blame] | 46 | class BaseTestCase(unittest.TestCase): | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 47 |     def setUp(self): | 
 | 48 |         # Try to minimize the number of children we have so this test | 
 | 49 |         # doesn't crash on some buildbots (Alphas in particular). | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 50 |         support.reap_children() | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 51 |  | 
| Florent Xicluna | f0cbd82 | 2010-03-04 21:50:56 +0000 | [diff] [blame] | 52 |     def tearDown(self): | 
 | 53 |         for inst in subprocess._active: | 
 | 54 |             inst.wait() | 
 | 55 |         subprocess._cleanup() | 
 | 56 |         self.assertFalse(subprocess._active, "subprocess._active not empty") | 
 | 57 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 58 |     def assertStderrEqual(self, stderr, expected, msg=None): | 
 | 59 |         # In a debug build, stuff like "[6580 refs]" is printed to stderr at | 
 | 60 |         # shutdown time.  That frustrates tests trying to check stderr produced | 
 | 61 |         # from a spawned Python process. | 
| Antoine Pitrou | 62f68ed | 2010-08-04 11:48:56 +0000 | [diff] [blame] | 62 |         actual = support.strip_python_stderr(stderr) | 
| Reid Kleckner | 31aa7dd | 2011-03-14 12:02:10 -0400 | [diff] [blame] | 63 |         # strip_python_stderr also strips whitespace, so we do too. | 
 | 64 |         expected = expected.strip() | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 65 |         self.assertEqual(actual, expected, msg) | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 66 |  | 
| Florent Xicluna | c049d87 | 2010-03-27 22:47:23 +0000 | [diff] [blame] | 67 |  | 
 | 68 | class ProcessTestCase(BaseTestCase): | 
 | 69 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 70 |     def test_call_seq(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 71 |         # call() function with sequence argument | 
| Tim Peters | 3b01a70 | 2004-10-12 22:19:32 +0000 | [diff] [blame] | 72 |         rc = subprocess.call([sys.executable, "-c", | 
 | 73 |                               "import sys; sys.exit(47)"]) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 74 |         self.assertEqual(rc, 47) | 
 | 75 |  | 
| Reid Kleckner | 31aa7dd | 2011-03-14 12:02:10 -0400 | [diff] [blame] | 76 |     def test_call_timeout(self): | 
 | 77 |         # call() function with timeout argument; we want to test that the child | 
 | 78 |         # process gets killed when the timeout expires.  If the child isn't | 
 | 79 |         # killed, this call will deadlock since subprocess.call waits for the | 
 | 80 |         # child. | 
 | 81 |         self.assertRaises(subprocess.TimeoutExpired, subprocess.call, | 
 | 82 |                           [sys.executable, "-c", "while True: pass"], | 
 | 83 |                           timeout=0.1) | 
 | 84 |  | 
| Peter Astrand | 454f767 | 2005-01-01 09:36:35 +0000 | [diff] [blame] | 85 |     def test_check_call_zero(self): | 
 | 86 |         # check_call() function with zero return code | 
 | 87 |         rc = subprocess.check_call([sys.executable, "-c", | 
 | 88 |                                     "import sys; sys.exit(0)"]) | 
 | 89 |         self.assertEqual(rc, 0) | 
 | 90 |  | 
 | 91 |     def test_check_call_nonzero(self): | 
 | 92 |         # check_call() function with non-zero return code | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 93 |         with self.assertRaises(subprocess.CalledProcessError) as c: | 
| Peter Astrand | 454f767 | 2005-01-01 09:36:35 +0000 | [diff] [blame] | 94 |             subprocess.check_call([sys.executable, "-c", | 
 | 95 |                                    "import sys; sys.exit(47)"]) | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 96 |         self.assertEqual(c.exception.returncode, 47) | 
| Peter Astrand | 454f767 | 2005-01-01 09:36:35 +0000 | [diff] [blame] | 97 |  | 
| Georg Brandl | f973407 | 2008-12-07 15:30:06 +0000 | [diff] [blame] | 98 |     def test_check_output(self): | 
 | 99 |         # check_output() function with zero return code | 
 | 100 |         output = subprocess.check_output( | 
 | 101 |                 [sys.executable, "-c", "print('BDFL')"]) | 
| Benjamin Peterson | 577473f | 2010-01-19 00:09:57 +0000 | [diff] [blame] | 102 |         self.assertIn(b'BDFL', output) | 
| Georg Brandl | f973407 | 2008-12-07 15:30:06 +0000 | [diff] [blame] | 103 |  | 
 | 104 |     def test_check_output_nonzero(self): | 
 | 105 |         # check_call() function with non-zero return code | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 106 |         with self.assertRaises(subprocess.CalledProcessError) as c: | 
| Georg Brandl | f973407 | 2008-12-07 15:30:06 +0000 | [diff] [blame] | 107 |             subprocess.check_output( | 
 | 108 |                     [sys.executable, "-c", "import sys; sys.exit(5)"]) | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 109 |         self.assertEqual(c.exception.returncode, 5) | 
| Georg Brandl | f973407 | 2008-12-07 15:30:06 +0000 | [diff] [blame] | 110 |  | 
 | 111 |     def test_check_output_stderr(self): | 
 | 112 |         # check_output() function stderr redirected to stdout | 
 | 113 |         output = subprocess.check_output( | 
 | 114 |                 [sys.executable, "-c", "import sys; sys.stderr.write('BDFL')"], | 
 | 115 |                 stderr=subprocess.STDOUT) | 
| Benjamin Peterson | 577473f | 2010-01-19 00:09:57 +0000 | [diff] [blame] | 116 |         self.assertIn(b'BDFL', output) | 
| Georg Brandl | f973407 | 2008-12-07 15:30:06 +0000 | [diff] [blame] | 117 |  | 
 | 118 |     def test_check_output_stdout_arg(self): | 
 | 119 |         # check_output() function stderr redirected to stdout | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 120 |         with self.assertRaises(ValueError) as c: | 
| Georg Brandl | f973407 | 2008-12-07 15:30:06 +0000 | [diff] [blame] | 121 |             output = subprocess.check_output( | 
 | 122 |                     [sys.executable, "-c", "print('will not be run')"], | 
 | 123 |                     stdout=sys.stdout) | 
| Georg Brandl | f973407 | 2008-12-07 15:30:06 +0000 | [diff] [blame] | 124 |             self.fail("Expected ValueError when stdout arg supplied.") | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 125 |         self.assertIn('stdout', c.exception.args[0]) | 
| Georg Brandl | f973407 | 2008-12-07 15:30:06 +0000 | [diff] [blame] | 126 |  | 
| Reid Kleckner | 31aa7dd | 2011-03-14 12:02:10 -0400 | [diff] [blame] | 127 |     def test_check_output_timeout(self): | 
 | 128 |         # check_output() function with timeout arg | 
 | 129 |         with self.assertRaises(subprocess.TimeoutExpired) as c: | 
 | 130 |             output = subprocess.check_output( | 
 | 131 |                     [sys.executable, "-c", | 
| Victor Stinner | 149b1c7 | 2011-06-06 23:43:02 +0200 | [diff] [blame] | 132 |                      "import sys, time\n" | 
 | 133 |                      "sys.stdout.write('BDFL')\n" | 
| Reid Kleckner | 31aa7dd | 2011-03-14 12:02:10 -0400 | [diff] [blame] | 134 |                      "sys.stdout.flush()\n" | 
| Victor Stinner | 149b1c7 | 2011-06-06 23:43:02 +0200 | [diff] [blame] | 135 |                      "time.sleep(3600)"], | 
| Reid Kleckner | da9ac72 | 2011-03-16 17:08:21 -0400 | [diff] [blame] | 136 |                     # Some heavily loaded buildbots (sparc Debian 3.x) require | 
 | 137 |                     # this much time to start and print. | 
 | 138 |                     timeout=3) | 
| Reid Kleckner | 31aa7dd | 2011-03-14 12:02:10 -0400 | [diff] [blame] | 139 |             self.fail("Expected TimeoutExpired.") | 
 | 140 |         self.assertEqual(c.exception.output, b'BDFL') | 
 | 141 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 142 |     def test_call_kwargs(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 143 |         # call() function with keyword args | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 144 |         newenv = os.environ.copy() | 
 | 145 |         newenv["FRUIT"] = "banana" | 
 | 146 |         rc = subprocess.call([sys.executable, "-c", | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 147 |                               'import sys, os;' | 
 | 148 |                               'sys.exit(os.getenv("FRUIT")=="banana")'], | 
 | 149 |                              env=newenv) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 150 |         self.assertEqual(rc, 1) | 
 | 151 |  | 
| Victor Stinner | 87b9bc3 | 2011-06-01 00:57:47 +0200 | [diff] [blame] | 152 |     def test_invalid_args(self): | 
 | 153 |         # Popen() called with invalid arguments should raise TypeError | 
 | 154 |         # but Popen.__del__ should not complain (issue #12085) | 
 | 155 |         with support.captured_stderr() as s: | 
 | 156 |             self.assertRaises(TypeError, subprocess.Popen, invalid_arg_name=1) | 
 | 157 |             argcount = subprocess.Popen.__init__.__code__.co_argcount | 
 | 158 |             too_many_args = [0] * (argcount + 1) | 
 | 159 |             self.assertRaises(TypeError, subprocess.Popen, *too_many_args) | 
 | 160 |         self.assertEqual(s.getvalue(), '') | 
 | 161 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 162 |     def test_stdin_none(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 163 |         # .stdin is None when not redirected | 
| Georg Brandl | 88fc664 | 2007-02-09 21:28:07 +0000 | [diff] [blame] | 164 |         p = subprocess.Popen([sys.executable, "-c", 'print("banana")'], | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 165 |                          stdout=subprocess.PIPE, stderr=subprocess.PIPE) | 
| Brian Curtin | 3c6a951 | 2010-11-05 03:58:52 +0000 | [diff] [blame] | 166 |         self.addCleanup(p.stdout.close) | 
 | 167 |         self.addCleanup(p.stderr.close) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 168 |         p.wait() | 
 | 169 |         self.assertEqual(p.stdin, None) | 
 | 170 |  | 
 | 171 |     def test_stdout_none(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 172 |         # .stdout is None when not redirected | 
| Tim Peters | 29b6b4f | 2004-10-13 03:43:40 +0000 | [diff] [blame] | 173 |         p = subprocess.Popen([sys.executable, "-c", | 
| Georg Brandl | 88fc664 | 2007-02-09 21:28:07 +0000 | [diff] [blame] | 174 |                              'print("    this bit of output is from a ' | 
| Tim Peters | 4052fe5 | 2004-10-13 03:29:54 +0000 | [diff] [blame] | 175 |                              'test of stdout in a different ' | 
| Georg Brandl | 88fc664 | 2007-02-09 21:28:07 +0000 | [diff] [blame] | 176 |                              'process ...")'], | 
| Tim Peters | 4052fe5 | 2004-10-13 03:29:54 +0000 | [diff] [blame] | 177 |                              stdin=subprocess.PIPE, stderr=subprocess.PIPE) | 
| Brian Curtin | 3c6a951 | 2010-11-05 03:58:52 +0000 | [diff] [blame] | 178 |         self.addCleanup(p.stdin.close) | 
 | 179 |         self.addCleanup(p.stderr.close) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 180 |         p.wait() | 
 | 181 |         self.assertEqual(p.stdout, None) | 
 | 182 |  | 
 | 183 |     def test_stderr_none(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 184 |         # .stderr is None when not redirected | 
| Georg Brandl | 88fc664 | 2007-02-09 21:28:07 +0000 | [diff] [blame] | 185 |         p = subprocess.Popen([sys.executable, "-c", 'print("banana")'], | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 186 |                          stdin=subprocess.PIPE, stdout=subprocess.PIPE) | 
| Brian Curtin | 3c6a951 | 2010-11-05 03:58:52 +0000 | [diff] [blame] | 187 |         self.addCleanup(p.stdout.close) | 
 | 188 |         self.addCleanup(p.stdin.close) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 189 |         p.wait() | 
 | 190 |         self.assertEqual(p.stderr, None) | 
 | 191 |  | 
| Ezio Melotti | 184bdfb | 2010-02-18 09:37:05 +0000 | [diff] [blame] | 192 |     def test_executable_with_cwd(self): | 
| Florent Xicluna | 1d1ab97 | 2010-03-11 01:53:10 +0000 | [diff] [blame] | 193 |         python_dir = os.path.dirname(os.path.realpath(sys.executable)) | 
| Ezio Melotti | 184bdfb | 2010-02-18 09:37:05 +0000 | [diff] [blame] | 194 |         p = subprocess.Popen(["somethingyoudonthave", "-c", | 
 | 195 |                               "import sys; sys.exit(47)"], | 
 | 196 |                              executable=sys.executable, cwd=python_dir) | 
 | 197 |         p.wait() | 
 | 198 |         self.assertEqual(p.returncode, 47) | 
 | 199 |  | 
 | 200 |     @unittest.skipIf(sysconfig.is_python_build(), | 
 | 201 |                      "need an installed Python. See #7774") | 
 | 202 |     def test_executable_without_cwd(self): | 
 | 203 |         # For a normal installation, it should work without 'cwd' | 
 | 204 |         # argument.  For test runs in the build directory, see #7774. | 
 | 205 |         p = subprocess.Popen(["somethingyoudonthave", "-c", | 
 | 206 |                               "import sys; sys.exit(47)"], | 
| Tim Peters | 3b01a70 | 2004-10-12 22:19:32 +0000 | [diff] [blame] | 207 |                              executable=sys.executable) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 208 |         p.wait() | 
 | 209 |         self.assertEqual(p.returncode, 47) | 
 | 210 |  | 
 | 211 |     def test_stdin_pipe(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 212 |         # stdin redirection | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 213 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 214 |                          'import sys; sys.exit(sys.stdin.read() == "pear")'], | 
 | 215 |                         stdin=subprocess.PIPE) | 
| Guido van Rossum | bb839ef | 2007-08-27 23:58:21 +0000 | [diff] [blame] | 216 |         p.stdin.write(b"pear") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 217 |         p.stdin.close() | 
 | 218 |         p.wait() | 
 | 219 |         self.assertEqual(p.returncode, 1) | 
 | 220 |  | 
 | 221 |     def test_stdin_filedes(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 222 |         # stdin is set to open file descriptor | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 223 |         tf = tempfile.TemporaryFile() | 
| Benjamin Peterson | cc221b2 | 2010-10-31 02:06:21 +0000 | [diff] [blame] | 224 |         self.addCleanup(tf.close) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 225 |         d = tf.fileno() | 
| Antoine Pitrou | 9cadb1b | 2008-09-15 23:02:56 +0000 | [diff] [blame] | 226 |         os.write(d, b"pear") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 227 |         os.lseek(d, 0, 0) | 
 | 228 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 229 |                          'import sys; sys.exit(sys.stdin.read() == "pear")'], | 
 | 230 |                          stdin=d) | 
 | 231 |         p.wait() | 
 | 232 |         self.assertEqual(p.returncode, 1) | 
 | 233 |  | 
 | 234 |     def test_stdin_fileobj(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 235 |         # stdin is set to open file object | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 236 |         tf = tempfile.TemporaryFile() | 
| Benjamin Peterson | cc221b2 | 2010-10-31 02:06:21 +0000 | [diff] [blame] | 237 |         self.addCleanup(tf.close) | 
| Guido van Rossum | bb839ef | 2007-08-27 23:58:21 +0000 | [diff] [blame] | 238 |         tf.write(b"pear") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 239 |         tf.seek(0) | 
 | 240 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 241 |                          'import sys; sys.exit(sys.stdin.read() == "pear")'], | 
 | 242 |                          stdin=tf) | 
 | 243 |         p.wait() | 
 | 244 |         self.assertEqual(p.returncode, 1) | 
 | 245 |  | 
 | 246 |     def test_stdout_pipe(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 247 |         # stdout redirection | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 248 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 249 |                           'import sys; sys.stdout.write("orange")'], | 
 | 250 |                          stdout=subprocess.PIPE) | 
| Brian Curtin | 3c6a951 | 2010-11-05 03:58:52 +0000 | [diff] [blame] | 251 |         self.addCleanup(p.stdout.close) | 
| Guido van Rossum | fa0054a | 2007-05-24 04:05:35 +0000 | [diff] [blame] | 252 |         self.assertEqual(p.stdout.read(), b"orange") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 253 |  | 
 | 254 |     def test_stdout_filedes(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 255 |         # stdout is set to open file descriptor | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 256 |         tf = tempfile.TemporaryFile() | 
| Benjamin Peterson | cc221b2 | 2010-10-31 02:06:21 +0000 | [diff] [blame] | 257 |         self.addCleanup(tf.close) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 258 |         d = tf.fileno() | 
 | 259 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 260 |                           'import sys; sys.stdout.write("orange")'], | 
 | 261 |                          stdout=d) | 
 | 262 |         p.wait() | 
 | 263 |         os.lseek(d, 0, 0) | 
| Guido van Rossum | c9e363c | 2007-05-15 23:18:55 +0000 | [diff] [blame] | 264 |         self.assertEqual(os.read(d, 1024), b"orange") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 265 |  | 
 | 266 |     def test_stdout_fileobj(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 267 |         # stdout is set to open file object | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 268 |         tf = tempfile.TemporaryFile() | 
| Benjamin Peterson | cc221b2 | 2010-10-31 02:06:21 +0000 | [diff] [blame] | 269 |         self.addCleanup(tf.close) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 270 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 271 |                           'import sys; sys.stdout.write("orange")'], | 
 | 272 |                          stdout=tf) | 
 | 273 |         p.wait() | 
 | 274 |         tf.seek(0) | 
| Guido van Rossum | fa0054a | 2007-05-24 04:05:35 +0000 | [diff] [blame] | 275 |         self.assertEqual(tf.read(), b"orange") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 276 |  | 
 | 277 |     def test_stderr_pipe(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 278 |         # stderr redirection | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 279 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 280 |                           'import sys; sys.stderr.write("strawberry")'], | 
 | 281 |                          stderr=subprocess.PIPE) | 
| Brian Curtin | 3c6a951 | 2010-11-05 03:58:52 +0000 | [diff] [blame] | 282 |         self.addCleanup(p.stderr.close) | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 283 |         self.assertStderrEqual(p.stderr.read(), b"strawberry") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 284 |  | 
 | 285 |     def test_stderr_filedes(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 286 |         # stderr is set to open file descriptor | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 287 |         tf = tempfile.TemporaryFile() | 
| Benjamin Peterson | cc221b2 | 2010-10-31 02:06:21 +0000 | [diff] [blame] | 288 |         self.addCleanup(tf.close) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 289 |         d = tf.fileno() | 
 | 290 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 291 |                           'import sys; sys.stderr.write("strawberry")'], | 
 | 292 |                          stderr=d) | 
 | 293 |         p.wait() | 
 | 294 |         os.lseek(d, 0, 0) | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 295 |         self.assertStderrEqual(os.read(d, 1024), b"strawberry") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 296 |  | 
 | 297 |     def test_stderr_fileobj(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 298 |         # stderr is set to open file object | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 299 |         tf = tempfile.TemporaryFile() | 
| Benjamin Peterson | cc221b2 | 2010-10-31 02:06:21 +0000 | [diff] [blame] | 300 |         self.addCleanup(tf.close) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 301 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 302 |                           'import sys; sys.stderr.write("strawberry")'], | 
 | 303 |                          stderr=tf) | 
 | 304 |         p.wait() | 
 | 305 |         tf.seek(0) | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 306 |         self.assertStderrEqual(tf.read(), b"strawberry") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 307 |  | 
 | 308 |     def test_stdout_stderr_pipe(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 309 |         # capture stdout and stderr to the same pipe | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 310 |         p = subprocess.Popen([sys.executable, "-c", | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 311 |                               'import sys;' | 
 | 312 |                               'sys.stdout.write("apple");' | 
 | 313 |                               'sys.stdout.flush();' | 
 | 314 |                               'sys.stderr.write("orange")'], | 
 | 315 |                              stdout=subprocess.PIPE, | 
 | 316 |                              stderr=subprocess.STDOUT) | 
| Brian Curtin | 3c6a951 | 2010-11-05 03:58:52 +0000 | [diff] [blame] | 317 |         self.addCleanup(p.stdout.close) | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 318 |         self.assertStderrEqual(p.stdout.read(), b"appleorange") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 319 |  | 
 | 320 |     def test_stdout_stderr_file(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 321 |         # capture stdout and stderr to the same open file | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 322 |         tf = tempfile.TemporaryFile() | 
| Benjamin Peterson | cc221b2 | 2010-10-31 02:06:21 +0000 | [diff] [blame] | 323 |         self.addCleanup(tf.close) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 324 |         p = subprocess.Popen([sys.executable, "-c", | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 325 |                               'import sys;' | 
 | 326 |                               'sys.stdout.write("apple");' | 
 | 327 |                               'sys.stdout.flush();' | 
 | 328 |                               'sys.stderr.write("orange")'], | 
 | 329 |                              stdout=tf, | 
 | 330 |                              stderr=tf) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 331 |         p.wait() | 
 | 332 |         tf.seek(0) | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 333 |         self.assertStderrEqual(tf.read(), b"appleorange") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 334 |  | 
| Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 335 |     def test_stdout_filedes_of_stdout(self): | 
 | 336 |         # stdout is set to 1 (#1531862). | 
| Antoine Pitrou | 9cadb1b | 2008-09-15 23:02:56 +0000 | [diff] [blame] | 337 |         cmd = r"import sys, os; sys.exit(os.write(sys.stdout.fileno(), b'.\n'))" | 
| Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 338 |         rc = subprocess.call([sys.executable, "-c", cmd], stdout=1) | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 339 |         self.assertEqual(rc, 2) | 
| Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 340 |  | 
| Ross Lagerwall | ba102ec | 2011-03-16 18:40:25 +0200 | [diff] [blame] | 341 |     def test_stdout_devnull(self): | 
 | 342 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 343 |                               'for i in range(10240):' | 
 | 344 |                               'print("x" * 1024)'], | 
 | 345 |                               stdout=subprocess.DEVNULL) | 
 | 346 |         p.wait() | 
 | 347 |         self.assertEqual(p.stdout, None) | 
 | 348 |  | 
 | 349 |     def test_stderr_devnull(self): | 
 | 350 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 351 |                               'import sys\n' | 
 | 352 |                               'for i in range(10240):' | 
 | 353 |                               'sys.stderr.write("x" * 1024)'], | 
 | 354 |                               stderr=subprocess.DEVNULL) | 
 | 355 |         p.wait() | 
 | 356 |         self.assertEqual(p.stderr, None) | 
 | 357 |  | 
 | 358 |     def test_stdin_devnull(self): | 
 | 359 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 360 |                               'import sys;' | 
 | 361 |                               'sys.stdin.read(1)'], | 
 | 362 |                               stdin=subprocess.DEVNULL) | 
 | 363 |         p.wait() | 
 | 364 |         self.assertEqual(p.stdin, None) | 
 | 365 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 366 |     def test_cwd(self): | 
| Christian Heimes | 5fb7c2a | 2007-12-24 08:52:31 +0000 | [diff] [blame] | 367 |         tmpdir = tempfile.gettempdir() | 
| Peter Astrand | 195404f | 2004-11-12 15:51:48 +0000 | [diff] [blame] | 368 |         # We cannot use os.path.realpath to canonicalize the path, | 
 | 369 |         # since it doesn't expand Tru64 {memb} strings. See bug 1063571. | 
 | 370 |         cwd = os.getcwd() | 
 | 371 |         os.chdir(tmpdir) | 
 | 372 |         tmpdir = os.getcwd() | 
 | 373 |         os.chdir(cwd) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 374 |         p = subprocess.Popen([sys.executable, "-c", | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 375 |                               'import sys,os;' | 
 | 376 |                               'sys.stdout.write(os.getcwd())'], | 
 | 377 |                              stdout=subprocess.PIPE, | 
 | 378 |                              cwd=tmpdir) | 
| Brian Curtin | 3c6a951 | 2010-11-05 03:58:52 +0000 | [diff] [blame] | 379 |         self.addCleanup(p.stdout.close) | 
| Fredrik Lundh | 59c0559 | 2004-10-13 06:55:40 +0000 | [diff] [blame] | 380 |         normcase = os.path.normcase | 
| Guido van Rossum | bb839ef | 2007-08-27 23:58:21 +0000 | [diff] [blame] | 381 |         self.assertEqual(normcase(p.stdout.read().decode("utf-8")), | 
 | 382 |                          normcase(tmpdir)) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 383 |  | 
 | 384 |     def test_env(self): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 385 |         newenv = os.environ.copy() | 
 | 386 |         newenv["FRUIT"] = "orange" | 
| Victor Stinner | f1512a2 | 2011-06-21 17:18:38 +0200 | [diff] [blame] | 387 |         with subprocess.Popen([sys.executable, "-c", | 
 | 388 |                                'import sys,os;' | 
 | 389 |                                'sys.stdout.write(os.getenv("FRUIT"))'], | 
 | 390 |                               stdout=subprocess.PIPE, | 
 | 391 |                               env=newenv) as p: | 
 | 392 |             stdout, stderr = p.communicate() | 
 | 393 |             self.assertEqual(stdout, b"orange") | 
 | 394 |  | 
| Victor Stinner | 62d5118 | 2011-06-23 01:02:25 +0200 | [diff] [blame] | 395 |     # Windows requires at least the SYSTEMROOT environment variable to start | 
 | 396 |     # Python | 
 | 397 |     @unittest.skipIf(sys.platform == 'win32', | 
 | 398 |                      'cannot test an empty env on Windows') | 
| Victor Stinner | 237e5cb | 2011-06-22 21:28:43 +0200 | [diff] [blame] | 399 |     @unittest.skipIf(sysconfig.get_config_var('Py_ENABLE_SHARED') is not None, | 
| Victor Stinner | 372309a | 2011-06-21 21:59:06 +0200 | [diff] [blame] | 400 |                      'the python library cannot be loaded ' | 
 | 401 |                      'with an empty environment') | 
| Victor Stinner | f1512a2 | 2011-06-21 17:18:38 +0200 | [diff] [blame] | 402 |     def test_empty_env(self): | 
 | 403 |         with subprocess.Popen([sys.executable, "-c", | 
 | 404 |                                'import os; ' | 
| Victor Stinner | 372309a | 2011-06-21 21:59:06 +0200 | [diff] [blame] | 405 |                                'print(list(os.environ.keys()))'], | 
| Victor Stinner | f1512a2 | 2011-06-21 17:18:38 +0200 | [diff] [blame] | 406 |                               stdout=subprocess.PIPE, | 
 | 407 |                               env={}) as p: | 
 | 408 |             stdout, stderr = p.communicate() | 
| Victor Stinner | 237e5cb | 2011-06-22 21:28:43 +0200 | [diff] [blame] | 409 |             self.assertIn(stdout.strip(), | 
 | 410 |                 (b"[]", | 
 | 411 |                  # Mac OS X adds __CF_USER_TEXT_ENCODING variable to an empty | 
 | 412 |                  # environment | 
 | 413 |                  b"['__CF_USER_TEXT_ENCODING']")) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 414 |  | 
| Peter Astrand | cbac93c | 2005-03-03 20:24:28 +0000 | [diff] [blame] | 415 |     def test_communicate_stdin(self): | 
 | 416 |         p = subprocess.Popen([sys.executable, "-c", | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 417 |                               'import sys;' | 
 | 418 |                               'sys.exit(sys.stdin.read() == "pear")'], | 
| Peter Astrand | cbac93c | 2005-03-03 20:24:28 +0000 | [diff] [blame] | 419 |                              stdin=subprocess.PIPE) | 
| Guido van Rossum | bb839ef | 2007-08-27 23:58:21 +0000 | [diff] [blame] | 420 |         p.communicate(b"pear") | 
| Peter Astrand | cbac93c | 2005-03-03 20:24:28 +0000 | [diff] [blame] | 421 |         self.assertEqual(p.returncode, 1) | 
 | 422 |  | 
 | 423 |     def test_communicate_stdout(self): | 
 | 424 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 425 |                               'import sys; sys.stdout.write("pineapple")'], | 
 | 426 |                              stdout=subprocess.PIPE) | 
 | 427 |         (stdout, stderr) = p.communicate() | 
| Guido van Rossum | fa0054a | 2007-05-24 04:05:35 +0000 | [diff] [blame] | 428 |         self.assertEqual(stdout, b"pineapple") | 
| Peter Astrand | cbac93c | 2005-03-03 20:24:28 +0000 | [diff] [blame] | 429 |         self.assertEqual(stderr, None) | 
 | 430 |  | 
 | 431 |     def test_communicate_stderr(self): | 
 | 432 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 433 |                               'import sys; sys.stderr.write("pineapple")'], | 
 | 434 |                              stderr=subprocess.PIPE) | 
 | 435 |         (stdout, stderr) = p.communicate() | 
 | 436 |         self.assertEqual(stdout, None) | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 437 |         self.assertStderrEqual(stderr, b"pineapple") | 
| Peter Astrand | cbac93c | 2005-03-03 20:24:28 +0000 | [diff] [blame] | 438 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 439 |     def test_communicate(self): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 440 |         p = subprocess.Popen([sys.executable, "-c", | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 441 |                               'import sys,os;' | 
 | 442 |                               'sys.stderr.write("pineapple");' | 
 | 443 |                               'sys.stdout.write(sys.stdin.read())'], | 
 | 444 |                              stdin=subprocess.PIPE, | 
 | 445 |                              stdout=subprocess.PIPE, | 
 | 446 |                              stderr=subprocess.PIPE) | 
| Brian Curtin | 19a5379 | 2010-11-05 17:09:05 +0000 | [diff] [blame] | 447 |         self.addCleanup(p.stdout.close) | 
 | 448 |         self.addCleanup(p.stderr.close) | 
 | 449 |         self.addCleanup(p.stdin.close) | 
| Georg Brandl | 1abcbf8 | 2008-07-01 19:28:43 +0000 | [diff] [blame] | 450 |         (stdout, stderr) = p.communicate(b"banana") | 
| Guido van Rossum | c9e363c | 2007-05-15 23:18:55 +0000 | [diff] [blame] | 451 |         self.assertEqual(stdout, b"banana") | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 452 |         self.assertStderrEqual(stderr, b"pineapple") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 453 |  | 
| Reid Kleckner | 31aa7dd | 2011-03-14 12:02:10 -0400 | [diff] [blame] | 454 |     def test_communicate_timeout(self): | 
 | 455 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 456 |                               'import sys,os,time;' | 
 | 457 |                               'sys.stderr.write("pineapple\\n");' | 
 | 458 |                               'time.sleep(1);' | 
 | 459 |                               'sys.stderr.write("pear\\n");' | 
 | 460 |                               'sys.stdout.write(sys.stdin.read())'], | 
 | 461 |                              universal_newlines=True, | 
 | 462 |                              stdin=subprocess.PIPE, | 
 | 463 |                              stdout=subprocess.PIPE, | 
 | 464 |                              stderr=subprocess.PIPE) | 
 | 465 |         self.assertRaises(subprocess.TimeoutExpired, p.communicate, "banana", | 
 | 466 |                           timeout=0.3) | 
 | 467 |         # Make sure we can keep waiting for it, and that we get the whole output | 
 | 468 |         # after it completes. | 
 | 469 |         (stdout, stderr) = p.communicate() | 
 | 470 |         self.assertEqual(stdout, "banana") | 
 | 471 |         self.assertStderrEqual(stderr.encode(), b"pineapple\npear\n") | 
 | 472 |  | 
 | 473 |     def test_communicate_timeout_large_ouput(self): | 
| Ross Lagerwall | 003c7a3 | 2012-02-12 09:02:01 +0200 | [diff] [blame] | 474 |         # Test an expiring timeout while the child is outputting lots of data. | 
| Reid Kleckner | 31aa7dd | 2011-03-14 12:02:10 -0400 | [diff] [blame] | 475 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 476 |                               'import sys,os,time;' | 
 | 477 |                               'sys.stdout.write("a" * (64 * 1024));' | 
 | 478 |                               'time.sleep(0.2);' | 
 | 479 |                               'sys.stdout.write("a" * (64 * 1024));' | 
 | 480 |                               'time.sleep(0.2);' | 
 | 481 |                               'sys.stdout.write("a" * (64 * 1024));' | 
 | 482 |                               'time.sleep(0.2);' | 
 | 483 |                               'sys.stdout.write("a" * (64 * 1024));'], | 
 | 484 |                              stdout=subprocess.PIPE) | 
 | 485 |         self.assertRaises(subprocess.TimeoutExpired, p.communicate, timeout=0.4) | 
 | 486 |         (stdout, _) = p.communicate() | 
 | 487 |         self.assertEqual(len(stdout), 4 * 64 * 1024) | 
 | 488 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 489 |     # Test for the fd leak reported in http://bugs.python.org/issue2791. | 
 | 490 |     def test_communicate_pipe_fd_leak(self): | 
| Victor Stinner | 667d4b5 | 2010-12-25 22:40:32 +0000 | [diff] [blame] | 491 |         for stdin_pipe in (False, True): | 
 | 492 |             for stdout_pipe in (False, True): | 
 | 493 |                 for stderr_pipe in (False, True): | 
 | 494 |                     options = {} | 
 | 495 |                     if stdin_pipe: | 
 | 496 |                         options['stdin'] = subprocess.PIPE | 
 | 497 |                     if stdout_pipe: | 
 | 498 |                         options['stdout'] = subprocess.PIPE | 
 | 499 |                     if stderr_pipe: | 
 | 500 |                         options['stderr'] = subprocess.PIPE | 
 | 501 |                     if not options: | 
 | 502 |                         continue | 
 | 503 |                     p = subprocess.Popen((sys.executable, "-c", "pass"), **options) | 
 | 504 |                     p.communicate() | 
 | 505 |                     if p.stdin is not None: | 
 | 506 |                         self.assertTrue(p.stdin.closed) | 
 | 507 |                     if p.stdout is not None: | 
 | 508 |                         self.assertTrue(p.stdout.closed) | 
 | 509 |                     if p.stderr is not None: | 
 | 510 |                         self.assertTrue(p.stderr.closed) | 
| Georg Brandl | f08a9dd | 2008-06-10 16:57:31 +0000 | [diff] [blame] | 511 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 512 |     def test_communicate_returns(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 513 |         # communicate() should return None if no redirection is active | 
| Tim Peters | 3b01a70 | 2004-10-12 22:19:32 +0000 | [diff] [blame] | 514 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 515 |                               "import sys; sys.exit(47)"]) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 516 |         (stdout, stderr) = p.communicate() | 
 | 517 |         self.assertEqual(stdout, None) | 
 | 518 |         self.assertEqual(stderr, None) | 
 | 519 |  | 
 | 520 |     def test_communicate_pipe_buf(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 521 |         # communicate() with writes larger than pipe_buf | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 522 |         # This test will probably deadlock rather than fail, if | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 523 |         # communicate() does not work properly. | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 524 |         x, y = os.pipe() | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 525 |         os.close(x) | 
 | 526 |         os.close(y) | 
 | 527 |         p = subprocess.Popen([sys.executable, "-c", | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 528 |                               'import sys,os;' | 
 | 529 |                               'sys.stdout.write(sys.stdin.read(47));' | 
| Charles-François Natali | 2d51721 | 2011-05-29 16:36:44 +0200 | [diff] [blame] | 530 |                               'sys.stderr.write("x" * %d);' | 
 | 531 |                               'sys.stdout.write(sys.stdin.read())' % | 
 | 532 |                               support.PIPE_MAX_SIZE], | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 533 |                              stdin=subprocess.PIPE, | 
 | 534 |                              stdout=subprocess.PIPE, | 
 | 535 |                              stderr=subprocess.PIPE) | 
| Brian Curtin | 19a5379 | 2010-11-05 17:09:05 +0000 | [diff] [blame] | 536 |         self.addCleanup(p.stdout.close) | 
 | 537 |         self.addCleanup(p.stderr.close) | 
 | 538 |         self.addCleanup(p.stdin.close) | 
| Charles-François Natali | 2d51721 | 2011-05-29 16:36:44 +0200 | [diff] [blame] | 539 |         string_to_write = b"a" * support.PIPE_MAX_SIZE | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 540 |         (stdout, stderr) = p.communicate(string_to_write) | 
 | 541 |         self.assertEqual(stdout, string_to_write) | 
 | 542 |  | 
 | 543 |     def test_writes_before_communicate(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 544 |         # stdin.write before communicate() | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 545 |         p = subprocess.Popen([sys.executable, "-c", | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 546 |                               'import sys,os;' | 
 | 547 |                               'sys.stdout.write(sys.stdin.read())'], | 
 | 548 |                              stdin=subprocess.PIPE, | 
 | 549 |                              stdout=subprocess.PIPE, | 
 | 550 |                              stderr=subprocess.PIPE) | 
| Brian Curtin | 19a5379 | 2010-11-05 17:09:05 +0000 | [diff] [blame] | 551 |         self.addCleanup(p.stdout.close) | 
 | 552 |         self.addCleanup(p.stderr.close) | 
 | 553 |         self.addCleanup(p.stdin.close) | 
| Guido van Rossum | bb839ef | 2007-08-27 23:58:21 +0000 | [diff] [blame] | 554 |         p.stdin.write(b"banana") | 
 | 555 |         (stdout, stderr) = p.communicate(b"split") | 
| Guido van Rossum | c9e363c | 2007-05-15 23:18:55 +0000 | [diff] [blame] | 556 |         self.assertEqual(stdout, b"bananasplit") | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 557 |         self.assertStderrEqual(stderr, b"") | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 558 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 559 |     def test_universal_newlines(self): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 560 |         p = subprocess.Popen([sys.executable, "-c", | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 561 |                               'import sys,os;' + SETBINARY + | 
| Antoine Pitrou | ab85ff3 | 2011-07-23 22:03:45 +0200 | [diff] [blame] | 562 |                               'sys.stdout.write(sys.stdin.readline());' | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 563 |                               'sys.stdout.flush();' | 
 | 564 |                               'sys.stdout.write("line2\\n");' | 
 | 565 |                               'sys.stdout.flush();' | 
| Antoine Pitrou | ab85ff3 | 2011-07-23 22:03:45 +0200 | [diff] [blame] | 566 |                               'sys.stdout.write(sys.stdin.read());' | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 567 |                               'sys.stdout.flush();' | 
| Antoine Pitrou | ab85ff3 | 2011-07-23 22:03:45 +0200 | [diff] [blame] | 568 |                               'sys.stdout.write("line4\\n");' | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 569 |                               'sys.stdout.flush();' | 
| Antoine Pitrou | ab85ff3 | 2011-07-23 22:03:45 +0200 | [diff] [blame] | 570 |                               'sys.stdout.write("line5\\r\\n");' | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 571 |                               'sys.stdout.flush();' | 
| Antoine Pitrou | ab85ff3 | 2011-07-23 22:03:45 +0200 | [diff] [blame] | 572 |                               'sys.stdout.write("line6\\r");' | 
 | 573 |                               'sys.stdout.flush();' | 
 | 574 |                               'sys.stdout.write("\\nline7");' | 
 | 575 |                               'sys.stdout.flush();' | 
 | 576 |                               'sys.stdout.write("\\nline8");'], | 
 | 577 |                              stdin=subprocess.PIPE, | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 578 |                              stdout=subprocess.PIPE, | 
 | 579 |                              universal_newlines=1) | 
| Antoine Pitrou | ab85ff3 | 2011-07-23 22:03:45 +0200 | [diff] [blame] | 580 |         p.stdin.write("line1\n") | 
 | 581 |         self.assertEqual(p.stdout.readline(), "line1\n") | 
 | 582 |         p.stdin.write("line3\n") | 
 | 583 |         p.stdin.close() | 
| Brian Curtin | 3c6a951 | 2010-11-05 03:58:52 +0000 | [diff] [blame] | 584 |         self.addCleanup(p.stdout.close) | 
| Antoine Pitrou | ab85ff3 | 2011-07-23 22:03:45 +0200 | [diff] [blame] | 585 |         self.assertEqual(p.stdout.readline(), | 
 | 586 |                          "line2\n") | 
 | 587 |         self.assertEqual(p.stdout.read(6), | 
 | 588 |                          "line3\n") | 
 | 589 |         self.assertEqual(p.stdout.read(), | 
 | 590 |                          "line4\nline5\nline6\nline7\nline8") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 591 |  | 
 | 592 |     def test_universal_newlines_communicate(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 593 |         # universal newlines through communicate() | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 594 |         p = subprocess.Popen([sys.executable, "-c", | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 595 |                               'import sys,os;' + SETBINARY + | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 596 |                               'sys.stdout.write("line2\\n");' | 
 | 597 |                               'sys.stdout.flush();' | 
| Antoine Pitrou | ab85ff3 | 2011-07-23 22:03:45 +0200 | [diff] [blame] | 598 |                               'sys.stdout.write("line4\\n");' | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 599 |                               'sys.stdout.flush();' | 
| Antoine Pitrou | ab85ff3 | 2011-07-23 22:03:45 +0200 | [diff] [blame] | 600 |                               'sys.stdout.write("line5\\r\\n");' | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 601 |                               'sys.stdout.flush();' | 
| Antoine Pitrou | ab85ff3 | 2011-07-23 22:03:45 +0200 | [diff] [blame] | 602 |                               'sys.stdout.write("line6\\r");' | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 603 |                               'sys.stdout.flush();' | 
| Antoine Pitrou | ab85ff3 | 2011-07-23 22:03:45 +0200 | [diff] [blame] | 604 |                               'sys.stdout.write("\\nline7");' | 
 | 605 |                               'sys.stdout.flush();' | 
 | 606 |                               'sys.stdout.write("\\nline8");'], | 
 | 607 |                              stderr=subprocess.PIPE, | 
 | 608 |                              stdout=subprocess.PIPE, | 
| Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 609 |                              universal_newlines=1) | 
| Brian Curtin | 19a5379 | 2010-11-05 17:09:05 +0000 | [diff] [blame] | 610 |         self.addCleanup(p.stdout.close) | 
 | 611 |         self.addCleanup(p.stderr.close) | 
| Antoine Pitrou | ab85ff3 | 2011-07-23 22:03:45 +0200 | [diff] [blame] | 612 |         # BUG: can't give a non-empty stdin because it breaks both the | 
 | 613 |         # select- and poll-based communicate() implementations. | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 614 |         (stdout, stderr) = p.communicate() | 
| Antoine Pitrou | ab85ff3 | 2011-07-23 22:03:45 +0200 | [diff] [blame] | 615 |         self.assertEqual(stdout, | 
 | 616 |                          "line2\nline4\nline5\nline6\nline7\nline8") | 
 | 617 |  | 
 | 618 |     def test_universal_newlines_communicate_stdin(self): | 
 | 619 |         # universal newlines through communicate(), with only stdin | 
 | 620 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 621 |                               'import sys,os;' + SETBINARY + '''\nif True: | 
 | 622 |                                   s = sys.stdin.readline() | 
 | 623 |                                   assert s == "line1\\n", repr(s) | 
 | 624 |                                   s = sys.stdin.read() | 
 | 625 |                                   assert s == "line3\\n", repr(s) | 
 | 626 |                               '''], | 
 | 627 |                              stdin=subprocess.PIPE, | 
 | 628 |                              universal_newlines=1) | 
 | 629 |         (stdout, stderr) = p.communicate("line1\nline3\n") | 
 | 630 |         self.assertEqual(p.returncode, 0) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 631 |  | 
 | 632 |     def test_no_leaking(self): | 
| Tim Peters | 7b759da | 2004-10-12 22:29:54 +0000 | [diff] [blame] | 633 |         # Make sure we leak no resources | 
| Antoine Pitrou | 8db3027 | 2010-09-18 22:38:48 +0000 | [diff] [blame] | 634 |         if not mswindows: | 
| Peter Astrand | f7f1bb7 | 2005-03-03 20:47:37 +0000 | [diff] [blame] | 635 |             max_handles = 1026 # too much for most UNIX systems | 
 | 636 |         else: | 
| Antoine Pitrou | 8db3027 | 2010-09-18 22:38:48 +0000 | [diff] [blame] | 637 |             max_handles = 2050 # too much for (at least some) Windows setups | 
 | 638 |         handles = [] | 
| Gregory P. Smith | 81ce685 | 2011-03-15 02:04:11 -0400 | [diff] [blame] | 639 |         tmpdir = tempfile.mkdtemp() | 
| Antoine Pitrou | 8db3027 | 2010-09-18 22:38:48 +0000 | [diff] [blame] | 640 |         try: | 
 | 641 |             for i in range(max_handles): | 
 | 642 |                 try: | 
| Gregory P. Smith | 81ce685 | 2011-03-15 02:04:11 -0400 | [diff] [blame] | 643 |                     tmpfile = os.path.join(tmpdir, support.TESTFN) | 
 | 644 |                     handles.append(os.open(tmpfile, os.O_WRONLY|os.O_CREAT)) | 
| Antoine Pitrou | 8db3027 | 2010-09-18 22:38:48 +0000 | [diff] [blame] | 645 |                 except OSError as e: | 
 | 646 |                     if e.errno != errno.EMFILE: | 
 | 647 |                         raise | 
 | 648 |                     break | 
 | 649 |             else: | 
 | 650 |                 self.skipTest("failed to reach the file descriptor limit " | 
 | 651 |                     "(tried %d)" % max_handles) | 
 | 652 |             # Close a couple of them (should be enough for a subprocess) | 
 | 653 |             for i in range(10): | 
 | 654 |                 os.close(handles.pop()) | 
 | 655 |             # Loop creating some subprocesses. If one of them leaks some fds, | 
 | 656 |             # the next loop iteration will fail by reaching the max fd limit. | 
 | 657 |             for i in range(15): | 
 | 658 |                 p = subprocess.Popen([sys.executable, "-c", | 
 | 659 |                                       "import sys;" | 
 | 660 |                                       "sys.stdout.write(sys.stdin.read())"], | 
 | 661 |                                      stdin=subprocess.PIPE, | 
 | 662 |                                      stdout=subprocess.PIPE, | 
 | 663 |                                      stderr=subprocess.PIPE) | 
 | 664 |                 data = p.communicate(b"lime")[0] | 
 | 665 |                 self.assertEqual(data, b"lime") | 
 | 666 |         finally: | 
 | 667 |             for h in handles: | 
 | 668 |                 os.close(h) | 
| Gregory P. Smith | 81ce685 | 2011-03-15 02:04:11 -0400 | [diff] [blame] | 669 |             shutil.rmtree(tmpdir) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 670 |  | 
 | 671 |     def test_list2cmdline(self): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 672 |         self.assertEqual(subprocess.list2cmdline(['a b c', 'd', 'e']), | 
 | 673 |                          '"a b c" d e') | 
 | 674 |         self.assertEqual(subprocess.list2cmdline(['ab"c', '\\', 'd']), | 
 | 675 |                          'ab\\"c \\ d') | 
| Christian Heimes | fdab48e | 2008-01-20 09:06:41 +0000 | [diff] [blame] | 676 |         self.assertEqual(subprocess.list2cmdline(['ab"c', ' \\', 'd']), | 
 | 677 |                          'ab\\"c " \\\\" d') | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 678 |         self.assertEqual(subprocess.list2cmdline(['a\\\\\\b', 'de fg', 'h']), | 
 | 679 |                          'a\\\\\\b "de fg" h') | 
 | 680 |         self.assertEqual(subprocess.list2cmdline(['a\\"b', 'c', 'd']), | 
 | 681 |                          'a\\\\\\"b c d') | 
 | 682 |         self.assertEqual(subprocess.list2cmdline(['a\\\\b c', 'd', 'e']), | 
 | 683 |                          '"a\\\\b c" d e') | 
 | 684 |         self.assertEqual(subprocess.list2cmdline(['a\\\\b\\ c', 'd', 'e']), | 
 | 685 |                          '"a\\\\b\\ c" d e') | 
| Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 686 |         self.assertEqual(subprocess.list2cmdline(['ab', '']), | 
 | 687 |                          'ab ""') | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 688 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 689 |     def test_poll(self): | 
| Ross Lagerwall | ab66d2a | 2012-02-12 09:01:30 +0200 | [diff] [blame] | 690 |         p = subprocess.Popen([sys.executable, "-c", | 
| Ross Lagerwall | e7ad419 | 2012-02-22 06:02:07 +0200 | [diff] [blame] | 691 |                               "import os; os.read(0, 1)"], | 
 | 692 |                              stdin=subprocess.PIPE) | 
| Ross Lagerwall | ab66d2a | 2012-02-12 09:01:30 +0200 | [diff] [blame] | 693 |         self.addCleanup(p.stdin.close) | 
 | 694 |         self.assertIsNone(p.poll()) | 
 | 695 |         os.write(p.stdin.fileno(), b'A') | 
 | 696 |         p.wait() | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 697 |         # Subsequent invocations should just return the returncode | 
 | 698 |         self.assertEqual(p.poll(), 0) | 
 | 699 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 700 |     def test_wait(self): | 
| Ross Lagerwall | ab66d2a | 2012-02-12 09:01:30 +0200 | [diff] [blame] | 701 |         p = subprocess.Popen([sys.executable, "-c", "pass"]) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 702 |         self.assertEqual(p.wait(), 0) | 
 | 703 |         # Subsequent invocations should just return the returncode | 
 | 704 |         self.assertEqual(p.wait(), 0) | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 705 |  | 
| Reid Kleckner | 31aa7dd | 2011-03-14 12:02:10 -0400 | [diff] [blame] | 706 |     def test_wait_timeout(self): | 
 | 707 |         p = subprocess.Popen([sys.executable, | 
| Reid Kleckner | 93479cc | 2011-03-14 19:32:41 -0400 | [diff] [blame] | 708 |                               "-c", "import time; time.sleep(0.1)"]) | 
| Reid Kleckner | 2b228f0 | 2011-03-16 16:57:54 -0400 | [diff] [blame] | 709 |         with self.assertRaises(subprocess.TimeoutExpired) as c: | 
 | 710 |             p.wait(timeout=0.01) | 
 | 711 |         self.assertIn("0.01", str(c.exception))  # For coverage of __str__. | 
| Reid Kleckner | da9ac72 | 2011-03-16 17:08:21 -0400 | [diff] [blame] | 712 |         # Some heavily loaded buildbots (sparc Debian 3.x) require this much | 
 | 713 |         # time to start. | 
 | 714 |         self.assertEqual(p.wait(timeout=3), 0) | 
| Reid Kleckner | 31aa7dd | 2011-03-14 12:02:10 -0400 | [diff] [blame] | 715 |  | 
| Peter Astrand | 738131d | 2004-11-30 21:04:45 +0000 | [diff] [blame] | 716 |     def test_invalid_bufsize(self): | 
 | 717 |         # an invalid type of the bufsize argument should raise | 
 | 718 |         # TypeError. | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 719 |         with self.assertRaises(TypeError): | 
| Peter Astrand | 738131d | 2004-11-30 21:04:45 +0000 | [diff] [blame] | 720 |             subprocess.Popen([sys.executable, "-c", "pass"], "orange") | 
| Peter Astrand | 738131d | 2004-11-30 21:04:45 +0000 | [diff] [blame] | 721 |  | 
| Guido van Rossum | 46a05a7 | 2007-06-07 21:56:45 +0000 | [diff] [blame] | 722 |     def test_bufsize_is_none(self): | 
 | 723 |         # bufsize=None should be the same as bufsize=0. | 
 | 724 |         p = subprocess.Popen([sys.executable, "-c", "pass"], None) | 
 | 725 |         self.assertEqual(p.wait(), 0) | 
 | 726 |         # Again with keyword arg | 
 | 727 |         p = subprocess.Popen([sys.executable, "-c", "pass"], bufsize=None) | 
 | 728 |         self.assertEqual(p.wait(), 0) | 
 | 729 |  | 
| Benjamin Peterson | d75fcb4 | 2009-02-19 04:22:03 +0000 | [diff] [blame] | 730 |     def test_leaking_fds_on_error(self): | 
 | 731 |         # see bug #5179: Popen leaks file descriptors to PIPEs if | 
 | 732 |         # the child fails to execute; this will eventually exhaust | 
 | 733 |         # the maximum number of open fds. 1024 seems a very common | 
 | 734 |         # value for that limit, but Windows has 2048, so we loop | 
 | 735 |         # 1024 times (each call leaked two fds). | 
 | 736 |         for i in range(1024): | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 737 |             # Windows raises IOError.  Others raise OSError. | 
 | 738 |             with self.assertRaises(EnvironmentError) as c: | 
| Benjamin Peterson | d75fcb4 | 2009-02-19 04:22:03 +0000 | [diff] [blame] | 739 |                 subprocess.Popen(['nonexisting_i_hope'], | 
 | 740 |                                  stdout=subprocess.PIPE, | 
 | 741 |                                  stderr=subprocess.PIPE) | 
| R David Murray | 384069c | 2011-03-13 22:26:53 -0400 | [diff] [blame] | 742 |             # ignore errors that indicate the command was not found | 
| R David Murray | 6924bd7 | 2011-03-13 22:48:55 -0400 | [diff] [blame] | 743 |             if c.exception.errno not in (errno.ENOENT, errno.EACCES): | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 744 |                 raise c.exception | 
| Benjamin Peterson | d75fcb4 | 2009-02-19 04:22:03 +0000 | [diff] [blame] | 745 |  | 
| Victor Stinner | b369358 | 2010-05-21 20:13:12 +0000 | [diff] [blame] | 746 |     def test_issue8780(self): | 
 | 747 |         # Ensure that stdout is inherited from the parent | 
 | 748 |         # if stdout=PIPE is not used | 
 | 749 |         code = ';'.join(( | 
 | 750 |             'import subprocess, sys', | 
 | 751 |             'retcode = subprocess.call(' | 
 | 752 |                 "[sys.executable, '-c', 'print(\"Hello World!\")'])", | 
 | 753 |             'assert retcode == 0')) | 
 | 754 |         output = subprocess.check_output([sys.executable, '-c', code]) | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 755 |         self.assertTrue(output.startswith(b'Hello World!'), ascii(output)) | 
| Victor Stinner | b369358 | 2010-05-21 20:13:12 +0000 | [diff] [blame] | 756 |  | 
| Tim Golden | af5ac39 | 2010-08-06 13:03:56 +0000 | [diff] [blame] | 757 |     def test_handles_closed_on_exception(self): | 
 | 758 |         # If CreateProcess exits with an error, ensure the | 
 | 759 |         # duplicate output handles are released | 
 | 760 |         ifhandle, ifname = mkstemp() | 
 | 761 |         ofhandle, ofname = mkstemp() | 
 | 762 |         efhandle, efname = mkstemp() | 
 | 763 |         try: | 
 | 764 |             subprocess.Popen (["*"], stdin=ifhandle, stdout=ofhandle, | 
 | 765 |               stderr=efhandle) | 
 | 766 |         except OSError: | 
 | 767 |             os.close(ifhandle) | 
 | 768 |             os.remove(ifname) | 
 | 769 |             os.close(ofhandle) | 
 | 770 |             os.remove(ofname) | 
 | 771 |             os.close(efhandle) | 
 | 772 |             os.remove(efname) | 
 | 773 |         self.assertFalse(os.path.exists(ifname)) | 
 | 774 |         self.assertFalse(os.path.exists(ofname)) | 
 | 775 |         self.assertFalse(os.path.exists(efname)) | 
 | 776 |  | 
| Ross Lagerwall | 4f61b02 | 2011-04-05 15:34:00 +0200 | [diff] [blame] | 777 |     def test_communicate_epipe(self): | 
 | 778 |         # Issue 10963: communicate() should hide EPIPE | 
 | 779 |         p = subprocess.Popen([sys.executable, "-c", 'pass'], | 
 | 780 |                              stdin=subprocess.PIPE, | 
 | 781 |                              stdout=subprocess.PIPE, | 
 | 782 |                              stderr=subprocess.PIPE) | 
 | 783 |         self.addCleanup(p.stdout.close) | 
 | 784 |         self.addCleanup(p.stderr.close) | 
 | 785 |         self.addCleanup(p.stdin.close) | 
 | 786 |         p.communicate(b"x" * 2**20) | 
 | 787 |  | 
 | 788 |     def test_communicate_epipe_only_stdin(self): | 
 | 789 |         # Issue 10963: communicate() should hide EPIPE | 
 | 790 |         p = subprocess.Popen([sys.executable, "-c", 'pass'], | 
 | 791 |                              stdin=subprocess.PIPE) | 
 | 792 |         self.addCleanup(p.stdin.close) | 
| Ross Lagerwall | ab66d2a | 2012-02-12 09:01:30 +0200 | [diff] [blame] | 793 |         p.wait() | 
| Ross Lagerwall | 4f61b02 | 2011-04-05 15:34:00 +0200 | [diff] [blame] | 794 |         p.communicate(b"x" * 2**20) | 
 | 795 |  | 
| Ross Lagerwall | ab66d2a | 2012-02-12 09:01:30 +0200 | [diff] [blame] | 796 |     @unittest.skipUnless(hasattr(signal, 'SIGUSR1'), | 
 | 797 |                          "Requires signal.SIGUSR1") | 
 | 798 |     @unittest.skipUnless(hasattr(os, 'kill'), | 
 | 799 |                          "Requires os.kill") | 
 | 800 |     @unittest.skipUnless(hasattr(os, 'getppid'), | 
 | 801 |                          "Requires os.getppid") | 
| Victor Stinner | 2cfb6f3 | 2011-07-05 14:00:56 +0200 | [diff] [blame] | 802 |     def test_communicate_eintr(self): | 
 | 803 |         # Issue #12493: communicate() should handle EINTR | 
 | 804 |         def handler(signum, frame): | 
 | 805 |             pass | 
| Ross Lagerwall | ab66d2a | 2012-02-12 09:01:30 +0200 | [diff] [blame] | 806 |         old_handler = signal.signal(signal.SIGUSR1, handler) | 
 | 807 |         self.addCleanup(signal.signal, signal.SIGUSR1, old_handler) | 
| Victor Stinner | 2cfb6f3 | 2011-07-05 14:00:56 +0200 | [diff] [blame] | 808 |  | 
| Ross Lagerwall | ab66d2a | 2012-02-12 09:01:30 +0200 | [diff] [blame] | 809 |         args = [sys.executable, "-c", | 
 | 810 |                 'import os, signal;' | 
 | 811 |                 'os.kill(os.getppid(), signal.SIGUSR1)'] | 
| Victor Stinner | 2cfb6f3 | 2011-07-05 14:00:56 +0200 | [diff] [blame] | 812 |         for stream in ('stdout', 'stderr'): | 
 | 813 |             kw = {stream: subprocess.PIPE} | 
 | 814 |             with subprocess.Popen(args, **kw) as process: | 
| Ross Lagerwall | ab66d2a | 2012-02-12 09:01:30 +0200 | [diff] [blame] | 815 |                 # communicate() will be interrupted by SIGUSR1 | 
| Victor Stinner | 2cfb6f3 | 2011-07-05 14:00:56 +0200 | [diff] [blame] | 816 |                 process.communicate() | 
 | 817 |  | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 818 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 819 | # context manager | 
 | 820 | class _SuppressCoreFiles(object): | 
 | 821 |     """Try to prevent core files from being created.""" | 
 | 822 |     old_limit = None | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 823 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 824 |     def __enter__(self): | 
 | 825 |         """Try to save previous ulimit, then set it to (0, 0).""" | 
| Benjamin Peterson | 964561b | 2011-12-10 12:31:42 -0500 | [diff] [blame] | 826 |         if resource is not None: | 
 | 827 |             try: | 
 | 828 |                 self.old_limit = resource.getrlimit(resource.RLIMIT_CORE) | 
 | 829 |                 resource.setrlimit(resource.RLIMIT_CORE, (0, 0)) | 
 | 830 |             except (ValueError, resource.error): | 
 | 831 |                 pass | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 832 |  | 
| Ronald Oussoren | 102d11a | 2010-07-23 09:50:05 +0000 | [diff] [blame] | 833 |         if sys.platform == 'darwin': | 
 | 834 |             # Check if the 'Crash Reporter' on OSX was configured | 
 | 835 |             # in 'Developer' mode and warn that it will get triggered | 
 | 836 |             # when it is. | 
 | 837 |             # | 
 | 838 |             # This assumes that this context manager is used in tests | 
 | 839 |             # that might trigger the next manager. | 
 | 840 |             value = subprocess.Popen(['/usr/bin/defaults', 'read', | 
 | 841 |                     'com.apple.CrashReporter', 'DialogType'], | 
 | 842 |                     stdout=subprocess.PIPE).communicate()[0] | 
 | 843 |             if value.strip() == b'developer': | 
 | 844 |                 print("this tests triggers the Crash Reporter, " | 
 | 845 |                       "that is intentional", end='') | 
 | 846 |                 sys.stdout.flush() | 
 | 847 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 848 |     def __exit__(self, *args): | 
 | 849 |         """Return core file behavior to default.""" | 
 | 850 |         if self.old_limit is None: | 
 | 851 |             return | 
| Benjamin Peterson | 964561b | 2011-12-10 12:31:42 -0500 | [diff] [blame] | 852 |         if resource is not None: | 
 | 853 |             try: | 
 | 854 |                 resource.setrlimit(resource.RLIMIT_CORE, self.old_limit) | 
 | 855 |             except (ValueError, resource.error): | 
 | 856 |                 pass | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 857 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 858 |  | 
| Florent Xicluna | f0cbd82 | 2010-03-04 21:50:56 +0000 | [diff] [blame] | 859 | @unittest.skipIf(mswindows, "POSIX specific tests") | 
| Florent Xicluna | c049d87 | 2010-03-27 22:47:23 +0000 | [diff] [blame] | 860 | class POSIXProcessTestCase(BaseTestCase): | 
| Florent Xicluna | f0cbd82 | 2010-03-04 21:50:56 +0000 | [diff] [blame] | 861 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 862 |     def test_exceptions(self): | 
| Gregory P. Smith | fb94c5f | 2010-03-14 06:49:55 +0000 | [diff] [blame] | 863 |         nonexistent_dir = "/_this/pa.th/does/not/exist" | 
 | 864 |         try: | 
 | 865 |             os.chdir(nonexistent_dir) | 
 | 866 |         except OSError as e: | 
 | 867 |             # This avoids hard coding the errno value or the OS perror() | 
 | 868 |             # string and instead capture the exception that we want to see | 
 | 869 |             # below for comparison. | 
 | 870 |             desired_exception = e | 
| Benjamin Peterson | 5f78040 | 2010-11-20 18:07:52 +0000 | [diff] [blame] | 871 |             desired_exception.strerror += ': ' + repr(sys.executable) | 
| Gregory P. Smith | fb94c5f | 2010-03-14 06:49:55 +0000 | [diff] [blame] | 872 |         else: | 
 | 873 |             self.fail("chdir to nonexistant directory %s succeeded." % | 
 | 874 |                       nonexistent_dir) | 
 | 875 |  | 
 | 876 |         # Error in the child re-raised in the parent. | 
 | 877 |         try: | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 878 |             p = subprocess.Popen([sys.executable, "-c", ""], | 
| Gregory P. Smith | fb94c5f | 2010-03-14 06:49:55 +0000 | [diff] [blame] | 879 |                                  cwd=nonexistent_dir) | 
 | 880 |         except OSError as e: | 
 | 881 |             # Test that the child process chdir failure actually makes | 
 | 882 |             # it up to the parent process as the correct exception. | 
 | 883 |             self.assertEqual(desired_exception.errno, e.errno) | 
 | 884 |             self.assertEqual(desired_exception.strerror, e.strerror) | 
 | 885 |         else: | 
 | 886 |             self.fail("Expected OSError: %s" % desired_exception) | 
 | 887 |  | 
 | 888 |     def test_restore_signals(self): | 
 | 889 |         # Code coverage for both values of restore_signals to make sure it | 
 | 890 |         # at least does not blow up. | 
 | 891 |         # A test for behavior would be complex.  Contributions welcome. | 
 | 892 |         subprocess.call([sys.executable, "-c", ""], restore_signals=True) | 
 | 893 |         subprocess.call([sys.executable, "-c", ""], restore_signals=False) | 
 | 894 |  | 
 | 895 |     def test_start_new_session(self): | 
 | 896 |         # For code coverage of calling setsid().  We don't care if we get an | 
 | 897 |         # EPERM error from it depending on the test execution environment, that | 
 | 898 |         # still indicates that it was called. | 
 | 899 |         try: | 
 | 900 |             output = subprocess.check_output( | 
 | 901 |                     [sys.executable, "-c", | 
 | 902 |                      "import os; print(os.getpgid(os.getpid()))"], | 
 | 903 |                     start_new_session=True) | 
 | 904 |         except OSError as e: | 
 | 905 |             if e.errno != errno.EPERM: | 
 | 906 |                 raise | 
 | 907 |         else: | 
 | 908 |             parent_pgid = os.getpgid(os.getpid()) | 
 | 909 |             child_pgid = int(output) | 
 | 910 |             self.assertNotEqual(parent_pgid, child_pgid) | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 911 |  | 
 | 912 |     def test_run_abort(self): | 
 | 913 |         # returncode handles signal termination | 
 | 914 |         with _SuppressCoreFiles(): | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 915 |             p = subprocess.Popen([sys.executable, "-c", | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 916 |                                   'import os; os.abort()']) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 917 |             p.wait() | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 918 |         self.assertEqual(-p.returncode, signal.SIGABRT) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 919 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 920 |     def test_preexec(self): | 
| Gregory P. Smith | fb94c5f | 2010-03-14 06:49:55 +0000 | [diff] [blame] | 921 |         # DISCLAIMER: Setting environment variables is *not* a good use | 
 | 922 |         # of a preexec_fn.  This is merely a test. | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 923 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 924 |                               'import sys,os;' | 
 | 925 |                               'sys.stdout.write(os.getenv("FRUIT"))'], | 
 | 926 |                              stdout=subprocess.PIPE, | 
 | 927 |                              preexec_fn=lambda: os.putenv("FRUIT", "apple")) | 
| Brian Curtin | 3c6a951 | 2010-11-05 03:58:52 +0000 | [diff] [blame] | 928 |         self.addCleanup(p.stdout.close) | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 929 |         self.assertEqual(p.stdout.read(), b"apple") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 930 |  | 
| Gregory P. Smith | fb94c5f | 2010-03-14 06:49:55 +0000 | [diff] [blame] | 931 |     def test_preexec_exception(self): | 
 | 932 |         def raise_it(): | 
 | 933 |             raise ValueError("What if two swallows carried a coconut?") | 
 | 934 |         try: | 
 | 935 |             p = subprocess.Popen([sys.executable, "-c", ""], | 
 | 936 |                                  preexec_fn=raise_it) | 
 | 937 |         except RuntimeError as e: | 
 | 938 |             self.assertTrue( | 
 | 939 |                     subprocess._posixsubprocess, | 
 | 940 |                     "Expected a ValueError from the preexec_fn") | 
 | 941 |         except ValueError as e: | 
 | 942 |             self.assertIn("coconut", e.args[0]) | 
 | 943 |         else: | 
 | 944 |             self.fail("Exception raised by preexec_fn did not make it " | 
 | 945 |                       "to the parent process.") | 
 | 946 |  | 
| Gregory P. Smith | 32ec9da | 2010-03-19 16:53:08 +0000 | [diff] [blame] | 947 |     def test_preexec_gc_module_failure(self): | 
 | 948 |         # This tests the code that disables garbage collection if the child | 
 | 949 |         # process will execute any Python. | 
 | 950 |         def raise_runtime_error(): | 
 | 951 |             raise RuntimeError("this shouldn't escape") | 
 | 952 |         enabled = gc.isenabled() | 
 | 953 |         orig_gc_disable = gc.disable | 
 | 954 |         orig_gc_isenabled = gc.isenabled | 
 | 955 |         try: | 
 | 956 |             gc.disable() | 
 | 957 |             self.assertFalse(gc.isenabled()) | 
 | 958 |             subprocess.call([sys.executable, '-c', ''], | 
 | 959 |                             preexec_fn=lambda: None) | 
 | 960 |             self.assertFalse(gc.isenabled(), | 
 | 961 |                              "Popen enabled gc when it shouldn't.") | 
 | 962 |  | 
 | 963 |             gc.enable() | 
 | 964 |             self.assertTrue(gc.isenabled()) | 
 | 965 |             subprocess.call([sys.executable, '-c', ''], | 
 | 966 |                             preexec_fn=lambda: None) | 
 | 967 |             self.assertTrue(gc.isenabled(), "Popen left gc disabled.") | 
 | 968 |  | 
 | 969 |             gc.disable = raise_runtime_error | 
 | 970 |             self.assertRaises(RuntimeError, subprocess.Popen, | 
 | 971 |                               [sys.executable, '-c', ''], | 
 | 972 |                               preexec_fn=lambda: None) | 
 | 973 |  | 
 | 974 |             del gc.isenabled  # force an AttributeError | 
 | 975 |             self.assertRaises(AttributeError, subprocess.Popen, | 
 | 976 |                               [sys.executable, '-c', ''], | 
 | 977 |                               preexec_fn=lambda: None) | 
 | 978 |         finally: | 
 | 979 |             gc.disable = orig_gc_disable | 
 | 980 |             gc.isenabled = orig_gc_isenabled | 
 | 981 |             if not enabled: | 
 | 982 |                 gc.disable() | 
 | 983 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 984 |     def test_args_string(self): | 
 | 985 |         # args is a string | 
 | 986 |         fd, fname = mkstemp() | 
 | 987 |         # reopen in text mode | 
| Victor Stinner | f6782ac | 2010-10-16 23:46:43 +0000 | [diff] [blame] | 988 |         with open(fd, "w", errors="surrogateescape") as fobj: | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 989 |             fobj.write("#!/bin/sh\n") | 
 | 990 |             fobj.write("exec '%s' -c 'import sys; sys.exit(47)'\n" % | 
 | 991 |                        sys.executable) | 
 | 992 |         os.chmod(fname, 0o700) | 
 | 993 |         p = subprocess.Popen(fname) | 
 | 994 |         p.wait() | 
 | 995 |         os.remove(fname) | 
 | 996 |         self.assertEqual(p.returncode, 47) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 997 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 998 |     def test_invalid_args(self): | 
 | 999 |         # invalid arguments should raise ValueError | 
 | 1000 |         self.assertRaises(ValueError, subprocess.call, | 
 | 1001 |                           [sys.executable, "-c", | 
 | 1002 |                            "import sys; sys.exit(47)"], | 
 | 1003 |                           startupinfo=47) | 
 | 1004 |         self.assertRaises(ValueError, subprocess.call, | 
 | 1005 |                           [sys.executable, "-c", | 
 | 1006 |                            "import sys; sys.exit(47)"], | 
 | 1007 |                           creationflags=47) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1008 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1009 |     def test_shell_sequence(self): | 
 | 1010 |         # Run command through the shell (sequence) | 
 | 1011 |         newenv = os.environ.copy() | 
 | 1012 |         newenv["FRUIT"] = "apple" | 
 | 1013 |         p = subprocess.Popen(["echo $FRUIT"], shell=1, | 
 | 1014 |                              stdout=subprocess.PIPE, | 
 | 1015 |                              env=newenv) | 
| Brian Curtin | 3c6a951 | 2010-11-05 03:58:52 +0000 | [diff] [blame] | 1016 |         self.addCleanup(p.stdout.close) | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1017 |         self.assertEqual(p.stdout.read().strip(b" \t\r\n\f"), b"apple") | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1018 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1019 |     def test_shell_string(self): | 
 | 1020 |         # Run command through the shell (string) | 
 | 1021 |         newenv = os.environ.copy() | 
 | 1022 |         newenv["FRUIT"] = "apple" | 
 | 1023 |         p = subprocess.Popen("echo $FRUIT", shell=1, | 
 | 1024 |                              stdout=subprocess.PIPE, | 
 | 1025 |                              env=newenv) | 
| Brian Curtin | 3c6a951 | 2010-11-05 03:58:52 +0000 | [diff] [blame] | 1026 |         self.addCleanup(p.stdout.close) | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1027 |         self.assertEqual(p.stdout.read().strip(b" \t\r\n\f"), b"apple") | 
| Christian Heimes | a342c01 | 2008-04-20 21:01:16 +0000 | [diff] [blame] | 1028 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1029 |     def test_call_string(self): | 
 | 1030 |         # call() function with string argument on UNIX | 
 | 1031 |         fd, fname = mkstemp() | 
 | 1032 |         # reopen in text mode | 
| Victor Stinner | f6782ac | 2010-10-16 23:46:43 +0000 | [diff] [blame] | 1033 |         with open(fd, "w", errors="surrogateescape") as fobj: | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1034 |             fobj.write("#!/bin/sh\n") | 
 | 1035 |             fobj.write("exec '%s' -c 'import sys; sys.exit(47)'\n" % | 
 | 1036 |                        sys.executable) | 
 | 1037 |         os.chmod(fname, 0o700) | 
 | 1038 |         rc = subprocess.call(fname) | 
 | 1039 |         os.remove(fname) | 
 | 1040 |         self.assertEqual(rc, 47) | 
| Christian Heimes | a342c01 | 2008-04-20 21:01:16 +0000 | [diff] [blame] | 1041 |  | 
| Stefan Krah | 9542cc6 | 2010-07-19 14:20:53 +0000 | [diff] [blame] | 1042 |     def test_specific_shell(self): | 
 | 1043 |         # Issue #9265: Incorrect name passed as arg[0]. | 
 | 1044 |         shells = [] | 
 | 1045 |         for prefix in ['/bin', '/usr/bin/', '/usr/local/bin']: | 
 | 1046 |             for name in ['bash', 'ksh']: | 
 | 1047 |                 sh = os.path.join(prefix, name) | 
 | 1048 |                 if os.path.isfile(sh): | 
 | 1049 |                     shells.append(sh) | 
 | 1050 |         if not shells: # Will probably work for any shell but csh. | 
 | 1051 |             self.skipTest("bash or ksh required for this test") | 
 | 1052 |         sh = '/bin/sh' | 
 | 1053 |         if os.path.isfile(sh) and not os.path.islink(sh): | 
 | 1054 |             # Test will fail if /bin/sh is a symlink to csh. | 
 | 1055 |             shells.append(sh) | 
 | 1056 |         for sh in shells: | 
 | 1057 |             p = subprocess.Popen("echo $0", executable=sh, shell=True, | 
 | 1058 |                                  stdout=subprocess.PIPE) | 
| Brian Curtin | 3c6a951 | 2010-11-05 03:58:52 +0000 | [diff] [blame] | 1059 |             self.addCleanup(p.stdout.close) | 
| Stefan Krah | 9542cc6 | 2010-07-19 14:20:53 +0000 | [diff] [blame] | 1060 |             self.assertEqual(p.stdout.read().strip(), bytes(sh, 'ascii')) | 
 | 1061 |  | 
| Florent Xicluna | 4886d24 | 2010-03-08 13:27:26 +0000 | [diff] [blame] | 1062 |     def _kill_process(self, method, *args): | 
| Florent Xicluna | 1d8ee3a | 2010-03-05 20:26:54 +0000 | [diff] [blame] | 1063 |         # Do not inherit file handles from the parent. | 
 | 1064 |         # It should fix failures on some platforms. | 
| Antoine Pitrou | 3d8580f | 2010-09-20 01:33:21 +0000 | [diff] [blame] | 1065 |         p = subprocess.Popen([sys.executable, "-c", """if 1: | 
 | 1066 |                              import sys, time | 
 | 1067 |                              sys.stdout.write('x\\n') | 
 | 1068 |                              sys.stdout.flush() | 
 | 1069 |                              time.sleep(30) | 
 | 1070 |                              """], | 
 | 1071 |                              close_fds=True, | 
 | 1072 |                              stdin=subprocess.PIPE, | 
 | 1073 |                              stdout=subprocess.PIPE, | 
 | 1074 |                              stderr=subprocess.PIPE) | 
 | 1075 |         # Wait for the interpreter to be completely initialized before | 
 | 1076 |         # sending any signal. | 
 | 1077 |         p.stdout.read(1) | 
 | 1078 |         getattr(p, method)(*args) | 
| Florent Xicluna | 4886d24 | 2010-03-08 13:27:26 +0000 | [diff] [blame] | 1079 |         return p | 
 | 1080 |  | 
 | 1081 |     def test_send_signal(self): | 
 | 1082 |         p = self._kill_process('send_signal', signal.SIGINT) | 
| Florent Xicluna | c049d87 | 2010-03-27 22:47:23 +0000 | [diff] [blame] | 1083 |         _, stderr = p.communicate() | 
 | 1084 |         self.assertIn(b'KeyboardInterrupt', stderr) | 
| Florent Xicluna | f0cbd82 | 2010-03-04 21:50:56 +0000 | [diff] [blame] | 1085 |         self.assertNotEqual(p.wait(), 0) | 
| Christian Heimes | a342c01 | 2008-04-20 21:01:16 +0000 | [diff] [blame] | 1086 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1087 |     def test_kill(self): | 
| Florent Xicluna | 4886d24 | 2010-03-08 13:27:26 +0000 | [diff] [blame] | 1088 |         p = self._kill_process('kill') | 
| Florent Xicluna | c049d87 | 2010-03-27 22:47:23 +0000 | [diff] [blame] | 1089 |         _, stderr = p.communicate() | 
 | 1090 |         self.assertStderrEqual(stderr, b'') | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1091 |         self.assertEqual(p.wait(), -signal.SIGKILL) | 
| Tim Peters | e718f61 | 2004-10-12 21:51:32 +0000 | [diff] [blame] | 1092 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1093 |     def test_terminate(self): | 
| Florent Xicluna | 4886d24 | 2010-03-08 13:27:26 +0000 | [diff] [blame] | 1094 |         p = self._kill_process('terminate') | 
| Florent Xicluna | c049d87 | 2010-03-27 22:47:23 +0000 | [diff] [blame] | 1095 |         _, stderr = p.communicate() | 
 | 1096 |         self.assertStderrEqual(stderr, b'') | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1097 |         self.assertEqual(p.wait(), -signal.SIGTERM) | 
 | 1098 |  | 
| Antoine Pitrou | c9c83ba | 2011-01-03 18:23:55 +0000 | [diff] [blame] | 1099 |     def check_close_std_fds(self, fds): | 
 | 1100 |         # Issue #9905: test that subprocess pipes still work properly with | 
 | 1101 |         # some standard fds closed | 
 | 1102 |         stdin = 0 | 
 | 1103 |         newfds = [] | 
 | 1104 |         for a in fds: | 
 | 1105 |             b = os.dup(a) | 
 | 1106 |             newfds.append(b) | 
 | 1107 |             if a == 0: | 
 | 1108 |                 stdin = b | 
 | 1109 |         try: | 
 | 1110 |             for fd in fds: | 
 | 1111 |                 os.close(fd) | 
 | 1112 |             out, err = subprocess.Popen([sys.executable, "-c", | 
 | 1113 |                               'import sys;' | 
 | 1114 |                               'sys.stdout.write("apple");' | 
 | 1115 |                               'sys.stdout.flush();' | 
 | 1116 |                               'sys.stderr.write("orange")'], | 
 | 1117 |                        stdin=stdin, | 
 | 1118 |                        stdout=subprocess.PIPE, | 
 | 1119 |                        stderr=subprocess.PIPE).communicate() | 
 | 1120 |             err = support.strip_python_stderr(err) | 
 | 1121 |             self.assertEqual((out, err), (b'apple', b'orange')) | 
 | 1122 |         finally: | 
 | 1123 |             for b, a in zip(newfds, fds): | 
 | 1124 |                 os.dup2(b, a) | 
 | 1125 |             for b in newfds: | 
 | 1126 |                 os.close(b) | 
 | 1127 |  | 
 | 1128 |     def test_close_fd_0(self): | 
 | 1129 |         self.check_close_std_fds([0]) | 
 | 1130 |  | 
 | 1131 |     def test_close_fd_1(self): | 
 | 1132 |         self.check_close_std_fds([1]) | 
 | 1133 |  | 
 | 1134 |     def test_close_fd_2(self): | 
 | 1135 |         self.check_close_std_fds([2]) | 
 | 1136 |  | 
 | 1137 |     def test_close_fds_0_1(self): | 
 | 1138 |         self.check_close_std_fds([0, 1]) | 
 | 1139 |  | 
 | 1140 |     def test_close_fds_0_2(self): | 
 | 1141 |         self.check_close_std_fds([0, 2]) | 
 | 1142 |  | 
 | 1143 |     def test_close_fds_1_2(self): | 
 | 1144 |         self.check_close_std_fds([1, 2]) | 
 | 1145 |  | 
 | 1146 |     def test_close_fds_0_1_2(self): | 
 | 1147 |         # Issue #10806: test that subprocess pipes still work properly with | 
 | 1148 |         # all standard fds closed. | 
 | 1149 |         self.check_close_std_fds([0, 1, 2]) | 
 | 1150 |  | 
| Antoine Pitrou | 95aaeee | 2011-01-03 21:15:48 +0000 | [diff] [blame] | 1151 |     def test_remapping_std_fds(self): | 
 | 1152 |         # open up some temporary files | 
 | 1153 |         temps = [mkstemp() for i in range(3)] | 
 | 1154 |         try: | 
 | 1155 |             temp_fds = [fd for fd, fname in temps] | 
 | 1156 |  | 
 | 1157 |             # unlink the files -- we won't need to reopen them | 
 | 1158 |             for fd, fname in temps: | 
 | 1159 |                 os.unlink(fname) | 
 | 1160 |  | 
 | 1161 |             # write some data to what will become stdin, and rewind | 
 | 1162 |             os.write(temp_fds[1], b"STDIN") | 
 | 1163 |             os.lseek(temp_fds[1], 0, 0) | 
 | 1164 |  | 
 | 1165 |             # move the standard file descriptors out of the way | 
 | 1166 |             saved_fds = [os.dup(fd) for fd in range(3)] | 
 | 1167 |             try: | 
 | 1168 |                 # duplicate the file objects over the standard fd's | 
 | 1169 |                 for fd, temp_fd in enumerate(temp_fds): | 
 | 1170 |                     os.dup2(temp_fd, fd) | 
 | 1171 |  | 
 | 1172 |                 # now use those files in the "wrong" order, so that subprocess | 
 | 1173 |                 # has to rearrange them in the child | 
 | 1174 |                 p = subprocess.Popen([sys.executable, "-c", | 
 | 1175 |                     'import sys; got = sys.stdin.read();' | 
 | 1176 |                     'sys.stdout.write("got %s"%got); sys.stderr.write("err")'], | 
 | 1177 |                     stdin=temp_fds[1], | 
 | 1178 |                     stdout=temp_fds[2], | 
 | 1179 |                     stderr=temp_fds[0]) | 
 | 1180 |                 p.wait() | 
 | 1181 |             finally: | 
 | 1182 |                 # restore the original fd's underneath sys.stdin, etc. | 
 | 1183 |                 for std, saved in enumerate(saved_fds): | 
 | 1184 |                     os.dup2(saved, std) | 
 | 1185 |                     os.close(saved) | 
 | 1186 |  | 
 | 1187 |             for fd in temp_fds: | 
 | 1188 |                 os.lseek(fd, 0, 0) | 
 | 1189 |  | 
 | 1190 |             out = os.read(temp_fds[2], 1024) | 
 | 1191 |             err = support.strip_python_stderr(os.read(temp_fds[0], 1024)) | 
 | 1192 |             self.assertEqual(out, b"got STDIN") | 
 | 1193 |             self.assertEqual(err, b"err") | 
 | 1194 |  | 
 | 1195 |         finally: | 
 | 1196 |             for fd in temp_fds: | 
 | 1197 |                 os.close(fd) | 
 | 1198 |  | 
| Ross Lagerwall | d98646e | 2011-07-27 07:16:31 +0200 | [diff] [blame] | 1199 |     def check_swap_fds(self, stdin_no, stdout_no, stderr_no): | 
 | 1200 |         # open up some temporary files | 
 | 1201 |         temps = [mkstemp() for i in range(3)] | 
 | 1202 |         temp_fds = [fd for fd, fname in temps] | 
 | 1203 |         try: | 
 | 1204 |             # unlink the files -- we won't need to reopen them | 
 | 1205 |             for fd, fname in temps: | 
 | 1206 |                 os.unlink(fname) | 
 | 1207 |  | 
 | 1208 |             # save a copy of the standard file descriptors | 
 | 1209 |             saved_fds = [os.dup(fd) for fd in range(3)] | 
 | 1210 |             try: | 
 | 1211 |                 # duplicate the temp files over the standard fd's 0, 1, 2 | 
 | 1212 |                 for fd, temp_fd in enumerate(temp_fds): | 
 | 1213 |                     os.dup2(temp_fd, fd) | 
 | 1214 |  | 
 | 1215 |                 # write some data to what will become stdin, and rewind | 
 | 1216 |                 os.write(stdin_no, b"STDIN") | 
 | 1217 |                 os.lseek(stdin_no, 0, 0) | 
 | 1218 |  | 
 | 1219 |                 # now use those files in the given order, so that subprocess | 
 | 1220 |                 # has to rearrange them in the child | 
 | 1221 |                 p = subprocess.Popen([sys.executable, "-c", | 
 | 1222 |                     'import sys; got = sys.stdin.read();' | 
 | 1223 |                     'sys.stdout.write("got %s"%got); sys.stderr.write("err")'], | 
 | 1224 |                     stdin=stdin_no, | 
 | 1225 |                     stdout=stdout_no, | 
 | 1226 |                     stderr=stderr_no) | 
 | 1227 |                 p.wait() | 
 | 1228 |  | 
 | 1229 |                 for fd in temp_fds: | 
 | 1230 |                     os.lseek(fd, 0, 0) | 
 | 1231 |  | 
 | 1232 |                 out = os.read(stdout_no, 1024) | 
 | 1233 |                 err = support.strip_python_stderr(os.read(stderr_no, 1024)) | 
 | 1234 |             finally: | 
 | 1235 |                 for std, saved in enumerate(saved_fds): | 
 | 1236 |                     os.dup2(saved, std) | 
 | 1237 |                     os.close(saved) | 
 | 1238 |  | 
 | 1239 |             self.assertEqual(out, b"got STDIN") | 
 | 1240 |             self.assertEqual(err, b"err") | 
 | 1241 |  | 
 | 1242 |         finally: | 
 | 1243 |             for fd in temp_fds: | 
 | 1244 |                 os.close(fd) | 
 | 1245 |  | 
 | 1246 |     # When duping fds, if there arises a situation where one of the fds is | 
 | 1247 |     # either 0, 1 or 2, it is possible that it is overwritten (#12607). | 
 | 1248 |     # This tests all combinations of this. | 
 | 1249 |     def test_swap_fds(self): | 
 | 1250 |         self.check_swap_fds(0, 1, 2) | 
 | 1251 |         self.check_swap_fds(0, 2, 1) | 
 | 1252 |         self.check_swap_fds(1, 0, 2) | 
 | 1253 |         self.check_swap_fds(1, 2, 0) | 
 | 1254 |         self.check_swap_fds(2, 0, 1) | 
 | 1255 |         self.check_swap_fds(2, 1, 0) | 
 | 1256 |  | 
| Victor Stinner | 13bb71c | 2010-04-23 21:41:56 +0000 | [diff] [blame] | 1257 |     def test_surrogates_error_message(self): | 
| Victor Stinner | 4d07804 | 2010-04-23 19:28:32 +0000 | [diff] [blame] | 1258 |         def prepare(): | 
 | 1259 |             raise ValueError("surrogate:\uDCff") | 
 | 1260 |  | 
 | 1261 |         try: | 
 | 1262 |             subprocess.call( | 
 | 1263 |                 [sys.executable, "-c", "pass"], | 
 | 1264 |                 preexec_fn=prepare) | 
 | 1265 |         except ValueError as err: | 
 | 1266 |             # Pure Python implementations keeps the message | 
 | 1267 |             self.assertIsNone(subprocess._posixsubprocess) | 
 | 1268 |             self.assertEqual(str(err), "surrogate:\uDCff") | 
 | 1269 |         except RuntimeError as err: | 
 | 1270 |             # _posixsubprocess uses a default message | 
 | 1271 |             self.assertIsNotNone(subprocess._posixsubprocess) | 
 | 1272 |             self.assertEqual(str(err), "Exception occurred in preexec_fn.") | 
 | 1273 |         else: | 
 | 1274 |             self.fail("Expected ValueError or RuntimeError") | 
 | 1275 |  | 
| Victor Stinner | 13bb71c | 2010-04-23 21:41:56 +0000 | [diff] [blame] | 1276 |     def test_undecodable_env(self): | 
 | 1277 |         for key, value in (('test', 'abc\uDCFF'), ('test\uDCFF', '42')): | 
| Victor Stinner | 13bb71c | 2010-04-23 21:41:56 +0000 | [diff] [blame] | 1278 |             # test str with surrogates | 
| Antoine Pitrou | fb8db8f | 2010-09-19 22:46:05 +0000 | [diff] [blame] | 1279 |             script = "import os; print(ascii(os.getenv(%s)))" % repr(key) | 
| Victor Stinner | ce2d24d | 2010-04-23 22:55:39 +0000 | [diff] [blame] | 1280 |             env = os.environ.copy() | 
 | 1281 |             env[key] = value | 
| Victor Stinner | 89f3ad1 | 2010-10-14 10:43:31 +0000 | [diff] [blame] | 1282 |             # Use C locale to get ascii for the locale encoding to force | 
 | 1283 |             # surrogate-escaping of \xFF in the child process; otherwise it can | 
 | 1284 |             # be decoded as-is if the default locale is latin-1. | 
| Victor Stinner | ebc78d2 | 2010-10-14 10:38:17 +0000 | [diff] [blame] | 1285 |             env['LC_ALL'] = 'C' | 
| Victor Stinner | 13bb71c | 2010-04-23 21:41:56 +0000 | [diff] [blame] | 1286 |             stdout = subprocess.check_output( | 
 | 1287 |                 [sys.executable, "-c", script], | 
| Victor Stinner | ce2d24d | 2010-04-23 22:55:39 +0000 | [diff] [blame] | 1288 |                 env=env) | 
| Victor Stinner | 13bb71c | 2010-04-23 21:41:56 +0000 | [diff] [blame] | 1289 |             stdout = stdout.rstrip(b'\n\r') | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 1290 |             self.assertEqual(stdout.decode('ascii'), ascii(value)) | 
| Victor Stinner | 13bb71c | 2010-04-23 21:41:56 +0000 | [diff] [blame] | 1291 |  | 
 | 1292 |             # test bytes | 
 | 1293 |             key = key.encode("ascii", "surrogateescape") | 
 | 1294 |             value = value.encode("ascii", "surrogateescape") | 
| Antoine Pitrou | fb8db8f | 2010-09-19 22:46:05 +0000 | [diff] [blame] | 1295 |             script = "import os; print(ascii(os.getenvb(%s)))" % repr(key) | 
| Victor Stinner | ce2d24d | 2010-04-23 22:55:39 +0000 | [diff] [blame] | 1296 |             env = os.environ.copy() | 
 | 1297 |             env[key] = value | 
| Victor Stinner | 13bb71c | 2010-04-23 21:41:56 +0000 | [diff] [blame] | 1298 |             stdout = subprocess.check_output( | 
 | 1299 |                 [sys.executable, "-c", script], | 
| Victor Stinner | ce2d24d | 2010-04-23 22:55:39 +0000 | [diff] [blame] | 1300 |                 env=env) | 
| Victor Stinner | 13bb71c | 2010-04-23 21:41:56 +0000 | [diff] [blame] | 1301 |             stdout = stdout.rstrip(b'\n\r') | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 1302 |             self.assertEqual(stdout.decode('ascii'), ascii(value)) | 
| Victor Stinner | 13bb71c | 2010-04-23 21:41:56 +0000 | [diff] [blame] | 1303 |  | 
| Victor Stinner | b745a74 | 2010-05-18 17:17:23 +0000 | [diff] [blame] | 1304 |     def test_bytes_program(self): | 
 | 1305 |         abs_program = os.fsencode(sys.executable) | 
 | 1306 |         path, program = os.path.split(sys.executable) | 
 | 1307 |         program = os.fsencode(program) | 
 | 1308 |  | 
 | 1309 |         # absolute bytes path | 
 | 1310 |         exitcode = subprocess.call([abs_program, "-c", "pass"]) | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 1311 |         self.assertEqual(exitcode, 0) | 
| Victor Stinner | b745a74 | 2010-05-18 17:17:23 +0000 | [diff] [blame] | 1312 |  | 
| Victor Stinner | 7b3b20a | 2011-03-03 12:54:05 +0000 | [diff] [blame] | 1313 |         # absolute bytes path as a string | 
 | 1314 |         cmd = b"'" + abs_program + b"' -c pass" | 
 | 1315 |         exitcode = subprocess.call(cmd, shell=True) | 
 | 1316 |         self.assertEqual(exitcode, 0) | 
 | 1317 |  | 
| Victor Stinner | b745a74 | 2010-05-18 17:17:23 +0000 | [diff] [blame] | 1318 |         # bytes program, unicode PATH | 
 | 1319 |         env = os.environ.copy() | 
 | 1320 |         env["PATH"] = path | 
 | 1321 |         exitcode = subprocess.call([program, "-c", "pass"], env=env) | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 1322 |         self.assertEqual(exitcode, 0) | 
| Victor Stinner | b745a74 | 2010-05-18 17:17:23 +0000 | [diff] [blame] | 1323 |  | 
 | 1324 |         # bytes program, bytes PATH | 
 | 1325 |         envb = os.environb.copy() | 
 | 1326 |         envb[b"PATH"] = os.fsencode(path) | 
 | 1327 |         exitcode = subprocess.call([program, "-c", "pass"], env=envb) | 
| Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 1328 |         self.assertEqual(exitcode, 0) | 
| Victor Stinner | b745a74 | 2010-05-18 17:17:23 +0000 | [diff] [blame] | 1329 |  | 
| Gregory P. Smith | 51ee270 | 2010-12-13 07:59:39 +0000 | [diff] [blame] | 1330 |     def test_pipe_cloexec(self): | 
 | 1331 |         sleeper = support.findfile("input_reader.py", subdir="subprocessdata") | 
 | 1332 |         fd_status = support.findfile("fd_status.py", subdir="subprocessdata") | 
 | 1333 |  | 
 | 1334 |         p1 = subprocess.Popen([sys.executable, sleeper], | 
 | 1335 |                               stdin=subprocess.PIPE, stdout=subprocess.PIPE, | 
 | 1336 |                               stderr=subprocess.PIPE, close_fds=False) | 
 | 1337 |  | 
 | 1338 |         self.addCleanup(p1.communicate, b'') | 
 | 1339 |  | 
 | 1340 |         p2 = subprocess.Popen([sys.executable, fd_status], | 
 | 1341 |                               stdout=subprocess.PIPE, close_fds=False) | 
 | 1342 |  | 
 | 1343 |         output, error = p2.communicate() | 
 | 1344 |         result_fds = set(map(int, output.split(b','))) | 
 | 1345 |         unwanted_fds = set([p1.stdin.fileno(), p1.stdout.fileno(), | 
 | 1346 |                             p1.stderr.fileno()]) | 
 | 1347 |  | 
 | 1348 |         self.assertFalse(result_fds & unwanted_fds, | 
 | 1349 |                          "Expected no fds from %r to be open in child, " | 
 | 1350 |                          "found %r" % | 
 | 1351 |                               (unwanted_fds, result_fds & unwanted_fds)) | 
 | 1352 |  | 
 | 1353 |     def test_pipe_cloexec_real_tools(self): | 
 | 1354 |         qcat = support.findfile("qcat.py", subdir="subprocessdata") | 
 | 1355 |         qgrep = support.findfile("qgrep.py", subdir="subprocessdata") | 
 | 1356 |  | 
 | 1357 |         subdata = b'zxcvbn' | 
 | 1358 |         data = subdata * 4 + b'\n' | 
 | 1359 |  | 
 | 1360 |         p1 = subprocess.Popen([sys.executable, qcat], | 
 | 1361 |                               stdin=subprocess.PIPE, stdout=subprocess.PIPE, | 
 | 1362 |                               close_fds=False) | 
 | 1363 |  | 
 | 1364 |         p2 = subprocess.Popen([sys.executable, qgrep, subdata], | 
 | 1365 |                               stdin=p1.stdout, stdout=subprocess.PIPE, | 
 | 1366 |                               close_fds=False) | 
 | 1367 |  | 
 | 1368 |         self.addCleanup(p1.wait) | 
 | 1369 |         self.addCleanup(p2.wait) | 
| Gregory P. Smith | 886455c | 2012-01-21 22:05:10 -0800 | [diff] [blame] | 1370 |         def kill_p1(): | 
 | 1371 |             try: | 
 | 1372 |                 p1.terminate() | 
 | 1373 |             except ProcessLookupError: | 
 | 1374 |                 pass | 
 | 1375 |         def kill_p2(): | 
 | 1376 |             try: | 
 | 1377 |                 p2.terminate() | 
 | 1378 |             except ProcessLookupError: | 
 | 1379 |                 pass | 
 | 1380 |         self.addCleanup(kill_p1) | 
 | 1381 |         self.addCleanup(kill_p2) | 
| Gregory P. Smith | 51ee270 | 2010-12-13 07:59:39 +0000 | [diff] [blame] | 1382 |  | 
 | 1383 |         p1.stdin.write(data) | 
 | 1384 |         p1.stdin.close() | 
 | 1385 |  | 
 | 1386 |         readfiles, ignored1, ignored2 = select.select([p2.stdout], [], [], 10) | 
 | 1387 |  | 
 | 1388 |         self.assertTrue(readfiles, "The child hung") | 
 | 1389 |         self.assertEqual(p2.stdout.read(), data) | 
 | 1390 |  | 
| Victor Stinner | faa8c13 | 2011-01-03 16:36:00 +0000 | [diff] [blame] | 1391 |         p1.stdout.close() | 
 | 1392 |         p2.stdout.close() | 
 | 1393 |  | 
| Gregory P. Smith | 51ee270 | 2010-12-13 07:59:39 +0000 | [diff] [blame] | 1394 |     def test_close_fds(self): | 
 | 1395 |         fd_status = support.findfile("fd_status.py", subdir="subprocessdata") | 
 | 1396 |  | 
 | 1397 |         fds = os.pipe() | 
 | 1398 |         self.addCleanup(os.close, fds[0]) | 
 | 1399 |         self.addCleanup(os.close, fds[1]) | 
 | 1400 |  | 
 | 1401 |         open_fds = set(fds) | 
| Gregory P. Smith | 8facece | 2012-01-21 14:01:08 -0800 | [diff] [blame] | 1402 |         # add a bunch more fds | 
 | 1403 |         for _ in range(9): | 
 | 1404 |             fd = os.open("/dev/null", os.O_RDONLY) | 
 | 1405 |             self.addCleanup(os.close, fd) | 
 | 1406 |             open_fds.add(fd) | 
| Gregory P. Smith | 51ee270 | 2010-12-13 07:59:39 +0000 | [diff] [blame] | 1407 |  | 
 | 1408 |         p = subprocess.Popen([sys.executable, fd_status], | 
 | 1409 |                              stdout=subprocess.PIPE, close_fds=False) | 
 | 1410 |         output, ignored = p.communicate() | 
 | 1411 |         remaining_fds = set(map(int, output.split(b','))) | 
 | 1412 |  | 
 | 1413 |         self.assertEqual(remaining_fds & open_fds, open_fds, | 
 | 1414 |                          "Some fds were closed") | 
 | 1415 |  | 
 | 1416 |         p = subprocess.Popen([sys.executable, fd_status], | 
 | 1417 |                              stdout=subprocess.PIPE, close_fds=True) | 
 | 1418 |         output, ignored = p.communicate() | 
 | 1419 |         remaining_fds = set(map(int, output.split(b','))) | 
 | 1420 |  | 
 | 1421 |         self.assertFalse(remaining_fds & open_fds, | 
 | 1422 |                          "Some fds were left open") | 
 | 1423 |         self.assertIn(1, remaining_fds, "Subprocess failed") | 
 | 1424 |  | 
| Gregory P. Smith | 8facece | 2012-01-21 14:01:08 -0800 | [diff] [blame] | 1425 |         # Keep some of the fd's we opened open in the subprocess. | 
 | 1426 |         # This tests _posixsubprocess.c's proper handling of fds_to_keep. | 
 | 1427 |         fds_to_keep = set(open_fds.pop() for _ in range(8)) | 
 | 1428 |         p = subprocess.Popen([sys.executable, fd_status], | 
 | 1429 |                              stdout=subprocess.PIPE, close_fds=True, | 
 | 1430 |                              pass_fds=()) | 
 | 1431 |         output, ignored = p.communicate() | 
 | 1432 |         remaining_fds = set(map(int, output.split(b','))) | 
 | 1433 |  | 
 | 1434 |         self.assertFalse(remaining_fds & fds_to_keep & open_fds, | 
 | 1435 |                          "Some fds not in pass_fds were left open") | 
 | 1436 |         self.assertIn(1, remaining_fds, "Subprocess failed") | 
 | 1437 |  | 
| Victor Stinner | 88701e2 | 2011-06-01 13:13:04 +0200 | [diff] [blame] | 1438 |     # Mac OS X Tiger (10.4) has a kernel bug: sometimes, the file | 
 | 1439 |     # descriptor of a pipe closed in the parent process is valid in the | 
 | 1440 |     # child process according to fstat(), but the mode of the file | 
 | 1441 |     # descriptor is invalid, and read or write raise an error. | 
 | 1442 |     @support.requires_mac_ver(10, 5) | 
| Gregory P. Smith | 8edd99d | 2010-12-14 13:43:30 +0000 | [diff] [blame] | 1443 |     def test_pass_fds(self): | 
 | 1444 |         fd_status = support.findfile("fd_status.py", subdir="subprocessdata") | 
 | 1445 |  | 
 | 1446 |         open_fds = set() | 
 | 1447 |  | 
 | 1448 |         for x in range(5): | 
 | 1449 |             fds = os.pipe() | 
 | 1450 |             self.addCleanup(os.close, fds[0]) | 
 | 1451 |             self.addCleanup(os.close, fds[1]) | 
 | 1452 |             open_fds.update(fds) | 
 | 1453 |  | 
 | 1454 |         for fd in open_fds: | 
 | 1455 |             p = subprocess.Popen([sys.executable, fd_status], | 
 | 1456 |                                  stdout=subprocess.PIPE, close_fds=True, | 
 | 1457 |                                  pass_fds=(fd, )) | 
 | 1458 |             output, ignored = p.communicate() | 
 | 1459 |  | 
 | 1460 |             remaining_fds = set(map(int, output.split(b','))) | 
 | 1461 |             to_be_closed = open_fds - {fd} | 
 | 1462 |  | 
 | 1463 |             self.assertIn(fd, remaining_fds, "fd to be passed not passed") | 
 | 1464 |             self.assertFalse(remaining_fds & to_be_closed, | 
 | 1465 |                              "fd to be closed passed") | 
 | 1466 |  | 
 | 1467 |             # pass_fds overrides close_fds with a warning. | 
 | 1468 |             with self.assertWarns(RuntimeWarning) as context: | 
 | 1469 |                 self.assertFalse(subprocess.call( | 
 | 1470 |                         [sys.executable, "-c", "import sys; sys.exit(0)"], | 
 | 1471 |                         close_fds=False, pass_fds=(fd, ))) | 
 | 1472 |             self.assertIn('overriding close_fds', str(context.warning)) | 
 | 1473 |  | 
| Gregory P. Smith | 112bb3a | 2011-03-15 14:55:17 -0400 | [diff] [blame] | 1474 |     def test_stdout_stdin_are_single_inout_fd(self): | 
 | 1475 |         with io.open(os.devnull, "r+") as inout: | 
 | 1476 |             p = subprocess.Popen([sys.executable, "-c", "import sys; sys.exit(0)"], | 
 | 1477 |                                  stdout=inout, stdin=inout) | 
 | 1478 |             p.wait() | 
 | 1479 |  | 
 | 1480 |     def test_stdout_stderr_are_single_inout_fd(self): | 
 | 1481 |         with io.open(os.devnull, "r+") as inout: | 
 | 1482 |             p = subprocess.Popen([sys.executable, "-c", "import sys; sys.exit(0)"], | 
 | 1483 |                                  stdout=inout, stderr=inout) | 
 | 1484 |             p.wait() | 
 | 1485 |  | 
 | 1486 |     def test_stderr_stdin_are_single_inout_fd(self): | 
 | 1487 |         with io.open(os.devnull, "r+") as inout: | 
 | 1488 |             p = subprocess.Popen([sys.executable, "-c", "import sys; sys.exit(0)"], | 
 | 1489 |                                  stderr=inout, stdin=inout) | 
 | 1490 |             p.wait() | 
 | 1491 |  | 
| Gregory P. Smith | e85db2b | 2010-12-14 14:38:00 +0000 | [diff] [blame] | 1492 |     def test_wait_when_sigchild_ignored(self): | 
 | 1493 |         # NOTE: sigchild_ignore.py may not be an effective test on all OSes. | 
 | 1494 |         sigchild_ignore = support.findfile("sigchild_ignore.py", | 
 | 1495 |                                            subdir="subprocessdata") | 
 | 1496 |         p = subprocess.Popen([sys.executable, sigchild_ignore], | 
 | 1497 |                              stdout=subprocess.PIPE, stderr=subprocess.PIPE) | 
 | 1498 |         stdout, stderr = p.communicate() | 
 | 1499 |         self.assertEqual(0, p.returncode, "sigchild_ignore.py exited" | 
| Gregory P. Smith | a80f4fb | 2010-12-14 15:23:02 +0000 | [diff] [blame] | 1500 |                          " non-zero with this error:\n%s" % | 
| Marc-André Lemburg | 8f36af7 | 2011-02-25 15:42:01 +0000 | [diff] [blame] | 1501 |                          stderr.decode('utf-8')) | 
| Gregory P. Smith | e85db2b | 2010-12-14 14:38:00 +0000 | [diff] [blame] | 1502 |  | 
| Antoine Pitrou | 7b98d02 | 2011-03-19 17:04:13 +0100 | [diff] [blame] | 1503 |     def test_select_unbuffered(self): | 
 | 1504 |         # Issue #11459: bufsize=0 should really set the pipes as | 
 | 1505 |         # unbuffered (and therefore let select() work properly). | 
 | 1506 |         select = support.import_module("select") | 
 | 1507 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 1508 |                               'import sys;' | 
 | 1509 |                               'sys.stdout.write("apple")'], | 
 | 1510 |                              stdout=subprocess.PIPE, | 
 | 1511 |                              bufsize=0) | 
 | 1512 |         f = p.stdout | 
| Ross Lagerwall | 17ace7a | 2011-03-26 21:21:46 +0200 | [diff] [blame] | 1513 |         self.addCleanup(f.close) | 
| Antoine Pitrou | 7b98d02 | 2011-03-19 17:04:13 +0100 | [diff] [blame] | 1514 |         try: | 
 | 1515 |             self.assertEqual(f.read(4), b"appl") | 
 | 1516 |             self.assertIn(f, select.select([f], [], [], 0.0)[0]) | 
 | 1517 |         finally: | 
 | 1518 |             p.wait() | 
 | 1519 |  | 
| Charles-François Natali | 134a8ba | 2011-08-18 18:49:39 +0200 | [diff] [blame] | 1520 |     def test_zombie_fast_process_del(self): | 
 | 1521 |         # Issue #12650: on Unix, if Popen.__del__() was called before the | 
 | 1522 |         # process exited, it wouldn't be added to subprocess._active, and would | 
 | 1523 |         # remain a zombie. | 
 | 1524 |         # spawn a Popen, and delete its reference before it exits | 
 | 1525 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 1526 |                               'import sys, time;' | 
 | 1527 |                               'time.sleep(0.2)'], | 
 | 1528 |                              stdout=subprocess.PIPE, | 
 | 1529 |                              stderr=subprocess.PIPE) | 
| Nadeem Vawda | 0d7cda3 | 2011-08-19 05:12:01 +0200 | [diff] [blame] | 1530 |         self.addCleanup(p.stdout.close) | 
 | 1531 |         self.addCleanup(p.stderr.close) | 
| Charles-François Natali | 134a8ba | 2011-08-18 18:49:39 +0200 | [diff] [blame] | 1532 |         ident = id(p) | 
 | 1533 |         pid = p.pid | 
 | 1534 |         del p | 
 | 1535 |         # check that p is in the active processes list | 
 | 1536 |         self.assertIn(ident, [id(o) for o in subprocess._active]) | 
 | 1537 |  | 
| Charles-François Natali | 134a8ba | 2011-08-18 18:49:39 +0200 | [diff] [blame] | 1538 |     def test_leak_fast_process_del_killed(self): | 
 | 1539 |         # Issue #12650: on Unix, if Popen.__del__() was called before the | 
 | 1540 |         # process exited, and the process got killed by a signal, it would never | 
 | 1541 |         # be removed from subprocess._active, which triggered a FD and memory | 
 | 1542 |         # leak. | 
 | 1543 |         # spawn a Popen, delete its reference and kill it | 
 | 1544 |         p = subprocess.Popen([sys.executable, "-c", | 
 | 1545 |                               'import time;' | 
 | 1546 |                               'time.sleep(3)'], | 
 | 1547 |                              stdout=subprocess.PIPE, | 
 | 1548 |                              stderr=subprocess.PIPE) | 
| Nadeem Vawda | 0d7cda3 | 2011-08-19 05:12:01 +0200 | [diff] [blame] | 1549 |         self.addCleanup(p.stdout.close) | 
 | 1550 |         self.addCleanup(p.stderr.close) | 
| Charles-François Natali | 134a8ba | 2011-08-18 18:49:39 +0200 | [diff] [blame] | 1551 |         ident = id(p) | 
 | 1552 |         pid = p.pid | 
 | 1553 |         del p | 
 | 1554 |         os.kill(pid, signal.SIGKILL) | 
 | 1555 |         # check that p is in the active processes list | 
 | 1556 |         self.assertIn(ident, [id(o) for o in subprocess._active]) | 
 | 1557 |  | 
 | 1558 |         # let some time for the process to exit, and create a new Popen: this | 
 | 1559 |         # should trigger the wait() of p | 
 | 1560 |         time.sleep(0.2) | 
 | 1561 |         with self.assertRaises(EnvironmentError) as c: | 
 | 1562 |             with subprocess.Popen(['nonexisting_i_hope'], | 
 | 1563 |                                   stdout=subprocess.PIPE, | 
 | 1564 |                                   stderr=subprocess.PIPE) as proc: | 
 | 1565 |                 pass | 
 | 1566 |         # p should have been wait()ed on, and removed from the _active list | 
 | 1567 |         self.assertRaises(OSError, os.waitpid, pid, 0) | 
 | 1568 |         self.assertNotIn(ident, [id(o) for o in subprocess._active]) | 
 | 1569 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1570 |  | 
| Florent Xicluna | f0cbd82 | 2010-03-04 21:50:56 +0000 | [diff] [blame] | 1571 | @unittest.skipUnless(mswindows, "Windows specific tests") | 
| Florent Xicluna | c049d87 | 2010-03-27 22:47:23 +0000 | [diff] [blame] | 1572 | class Win32ProcessTestCase(BaseTestCase): | 
| Florent Xicluna | f0cbd82 | 2010-03-04 21:50:56 +0000 | [diff] [blame] | 1573 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1574 |     def test_startupinfo(self): | 
 | 1575 |         # startupinfo argument | 
 | 1576 |         # We uses hardcoded constants, because we do not want to | 
 | 1577 |         # depend on win32all. | 
 | 1578 |         STARTF_USESHOWWINDOW = 1 | 
 | 1579 |         SW_MAXIMIZE = 3 | 
 | 1580 |         startupinfo = subprocess.STARTUPINFO() | 
 | 1581 |         startupinfo.dwFlags = STARTF_USESHOWWINDOW | 
 | 1582 |         startupinfo.wShowWindow = SW_MAXIMIZE | 
 | 1583 |         # Since Python is a console process, it won't be affected | 
 | 1584 |         # by wShowWindow, but the argument should be silently | 
 | 1585 |         # ignored | 
 | 1586 |         subprocess.call([sys.executable, "-c", "import sys; sys.exit(0)"], | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1587 |                         startupinfo=startupinfo) | 
 | 1588 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1589 |     def test_creationflags(self): | 
 | 1590 |         # creationflags argument | 
 | 1591 |         CREATE_NEW_CONSOLE = 16 | 
 | 1592 |         sys.stderr.write("    a DOS box should flash briefly ...\n") | 
 | 1593 |         subprocess.call(sys.executable + | 
 | 1594 |                         ' -c "import time; time.sleep(0.25)"', | 
 | 1595 |                         creationflags=CREATE_NEW_CONSOLE) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1596 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1597 |     def test_invalid_args(self): | 
 | 1598 |         # invalid arguments should raise ValueError | 
 | 1599 |         self.assertRaises(ValueError, subprocess.call, | 
 | 1600 |                           [sys.executable, "-c", | 
 | 1601 |                            "import sys; sys.exit(47)"], | 
 | 1602 |                           preexec_fn=lambda: 1) | 
 | 1603 |         self.assertRaises(ValueError, subprocess.call, | 
 | 1604 |                           [sys.executable, "-c", | 
 | 1605 |                            "import sys; sys.exit(47)"], | 
 | 1606 |                           stdout=subprocess.PIPE, | 
 | 1607 |                           close_fds=True) | 
 | 1608 |  | 
 | 1609 |     def test_close_fds(self): | 
 | 1610 |         # close file descriptors | 
 | 1611 |         rc = subprocess.call([sys.executable, "-c", | 
 | 1612 |                               "import sys; sys.exit(47)"], | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1613 |                               close_fds=True) | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1614 |         self.assertEqual(rc, 47) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1615 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1616 |     def test_shell_sequence(self): | 
 | 1617 |         # Run command through the shell (sequence) | 
 | 1618 |         newenv = os.environ.copy() | 
 | 1619 |         newenv["FRUIT"] = "physalis" | 
 | 1620 |         p = subprocess.Popen(["set"], shell=1, | 
 | 1621 |                              stdout=subprocess.PIPE, | 
 | 1622 |                              env=newenv) | 
| Brian Curtin | 19a5379 | 2010-11-05 17:09:05 +0000 | [diff] [blame] | 1623 |         self.addCleanup(p.stdout.close) | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1624 |         self.assertIn(b"physalis", p.stdout.read()) | 
| Guido van Rossum | e7ba495 | 2007-06-06 23:52:48 +0000 | [diff] [blame] | 1625 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1626 |     def test_shell_string(self): | 
 | 1627 |         # Run command through the shell (string) | 
 | 1628 |         newenv = os.environ.copy() | 
 | 1629 |         newenv["FRUIT"] = "physalis" | 
 | 1630 |         p = subprocess.Popen("set", shell=1, | 
 | 1631 |                              stdout=subprocess.PIPE, | 
 | 1632 |                              env=newenv) | 
| Brian Curtin | 19a5379 | 2010-11-05 17:09:05 +0000 | [diff] [blame] | 1633 |         self.addCleanup(p.stdout.close) | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1634 |         self.assertIn(b"physalis", p.stdout.read()) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1635 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1636 |     def test_call_string(self): | 
 | 1637 |         # call() function with string argument on Windows | 
 | 1638 |         rc = subprocess.call(sys.executable + | 
 | 1639 |                              ' -c "import sys; sys.exit(47)"') | 
 | 1640 |         self.assertEqual(rc, 47) | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1641 |  | 
| Florent Xicluna | 4886d24 | 2010-03-08 13:27:26 +0000 | [diff] [blame] | 1642 |     def _kill_process(self, method, *args): | 
 | 1643 |         # Some win32 buildbot raises EOFError if stdin is inherited | 
| Antoine Pitrou | a4024e2 | 2010-09-24 18:57:01 +0000 | [diff] [blame] | 1644 |         p = subprocess.Popen([sys.executable, "-c", """if 1: | 
 | 1645 |                              import sys, time | 
 | 1646 |                              sys.stdout.write('x\\n') | 
 | 1647 |                              sys.stdout.flush() | 
 | 1648 |                              time.sleep(30) | 
 | 1649 |                              """], | 
 | 1650 |                              stdin=subprocess.PIPE, | 
 | 1651 |                              stdout=subprocess.PIPE, | 
 | 1652 |                              stderr=subprocess.PIPE) | 
| Brian Curtin | 19a5379 | 2010-11-05 17:09:05 +0000 | [diff] [blame] | 1653 |         self.addCleanup(p.stdout.close) | 
 | 1654 |         self.addCleanup(p.stderr.close) | 
 | 1655 |         self.addCleanup(p.stdin.close) | 
| Antoine Pitrou | a4024e2 | 2010-09-24 18:57:01 +0000 | [diff] [blame] | 1656 |         # Wait for the interpreter to be completely initialized before | 
 | 1657 |         # sending any signal. | 
 | 1658 |         p.stdout.read(1) | 
 | 1659 |         getattr(p, method)(*args) | 
| Florent Xicluna | c049d87 | 2010-03-27 22:47:23 +0000 | [diff] [blame] | 1660 |         _, stderr = p.communicate() | 
 | 1661 |         self.assertStderrEqual(stderr, b'') | 
| Antoine Pitrou | a4024e2 | 2010-09-24 18:57:01 +0000 | [diff] [blame] | 1662 |         returncode = p.wait() | 
| Florent Xicluna | 4886d24 | 2010-03-08 13:27:26 +0000 | [diff] [blame] | 1663 |         self.assertNotEqual(returncode, 0) | 
 | 1664 |  | 
 | 1665 |     def test_send_signal(self): | 
 | 1666 |         self._kill_process('send_signal', signal.SIGTERM) | 
| Christian Heimes | a342c01 | 2008-04-20 21:01:16 +0000 | [diff] [blame] | 1667 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1668 |     def test_kill(self): | 
| Florent Xicluna | 4886d24 | 2010-03-08 13:27:26 +0000 | [diff] [blame] | 1669 |         self._kill_process('kill') | 
| Christian Heimes | a342c01 | 2008-04-20 21:01:16 +0000 | [diff] [blame] | 1670 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1671 |     def test_terminate(self): | 
| Florent Xicluna | 4886d24 | 2010-03-08 13:27:26 +0000 | [diff] [blame] | 1672 |         self._kill_process('terminate') | 
| Christian Heimes | a342c01 | 2008-04-20 21:01:16 +0000 | [diff] [blame] | 1673 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1674 |  | 
| Brett Cannon | a23810f | 2008-05-26 19:04:21 +0000 | [diff] [blame] | 1675 | # The module says: | 
 | 1676 | #   "NB This only works (and is only relevant) for UNIX." | 
 | 1677 | # | 
 | 1678 | # Actually, getoutput should work on any platform with an os.popen, but | 
 | 1679 | # I'll take the comment as given, and skip this suite. | 
| Florent Xicluna | f0cbd82 | 2010-03-04 21:50:56 +0000 | [diff] [blame] | 1680 | @unittest.skipUnless(os.name == 'posix', "only relevant for UNIX") | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1681 | class CommandTests(unittest.TestCase): | 
 | 1682 |     def test_getoutput(self): | 
 | 1683 |         self.assertEqual(subprocess.getoutput('echo xyzzy'), 'xyzzy') | 
 | 1684 |         self.assertEqual(subprocess.getstatusoutput('echo xyzzy'), | 
 | 1685 |                          (0, 'xyzzy')) | 
| Brett Cannon | a23810f | 2008-05-26 19:04:21 +0000 | [diff] [blame] | 1686 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1687 |         # we use mkdtemp in the next line to create an empty directory | 
 | 1688 |         # under our exclusive control; from that, we can invent a pathname | 
 | 1689 |         # that we _know_ won't exist.  This is guaranteed to fail. | 
 | 1690 |         dir = None | 
 | 1691 |         try: | 
 | 1692 |             dir = tempfile.mkdtemp() | 
 | 1693 |             name = os.path.join(dir, "foo") | 
| Brett Cannon | a23810f | 2008-05-26 19:04:21 +0000 | [diff] [blame] | 1694 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1695 |             status, output = subprocess.getstatusoutput('cat ' + name) | 
 | 1696 |             self.assertNotEqual(status, 0) | 
 | 1697 |         finally: | 
 | 1698 |             if dir is not None: | 
 | 1699 |                 os.rmdir(dir) | 
| Brett Cannon | a23810f | 2008-05-26 19:04:21 +0000 | [diff] [blame] | 1700 |  | 
| Gregory P. Smith | d06fa47 | 2009-07-04 02:46:54 +0000 | [diff] [blame] | 1701 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1702 | @unittest.skipUnless(getattr(subprocess, '_has_poll', False), | 
 | 1703 |                      "poll system call not supported") | 
 | 1704 | class ProcessTestCaseNoPoll(ProcessTestCase): | 
 | 1705 |     def setUp(self): | 
 | 1706 |         subprocess._has_poll = False | 
 | 1707 |         ProcessTestCase.setUp(self) | 
| Gregory P. Smith | d06fa47 | 2009-07-04 02:46:54 +0000 | [diff] [blame] | 1708 |  | 
| Florent Xicluna | b1e94e8 | 2010-02-27 22:12:37 +0000 | [diff] [blame] | 1709 |     def tearDown(self): | 
 | 1710 |         subprocess._has_poll = True | 
 | 1711 |         ProcessTestCase.tearDown(self) | 
| Gregory P. Smith | d06fa47 | 2009-07-04 02:46:54 +0000 | [diff] [blame] | 1712 |  | 
 | 1713 |  | 
| Gregory P. Smith | a59c59f | 2010-03-01 00:17:40 +0000 | [diff] [blame] | 1714 | class HelperFunctionTests(unittest.TestCase): | 
| Gregory P. Smith | af6d3b8 | 2010-03-01 02:56:44 +0000 | [diff] [blame] | 1715 |     @unittest.skipIf(mswindows, "errno and EINTR make no sense on windows") | 
| Gregory P. Smith | a59c59f | 2010-03-01 00:17:40 +0000 | [diff] [blame] | 1716 |     def test_eintr_retry_call(self): | 
 | 1717 |         record_calls = [] | 
 | 1718 |         def fake_os_func(*args): | 
 | 1719 |             record_calls.append(args) | 
 | 1720 |             if len(record_calls) == 2: | 
 | 1721 |                 raise OSError(errno.EINTR, "fake interrupted system call") | 
 | 1722 |             return tuple(reversed(args)) | 
 | 1723 |  | 
 | 1724 |         self.assertEqual((999, 256), | 
 | 1725 |                          subprocess._eintr_retry_call(fake_os_func, 256, 999)) | 
 | 1726 |         self.assertEqual([(256, 999)], record_calls) | 
 | 1727 |         # This time there will be an EINTR so it will loop once. | 
 | 1728 |         self.assertEqual((666,), | 
 | 1729 |                          subprocess._eintr_retry_call(fake_os_func, 666)) | 
 | 1730 |         self.assertEqual([(256, 999), (666,), (666,)], record_calls) | 
 | 1731 |  | 
 | 1732 |  | 
| Tim Golden | 126c296 | 2010-08-11 14:20:40 +0000 | [diff] [blame] | 1733 | @unittest.skipUnless(mswindows, "Windows-specific tests") | 
 | 1734 | class CommandsWithSpaces (BaseTestCase): | 
 | 1735 |  | 
 | 1736 |     def setUp(self): | 
 | 1737 |         super().setUp() | 
 | 1738 |         f, fname = mkstemp(".py", "te st") | 
 | 1739 |         self.fname = fname.lower () | 
 | 1740 |         os.write(f, b"import sys;" | 
 | 1741 |                     b"sys.stdout.write('%d %s' % (len(sys.argv), [a.lower () for a in sys.argv]))" | 
 | 1742 |         ) | 
 | 1743 |         os.close(f) | 
 | 1744 |  | 
 | 1745 |     def tearDown(self): | 
 | 1746 |         os.remove(self.fname) | 
 | 1747 |         super().tearDown() | 
 | 1748 |  | 
 | 1749 |     def with_spaces(self, *args, **kwargs): | 
 | 1750 |         kwargs['stdout'] = subprocess.PIPE | 
 | 1751 |         p = subprocess.Popen(*args, **kwargs) | 
| Brian Curtin | 19a5379 | 2010-11-05 17:09:05 +0000 | [diff] [blame] | 1752 |         self.addCleanup(p.stdout.close) | 
| Tim Golden | 126c296 | 2010-08-11 14:20:40 +0000 | [diff] [blame] | 1753 |         self.assertEqual( | 
 | 1754 |           p.stdout.read ().decode("mbcs"), | 
 | 1755 |           "2 [%r, 'ab cd']" % self.fname | 
 | 1756 |         ) | 
 | 1757 |  | 
 | 1758 |     def test_shell_string_with_spaces(self): | 
 | 1759 |         # call() function with string argument with spaces on Windows | 
| Brian Curtin | d835cf1 | 2010-08-13 20:42:57 +0000 | [diff] [blame] | 1760 |         self.with_spaces('"%s" "%s" "%s"' % (sys.executable, self.fname, | 
 | 1761 |                                              "ab cd"), shell=1) | 
| Tim Golden | 126c296 | 2010-08-11 14:20:40 +0000 | [diff] [blame] | 1762 |  | 
 | 1763 |     def test_shell_sequence_with_spaces(self): | 
 | 1764 |         # call() function with sequence argument with spaces on Windows | 
| Brian Curtin | d835cf1 | 2010-08-13 20:42:57 +0000 | [diff] [blame] | 1765 |         self.with_spaces([sys.executable, self.fname, "ab cd"], shell=1) | 
| Tim Golden | 126c296 | 2010-08-11 14:20:40 +0000 | [diff] [blame] | 1766 |  | 
 | 1767 |     def test_noshell_string_with_spaces(self): | 
 | 1768 |         # call() function with string argument with spaces on Windows | 
 | 1769 |         self.with_spaces('"%s" "%s" "%s"' % (sys.executable, self.fname, | 
 | 1770 |                              "ab cd")) | 
 | 1771 |  | 
 | 1772 |     def test_noshell_sequence_with_spaces(self): | 
 | 1773 |         # call() function with sequence argument with spaces on Windows | 
 | 1774 |         self.with_spaces([sys.executable, self.fname, "ab cd"]) | 
 | 1775 |  | 
| Brian Curtin | 79cdb66 | 2010-12-03 02:46:02 +0000 | [diff] [blame] | 1776 |  | 
| Georg Brandl | a86b262 | 2012-02-20 21:34:57 +0100 | [diff] [blame] | 1777 | class ContextManagerTests(BaseTestCase): | 
| Brian Curtin | 79cdb66 | 2010-12-03 02:46:02 +0000 | [diff] [blame] | 1778 |  | 
 | 1779 |     def test_pipe(self): | 
 | 1780 |         with subprocess.Popen([sys.executable, "-c", | 
 | 1781 |                                "import sys;" | 
 | 1782 |                                "sys.stdout.write('stdout');" | 
 | 1783 |                                "sys.stderr.write('stderr');"], | 
 | 1784 |                               stdout=subprocess.PIPE, | 
 | 1785 |                               stderr=subprocess.PIPE) as proc: | 
 | 1786 |             self.assertEqual(proc.stdout.read(), b"stdout") | 
 | 1787 |             self.assertStderrEqual(proc.stderr.read(), b"stderr") | 
 | 1788 |  | 
 | 1789 |         self.assertTrue(proc.stdout.closed) | 
 | 1790 |         self.assertTrue(proc.stderr.closed) | 
 | 1791 |  | 
 | 1792 |     def test_returncode(self): | 
 | 1793 |         with subprocess.Popen([sys.executable, "-c", | 
 | 1794 |                                "import sys; sys.exit(100)"]) as proc: | 
| Gregory P. Smith | 6b65745 | 2011-05-11 21:42:08 -0700 | [diff] [blame] | 1795 |             pass | 
 | 1796 |         # __exit__ calls wait(), so the returncode should be set | 
| Brian Curtin | 79cdb66 | 2010-12-03 02:46:02 +0000 | [diff] [blame] | 1797 |         self.assertEqual(proc.returncode, 100) | 
 | 1798 |  | 
 | 1799 |     def test_communicate_stdin(self): | 
 | 1800 |         with subprocess.Popen([sys.executable, "-c", | 
 | 1801 |                               "import sys;" | 
 | 1802 |                               "sys.exit(sys.stdin.read() == 'context')"], | 
 | 1803 |                              stdin=subprocess.PIPE) as proc: | 
 | 1804 |             proc.communicate(b"context") | 
 | 1805 |             self.assertEqual(proc.returncode, 1) | 
 | 1806 |  | 
 | 1807 |     def test_invalid_args(self): | 
 | 1808 |         with self.assertRaises(EnvironmentError) as c: | 
 | 1809 |             with subprocess.Popen(['nonexisting_i_hope'], | 
 | 1810 |                                   stdout=subprocess.PIPE, | 
 | 1811 |                                   stderr=subprocess.PIPE) as proc: | 
 | 1812 |                 pass | 
 | 1813 |  | 
 | 1814 |             if c.exception.errno != errno.ENOENT:  # ignore "no such file" | 
 | 1815 |                 raise c.exception | 
 | 1816 |  | 
 | 1817 |  | 
| Gregory P. Smith | 3b4652e | 2011-03-15 15:43:39 -0400 | [diff] [blame] | 1818 | def test_main(): | 
 | 1819 |     unit_tests = (ProcessTestCase, | 
 | 1820 |                   POSIXProcessTestCase, | 
 | 1821 |                   Win32ProcessTestCase, | 
| Gregory P. Smith | 3b4652e | 2011-03-15 15:43:39 -0400 | [diff] [blame] | 1822 |                   CommandTests, | 
 | 1823 |                   ProcessTestCaseNoPoll, | 
 | 1824 |                   HelperFunctionTests, | 
 | 1825 |                   CommandsWithSpaces, | 
| Antoine Pitrou | ab85ff3 | 2011-07-23 22:03:45 +0200 | [diff] [blame] | 1826 |                   ContextManagerTests, | 
 | 1827 |                   ) | 
| Gregory P. Smith | 3b4652e | 2011-03-15 15:43:39 -0400 | [diff] [blame] | 1828 |  | 
 | 1829 |     support.run_unittest(*unit_tests) | 
 | 1830 |     support.reap_children() | 
 | 1831 |  | 
| Fredrik Lundh | 5b3687d | 2004-10-12 15:26:28 +0000 | [diff] [blame] | 1832 | if __name__ == "__main__": | 
| Gregory P. Smith | 112bb3a | 2011-03-15 14:55:17 -0400 | [diff] [blame] | 1833 |     unittest.main() |