Make MockExecutive behave more like MockExecutive2.

third_party/WebKit/Tools/Scripts/webkitpy/common/system/executive_mock.py

 # 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
@@ skipping 49 matching lines @@
 # FIXME: This should be unified with MockExecutive2 (http://crbug.com/626115).
 class MockExecutive(object):
     PIPE = "MOCK PIPE"
     STDOUT = "MOCK STDOUT"
 
     @staticmethod
     def ignore_error(error):
         pass
 
     def __init__(self, should_log=False, should_throw=False,
-                 should_throw_when_run=None, should_return_zero_when_run=None):
+                 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
-        self._should_throw_when_run = should_throw_when_run or set()
-        self._should_return_zero_when_run = should_return_zero_when_run or set()
         # 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 = []
-        self._output = "MOCK output of child process"
-        self._proc = None
 
     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=unused-argument
-                    timeout_seconds=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._should_throw_when_run.intersection(args):
-            raise ScriptError("Exception for %s" % args, output="MOCK command output")
-
+        if self._exception:
+            raise self._exception  # pylint: disable=raising-bad-type
         if self._should_throw:
             raise ScriptError("MOCK ScriptError", output=self._output)
 
-        if return_exit_code and self._should_return_zero_when_run.intersection(args):
-            return 0
+        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 popen(self, args, cwd=None, env=None, **kwargs):
+    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, **kwargs):
+    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:
@@ skipping 41 matching lines @@
         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