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