1 # Copyright (C) 2011 Apple Inc. All rights reserved.
3 # Redistribution and use in source and binary forms, with or without
4 # modification, are permitted provided that the following conditions
6 # 1. Redistributions of source code must retain the above copyright
7 # notice, this list of conditions and the following disclaimer.
8 # 2. Redistributions in binary form must reproduce the above copyright
9 # notice, this list of conditions and the following disclaimer in the
10 # documentation and/or other materials provided with the distribution.
12 # THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' AND
13 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
14 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
15 # DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS BE LIABLE FOR
16 # ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
17 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
18 # SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
19 # CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
20 # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
21 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 from webkitpy.common.system.outputcapture import OutputCapture
29 _log = logging.getLogger(__name__)
32 class OutputCaptureTest(unittest.TestCase):
34 self.output = OutputCapture()
36 def log_all_levels(self):
40 _log.critical('CRITICAL')
42 def assertLogged(self, expected_logs):
43 actual_stdout, actual_stderr, actual_logs = self.output.restore_output()
44 self.assertEqual('', actual_stdout)
45 self.assertEqual('', actual_stderr)
46 self.assertMultiLineEqual(expected_logs, actual_logs)
48 def test_initial_log_level(self):
49 self.output.capture_output()
51 self.assertLogged('INFO\nWARN\nERROR\nCRITICAL\n')
53 def test_set_log_level(self):
54 self.output.set_log_level(logging.ERROR)
55 self.output.capture_output()
57 self.output.set_log_level(logging.WARN)
59 self.assertLogged('ERROR\nCRITICAL\nWARN\nERROR\nCRITICAL\n')