| # 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) |
