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() 53
