1# Copyright (C) 2010 Apple Inc. All rights reserved. 2# 3# Redistribution and use in source and binary forms, with or without 4# modification, are permitted provided that the following conditions 5# are met: 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. 11# 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. 22 23import StringIO 24import logging 25import unittest 26 27from webkitpy.test.skip import skip_if 28 29 30class SkipTest(unittest.TestCase): 31 def setUp(self): 32 self.logger = logging.getLogger(__name__) 33 34 self.old_level = self.logger.level 35 self.logger.setLevel(logging.INFO) 36 37 self.old_propagate = self.logger.propagate 38 self.logger.propagate = False 39 40 self.log_stream = StringIO.StringIO() 41 self.handler = logging.StreamHandler(self.log_stream) 42 self.logger.addHandler(self.handler) 43 44 self.foo_was_called = False 45 46 def tearDown(self): 47 self.logger.removeHandler(self.handler) 48 self.propagate = self.old_propagate 49 self.logger.setLevel(self.old_level) 50 51 def create_fixture_class(self): 52 class TestSkipFixture(object): 53 def __init__(self, callback): 54 self.callback = callback 55 56 def test_foo(self): 57 self.callback() 58 59 return TestSkipFixture 60 61 def foo_callback(self): 62 self.foo_was_called = True 63 64 def test_skip_if_false(self): 65 klass = skip_if(self.create_fixture_class(), False, 'Should not see this message.', logger=self.logger) 66 klass(self.foo_callback).test_foo() 67 self.assertEqual(self.log_stream.getvalue(), '') 68 self.assertTrue(self.foo_was_called) 69 70 def test_skip_if_true(self): 71 klass = skip_if(self.create_fixture_class(), True, 'Should see this message.', logger=self.logger) 72 klass(self.foo_callback).test_foo() 73 self.assertEqual(self.log_stream.getvalue(), 'Skipping webkitpy.test.skip_unittest.TestSkipFixture: Should see this message.\n') 74 self.assertFalse(self.foo_was_called) 75 76if __name__ == '__main__': 77 unittest.main() 78