| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970 |
- """Basic tests for os.popen()
- Particularly useful for platforms that fake popen.
- """
- import unittest
- from test import support
- import os, sys
- if not hasattr(os, 'popen'):
- raise unittest.SkipTest("need os.popen()")
- # Test that command-lines get down as we expect.
- # To do this we execute:
- # python -c "import sys;print(sys.argv)" {rest_of_commandline}
- # This results in Python being spawned and printing the sys.argv list.
- # We can then eval() the result of this, and see what each argv was.
- python = sys.executable
- if ' ' in python:
- python = '"' + python + '"' # quote embedded space for cmdline
- @support.requires_subprocess()
- class PopenTest(unittest.TestCase):
- def _do_test_commandline(self, cmdline, expected):
- cmd = '%s -c "import sys; print(sys.argv)" %s'
- cmd = cmd % (python, cmdline)
- with os.popen(cmd) as p:
- data = p.read()
- got = eval(data)[1:] # strip off argv[0]
- self.assertEqual(got, expected)
- def test_popen(self):
- self.assertRaises(TypeError, os.popen)
- self._do_test_commandline(
- "foo bar",
- ["foo", "bar"]
- )
- self._do_test_commandline(
- 'foo "spam and eggs" "silly walk"',
- ["foo", "spam and eggs", "silly walk"]
- )
- self._do_test_commandline(
- 'foo "a \\"quoted\\" arg" bar',
- ["foo", 'a "quoted" arg', "bar"]
- )
- support.reap_children()
- def test_return_code(self):
- self.assertEqual(os.popen("exit 0").close(), None)
- status = os.popen("exit 42").close()
- if os.name == 'nt':
- self.assertEqual(status, 42)
- else:
- self.assertEqual(os.waitstatus_to_exitcode(status), 42)
- def test_contextmanager(self):
- with os.popen("echo hello") as f:
- self.assertEqual(f.read(), "hello\n")
- def test_iterating(self):
- with os.popen("echo hello") as f:
- self.assertEqual(list(f), ["hello\n"])
- def test_keywords(self):
- with os.popen(cmd="exit 0", mode="w", buffering=-1):
- pass
- if __name__ == "__main__":
- unittest.main()
|
