| # Copyright (C) 2011 Google Inc. All rights reserved. | |
| # | |
| # Redistribution and use in source and binary forms, with or without | |
| # modification, are permitted provided that the following conditions are | |
| # met: | |
| # | |
| # * Redistributions of source code must retain the above copyright | |
| # notice, this list of conditions and the following disclaimer. | |
| # * Redistributions in binary form must reproduce the above | |
| # copyright notice, this list of conditions and the following disclaimer | |
| # in the documentation and/or other materials provided with the | |
| # distribution. | |
| # * Neither the name of Google Inc. nor the names of its | |
| # contributors may be used to endorse or promote products derived from | |
| # this software without specific prior written permission. | |
| # | |
| # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS | |
| # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT | |
| # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR | |
| # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT | |
| # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | |
| # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT | |
| # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | |
| # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | |
| # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | |
| # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | |
| # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
| import logging | |
| import os | |
| import StringIO | |
| from webkitpy.common.system.executive import ScriptError | |
| _log = logging.getLogger(__name__) | |
| class MockProcess(object): | |
| def __init__(self, stdout='MOCK STDOUT\n', stderr=''): | |
| self.pid = 42 | |
| self.stdout = StringIO.StringIO(stdout) | |
| self.stderr = StringIO.StringIO(stderr) | |
| self.stdin = StringIO.StringIO() | |
| self.returncode = 0 | |
| def wait(self): | |
| return | |
| def poll(self): | |
| # Consider the process completed when all the stdout and stderr has been read. | |
| if self.stdout.len != self.stdout.tell() or self.stderr.len != self.stderr.tell(): | |
| return None | |
| return self.returncode | |
| def communicate(self, *_): | |
| return (self.stdout.getvalue(), self.stderr.getvalue()) | |
| class MockExecutive(object): | |
| PIPE = 'MOCK PIPE' | |
| STDOUT = 'MOCK STDOUT' | |
| DEVNULL = 'MOCK_DEVNULL' | |
| @staticmethod | |
| def ignore_error(error): | |
| pass | |
| def __init__(self, should_log=False, should_throw=False, | |
| output='MOCK output of child process', stderr='', | |
| exit_code=0, exception=None, run_command_fn=None): | |
| self._should_log = should_log | |
| self._should_throw = should_throw | |
| # FIXME: Once executive wraps os.getpid() we can just use a static pid for "this" process. | |
| self._running_pids = {'test-webkitpy': os.getpid()} | |
| self._output = output | |
| self._stderr = stderr | |
| self._exit_code = exit_code | |
| self._exception = exception | |
| self._run_command_fn = run_command_fn | |
| self._proc = None | |
| self.calls = [] | |
| def check_running_pid(self, pid): | |
| return pid in self._running_pids.values() | |
| def running_pids(self, process_name_filter): | |
| running_pids = [] | |
| for process_name, process_pid in self._running_pids.iteritems(): | |
| if process_name_filter(process_name): | |
| running_pids.append(process_pid) | |
| _log.info('MOCK running_pids: %s', running_pids) | |
| return running_pids | |
| def command_for_printing(self, args): | |
| string_args = map(unicode, args) | |
| return ' '.join(string_args) | |
| # The argument list should match Executive.run_command, even if | |
| # some arguments are not used. pylint: disable=unused-argument | |
| def run_command(self, | |
| args, | |
| cwd=None, | |
| input=None, # pylint: disable=redefined-builtin | |
| timeout_seconds=False, | |
| error_handler=None, | |
| return_exit_code=False, | |
| return_stderr=True, | |
| decode_output=False, | |
| env=None, | |
| debug_logging=False): | |
| self.calls.append(args) | |
| assert isinstance(args, list) or isinstance(args, tuple) | |
| if self._should_log: | |
| env_string = '' | |
| if env: | |
| env_string = ', env=%s' % env | |
| input_string = '' | |
| if input: | |
| input_string = ', input=%s' % input | |
| _log.info('MOCK run_command: %s, cwd=%s%s%s', args, cwd, env_string, input_string) | |
| if self._exception: | |
| raise self._exception # pylint: disable=raising-bad-type | |
| if self._should_throw: | |
| raise ScriptError('MOCK ScriptError', output=self._output) | |
| if self._run_command_fn: | |
| return self._run_command_fn(args) | |
| if return_exit_code: | |
| return self._exit_code | |
| if self._exit_code and error_handler: | |
| script_error = ScriptError(script_args=args, exit_code=self._exit_code, output=self._output) | |
| error_handler(script_error) | |
| if return_stderr: | |
| return self._output + self._stderr | |
| return self._output | |
| def cpu_count(self): | |
| return 2 | |
| def kill_all(self, process_name): | |
| pass | |
| def kill_process(self, pid): | |
| pass | |
| def interrupt(self, pid): | |
| pass | |
| def popen(self, args, cwd=None, env=None, **_): | |
| assert all(isinstance(arg, basestring) for arg in args) | |
| self.calls.append(args) | |
| if self._should_log: | |
| cwd_string = '' | |
| if cwd: | |
| cwd_string = ', cwd=%s' % cwd | |
| env_string = '' | |
| if env: | |
| env_string = ', env=%s' % env | |
| _log.info('MOCK popen: %s%s%s', args, cwd_string, env_string) | |
| if not self._proc: | |
| self._proc = MockProcess(self._output) | |
| return self._proc | |
| def call(self, args, **_): | |
| assert all(isinstance(arg, basestring) for arg in args) | |
| self.calls.append(args) | |
| _log.info('Mock call: %s', args) | |
| def run_in_parallel(self, commands): | |
| assert len(commands) | |
| num_previous_calls = len(self.calls) | |
| command_outputs = [] | |
| for cmd_line, cwd in commands: | |
| assert all(isinstance(arg, basestring) for arg in cmd_line) | |
| command_outputs.append([0, self.run_command(cmd_line, cwd=cwd), '']) | |
| new_calls = self.calls[num_previous_calls:] | |
| self.calls = self.calls[:num_previous_calls] | |
| self.calls.append(new_calls) | |
| return command_outputs | |
| def map(self, thunk, arglist, processes=None): | |
| return map(thunk, arglist) | |
| def process_dump(self): | |
| return [] | |
| def mock_git_commands(vals, strict=False): | |
| def run_fn(args): | |
| sub_command = args[1] | |
| if strict and sub_command not in vals: | |
| raise AssertionError('{} not found in sub-command list {}'.format( | |
| sub_command, vals)) | |
| return vals.get(sub_command, '') | |
| return MockExecutive(run_command_fn=run_fn) |