Lib/test/test_popen.py

   1 #! /usr/bin/env python
   2 """Basic tests for os.popen()
   3
   4   Particularly useful for platforms that fake popen.
   5 """
   6
   7 import unittest
   8 from test import test_support
   9 import os, sys
  10
  11 # Test that command-lines get down as we expect.
  12 # To do this we execute:
  13 #    python -c "import sys;print sys.argv" {rest_of_commandline}
  14 # This results in Python being spawned and printing the sys.argv list.
  15 # We can then eval() the result of this, and see what each argv was.
  16 python = sys.executable
  17 if ' ' in python:
  18     python = '"' + python + '"'     # quote embedded space for cmdline
  19
  20 class PopenTest(unittest.TestCase):
  21     def _do_test_commandline(self, cmdline, expected):
  22         cmd = '%s -c "import sys;print sys.argv" %s' % (python, cmdline)
  23         data = os.popen(cmd).read()
  24         got = eval(data)[1:] # strip off argv[0]
  25         self.assertEqual(got, expected)
  26
  27     def test_popen(self):
  28         self.assertRaises(TypeError, os.popen)
  29         self._do_test_commandline(
  30             "foo bar",
  31             ["foo", "bar"]
  32         )
  33         self._do_test_commandline(
  34             'foo "spam and eggs" "silly walk"',
  35             ["foo", "spam and eggs", "silly walk"]
  36         )
  37         self._do_test_commandline(
  38             'foo "a \\"quoted\\" arg" bar',
  39             ["foo", 'a "quoted" arg', "bar"]
  40         )
  41         test_support.reap_children()
  42
  43     def test_return_code(self):
  44         self.assertEqual(os.popen("exit 0").close(), None)
  45         if os.name == 'nt':
  46             self.assertEqual(os.popen("exit 42").close(), 42)
  47         else:
  48             self.assertEqual(os.popen("exit 42").close(), 42 << 8)
  49
  50 def test_main():
  51     test_support.run_unittest(PopenTest)
  52
  53 if __name__ == "__main__":
  54     test_main()