| Index: third_party/WebKit/Tools/Scripts/webkitpy/common/system/outputcapture.py
|
| diff --git a/third_party/WebKit/Tools/Scripts/webkitpy/common/system/outputcapture.py b/third_party/WebKit/Tools/Scripts/webkitpy/common/system/outputcapture.py
|
| deleted file mode 100644
|
| index cd4c00ecbe409caa035a315810406ab8a6cab048..0000000000000000000000000000000000000000
|
| --- a/third_party/WebKit/Tools/Scripts/webkitpy/common/system/outputcapture.py
|
| +++ /dev/null
|
| @@ -1,102 +0,0 @@
|
| -# Copyright (c) 2009, 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.
|
| -#
|
| -# Class for unittest support. Used for capturing stderr/stdout.
|
| -
|
| -import logging
|
| -import sys
|
| -
|
| -from StringIO import StringIO
|
| -
|
| -
|
| -class OutputCapture(object):
|
| -
|
| - def __init__(self):
|
| - self.saved_outputs = dict()
|
| - self._log_level = logging.INFO
|
| -
|
| - def set_log_level(self, log_level):
|
| - self._log_level = log_level
|
| - if hasattr(self, '_logs_handler'):
|
| - self._logs_handler.setLevel(self._log_level)
|
| -
|
| - def _capture_output_with_name(self, output_name):
|
| - stream = getattr(sys, output_name)
|
| - captured_output = StringIO()
|
| - self.saved_outputs[output_name] = stream
|
| - setattr(sys, output_name, captured_output)
|
| - return captured_output
|
| -
|
| - def _restore_output_with_name(self, output_name):
|
| - captured_output = getattr(sys, output_name).getvalue()
|
| - setattr(sys, output_name, self.saved_outputs[output_name])
|
| - del self.saved_outputs[output_name]
|
| - return captured_output
|
| -
|
| - def capture_output(self):
|
| - self._logs = StringIO()
|
| - self._logs_handler = logging.StreamHandler(self._logs)
|
| - self._logs_handler.setLevel(self._log_level)
|
| - self._logger = logging.getLogger()
|
| - self._orig_log_level = self._logger.level
|
| - self._logger.addHandler(self._logs_handler)
|
| - self._logger.setLevel(min(self._log_level, self._orig_log_level))
|
| - return (self._capture_output_with_name("stdout"), self._capture_output_with_name("stderr"))
|
| -
|
| - def restore_output(self):
|
| - self._logger.removeHandler(self._logs_handler)
|
| - self._logger.setLevel(self._orig_log_level)
|
| - self._logs_handler.flush()
|
| - self._logs.flush()
|
| - logs_string = self._logs.getvalue()
|
| - delattr(self, '_logs_handler')
|
| - delattr(self, '_logs')
|
| - return (self._restore_output_with_name("stdout"), self._restore_output_with_name("stderr"), logs_string)
|
| -
|
| - def assert_outputs(self, testcase, function, args=[], kwargs={}, expected_stdout="",
|
| - expected_stderr="", expected_exception=None, expected_logs=None):
|
| - self.capture_output()
|
| - try:
|
| - if expected_exception:
|
| - return_value = testcase.assertRaises(expected_exception, function, *args, **kwargs)
|
| - else:
|
| - return_value = function(*args, **kwargs)
|
| - finally:
|
| - (stdout_string, stderr_string, logs_string) = self.restore_output()
|
| -
|
| - if hasattr(testcase, 'assertMultiLineEqual'):
|
| - testassert = testcase.assertMultiLineEqual
|
| - else:
|
| - testassert = testcase.assertEqual
|
| -
|
| - testassert(stdout_string, expected_stdout)
|
| - testassert(stderr_string, expected_stderr)
|
| - if expected_logs is not None:
|
| - testassert(logs_string, expected_logs)
|
| - # This is a little strange, but I don't know where else to return this information.
|
| - return return_value
|
|
|