1#! /usr/bin/env python
2"""Basic tests for os.popen()
3
4  Particularly useful for platforms that fake popen.
5"""
6
7import unittest
8from test import test_support
9import 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.
16python = sys.executable
17
18class 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
48def test_main():
49    test_support.run_unittest(PopenTest)
50
51if __name__ == "__main__":
52    test_main()
53