Mark Hammond | e7fefbf | 2002-04-03 01:47:00 +0000 | [diff] [blame] | 1 | """Basic tests for os.popen() |
| 2 | |
| 3 | Particularly useful for platforms that fake popen. |
| 4 | """ |
| 5 | |
Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 6 | import unittest |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 7 | from test import support |
Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 8 | import os, sys |
Mark Hammond | e7fefbf | 2002-04-03 01:47:00 +0000 | [diff] [blame] | 9 | |
| 10 | # Test that command-lines get down as we expect. |
| 11 | # To do this we execute: |
Neal Norwitz | 752abd0 | 2008-05-13 04:55:24 +0000 | [diff] [blame] | 12 | # python -c "import sys;print(sys.argv)" {rest_of_commandline} |
Mark Hammond | e7fefbf | 2002-04-03 01:47:00 +0000 | [diff] [blame] | 13 | # This results in Python being spawned and printing the sys.argv list. |
| 14 | # We can then eval() the result of this, and see what each argv was. |
Tim Peters | 61cd0db | 2003-03-07 21:10:21 +0000 | [diff] [blame] | 15 | python = sys.executable |
| 16 | if ' ' in python: |
| 17 | python = '"' + python + '"' # quote embedded space for cmdline |
Mark Hammond | e7fefbf | 2002-04-03 01:47:00 +0000 | [diff] [blame] | 18 | |
Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 19 | class PopenTest(unittest.TestCase): |
Guido van Rossum | aa588c4 | 2007-06-15 03:35:38 +0000 | [diff] [blame] | 20 | |
Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 21 | def _do_test_commandline(self, cmdline, expected): |
Guido van Rossum | 96ca691 | 2007-06-15 03:49:03 +0000 | [diff] [blame] | 22 | cmd = '%s -c "import sys; print(sys.argv)" %s' |
Guido van Rossum | aa588c4 | 2007-06-15 03:35:38 +0000 | [diff] [blame] | 23 | cmd = cmd % (python, cmdline) |
Benjamin Peterson | aa7cec0 | 2010-10-31 01:35:53 +0000 | [diff] [blame] | 24 | with os.popen(cmd) as p: |
| 25 | data = p.read() |
Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 26 | got = eval(data)[1:] # strip off argv[0] |
| 27 | self.assertEqual(got, expected) |
Mark Hammond | e7fefbf | 2002-04-03 01:47:00 +0000 | [diff] [blame] | 28 | |
Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 29 | def test_popen(self): |
| 30 | self.assertRaises(TypeError, os.popen) |
| 31 | self._do_test_commandline( |
| 32 | "foo bar", |
| 33 | ["foo", "bar"] |
| 34 | ) |
| 35 | self._do_test_commandline( |
| 36 | 'foo "spam and eggs" "silly walk"', |
| 37 | ["foo", "spam and eggs", "silly walk"] |
| 38 | ) |
| 39 | self._do_test_commandline( |
| 40 | 'foo "a \\"quoted\\" arg" bar', |
| 41 | ["foo", 'a "quoted" arg', "bar"] |
| 42 | ) |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 43 | support.reap_children() |
Mark Hammond | e7fefbf | 2002-04-03 01:47:00 +0000 | [diff] [blame] | 44 | |
Amaury Forgeot d'Arc | 97e5f28 | 2009-07-11 09:35:13 +0000 | [diff] [blame] | 45 | def test_return_code(self): |
| 46 | self.assertEqual(os.popen("exit 0").close(), None) |
| 47 | if os.name == 'nt': |
| 48 | self.assertEqual(os.popen("exit 42").close(), 42) |
| 49 | else: |
| 50 | self.assertEqual(os.popen("exit 42").close(), 42 << 8) |
| 51 | |
Antoine Pitrou | ac62535 | 2009-12-09 00:01:27 +0000 | [diff] [blame] | 52 | def test_contextmanager(self): |
| 53 | with os.popen("echo hello") as f: |
| 54 | self.assertEqual(f.read(), "hello\n") |
| 55 | |
| 56 | def test_iterating(self): |
| 57 | with os.popen("echo hello") as f: |
| 58 | self.assertEqual(list(f), ["hello\n"]) |
| 59 | |
Martin Panter | bf19d16 | 2015-09-09 01:01:13 +0000 | [diff] [blame] | 60 | def test_keywords(self): |
| 61 | with os.popen(cmd="exit 0", mode="w", buffering=-1): |
| 62 | pass |
| 63 | |
Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 64 | if __name__ == "__main__": |
Zachary Ware | 38c707e | 2015-04-13 15:00:43 -0500 | [diff] [blame] | 65 | unittest.main() |