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
  18 class PopenTest(unittest.TestCase):
  19     def _do_test_commandline(self, cmdline, expected):
  20         cmd = '"%s" -c "import sys;print sys.argv" %s' % (python, cmdline)
  21         data = os.popen(cmd).read() + '\n'
  22         got = eval(data)[1:] # strip off argv[0]
  23         self.assertEqual(got, expected)
  24
  25     def test_popen(self):
  26         self.assertRaises(TypeError, os.popen)
  27         self._do_test_commandline(
  28             "foo bar",
  29             ["foo", "bar"]
  30         )
  31         self._do_test_commandline(
  32             'foo "spam and eggs" "silly walk"',
  33             ["foo", "spam and eggs", "silly walk"]
  34         )
  35         self._do_test_commandline(
  36             'foo "a \\"quoted\\" arg" bar',
  37             ["foo", 'a "quoted" arg', "bar"]
  38         )
  39         test_support.reap_children()
  40
  41     def test_return_code(self):
  42         self.assertEqual(os.popen("exit 0").close(), None)
  43         if os.name == 'nt':
  44             self.assertEqual(os.popen("exit 42").close(), 42)
  45         else:
  46             self.assertEqual(os.popen("exit 42").close(), 42 << 8)
  47
  48 def test_main():
  49     test_support.run_unittest(PopenTest)
  50
  51 if __name__ == "__main__":
  52     test_main()