summaryrefslogtreecommitdiffstats
path: root/Tools/Scripts/webkitpy/common/system/outputcapture.py
diff options
context:
space:
mode:
Diffstat (limited to 'Tools/Scripts/webkitpy/common/system/outputcapture.py')
-rw-r--r--Tools/Scripts/webkitpy/common/system/outputcapture.py128
1 files changed, 0 insertions, 128 deletions
diff --git a/Tools/Scripts/webkitpy/common/system/outputcapture.py b/Tools/Scripts/webkitpy/common/system/outputcapture.py
deleted file mode 100644
index 893b5e528..000000000
--- a/Tools/Scripts/webkitpy/common/system/outputcapture.py
+++ /dev/null
@@ -1,128 +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 unittest # Don't use unittest2 here as the autoinstaller may not have it yet.
-import sys
-from StringIO import StringIO
-
-
-class OutputCapture(object):
- # By default we capture the output to a stream. Other modules may override
- # this function in order to do things like pass through the output. See
- # webkitpy.test.main for an example.
- @staticmethod
- def stream_wrapper(stream):
- return StringIO()
-
- 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 = self.stream_wrapper(stream)
- 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
-
-
-class OutputCaptureTestCaseBase(unittest.TestCase):
- maxDiff = None
-
- def setUp(self):
- unittest.TestCase.setUp(self)
- self.output_capture = OutputCapture()
- (self.__captured_stdout, self.__captured_stderr) = self.output_capture.capture_output()
-
- def tearDown(self):
- del self.__captured_stdout
- del self.__captured_stderr
- self.output_capture.restore_output()
- unittest.TestCase.tearDown(self)
-
- def assertStdout(self, expected_stdout):
- self.assertEqual(expected_stdout, self.__captured_stdout.getvalue())
-
- def assertStderr(self, expected_stderr):
- self.assertEqual(expected_stderr, self.__captured_stderr.getvalue())