OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright (c) 2011 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2011 The Chromium Authors. All rights reserved. |
3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
5 | 5 |
6 """Unit tests for subprocess2.py.""" | 6 """Unit tests for subprocess2.py.""" |
7 | 7 |
8 import logging | 8 import logging |
9 import optparse | 9 import optparse |
10 import os | 10 import os |
11 import sys | 11 import sys |
12 import time | 12 import time |
13 import unittest | 13 import unittest |
14 | 14 |
15 try: | 15 try: |
16 import fcntl | 16 import fcntl |
17 except ImportError: | 17 except ImportError: |
18 fcntl = None | 18 fcntl = None |
19 | 19 |
20 ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) | 20 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) |
21 sys.path.insert(0, ROOT_DIR) | |
22 | 21 |
23 import subprocess2 | 22 import subprocess2 |
24 | 23 |
25 # Method could be a function | |
26 # pylint: disable=R0201 | |
27 | |
28 | 24 |
29 def convert_to_crlf(string): | 25 def convert_to_crlf(string): |
30 """Unconditionally convert LF to CRLF.""" | 26 """Unconditionally convert LF to CRLF.""" |
31 return string.replace('\n', '\r\n') | 27 return string.replace('\n', '\r\n') |
32 | 28 |
33 | 29 |
34 def convert_to_cr(string): | 30 def convert_to_cr(string): |
35 """Unconditionally convert LF to CR.""" | 31 """Unconditionally convert LF to CR.""" |
36 return string.replace('\n', '\r') | 32 return string.replace('\n', '\r') |
37 | 33 |
(...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
77 | 73 |
78 @staticmethod | 74 @staticmethod |
79 def _fake_Popen(): | 75 def _fake_Popen(): |
80 results = {} | 76 results = {} |
81 class fake_Popen(object): | 77 class fake_Popen(object): |
82 returncode = -8 | 78 returncode = -8 |
83 def __init__(self, args, **kwargs): | 79 def __init__(self, args, **kwargs): |
84 assert not results | 80 assert not results |
85 results.update(kwargs) | 81 results.update(kwargs) |
86 results['args'] = args | 82 results['args'] = args |
87 def communicate(self): | 83 @staticmethod |
| 84 def communicate(): |
88 return None, None | 85 return None, None |
89 subprocess2.Popen = fake_Popen | 86 subprocess2.Popen = fake_Popen |
90 return results | 87 return results |
91 | 88 |
92 @staticmethod | 89 @staticmethod |
93 def _fake_subprocess_Popen(): | 90 def _fake_subprocess_Popen(): |
94 results = {} | 91 results = {} |
95 class fake_Popen(object): | 92 class fake_Popen(object): |
96 returncode = -8 | 93 returncode = -8 |
97 def __init__(self, args, **kwargs): | 94 def __init__(self, args, **kwargs): |
98 assert not results | 95 assert not results |
99 results.update(kwargs) | 96 results.update(kwargs) |
100 results['args'] = args | 97 results['args'] = args |
101 def communicate(self): | 98 @staticmethod |
| 99 def communicate(): |
102 return None, None | 100 return None, None |
103 subprocess2.subprocess.Popen = fake_Popen | 101 subprocess2.subprocess.Popen = fake_Popen |
104 return results | 102 return results |
105 | 103 |
106 def test_check_call_defaults(self): | 104 def test_check_call_defaults(self): |
107 results = self._fake_communicate() | 105 results = self._fake_communicate() |
108 self.assertEquals( | 106 self.assertEquals( |
109 ['stdout', 'stderr'], subprocess2.check_call_out(['foo'], a=True)) | 107 ['stdout', 'stderr'], subprocess2.check_call_out(['foo'], a=True)) |
110 expected = { | 108 expected = { |
111 'args': ['foo'], | 109 'args': ['foo'], |
(...skipping 338 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
450 return options.return_value | 448 return options.return_value |
451 | 449 |
452 | 450 |
453 if __name__ == '__main__': | 451 if __name__ == '__main__': |
454 logging.basicConfig(level= | 452 logging.basicConfig(level= |
455 [logging.WARNING, logging.INFO, logging.DEBUG][ | 453 [logging.WARNING, logging.INFO, logging.DEBUG][ |
456 min(2, sys.argv.count('-v'))]) | 454 min(2, sys.argv.count('-v'))]) |
457 if len(sys.argv) > 1 and sys.argv[1] == '--child': | 455 if len(sys.argv) > 1 and sys.argv[1] == '--child': |
458 sys.exit(child_main(sys.argv[2:])) | 456 sys.exit(child_main(sys.argv[2:])) |
459 unittest.main() | 457 unittest.main() |
OLD | NEW |