| OLD | NEW |
| (Empty) |
| 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 | |
| 23 import logging | |
| 24 | |
| 25 _log = logging.getLogger(__name__) | |
| 26 | |
| 27 | |
| 28 def skip_if(klass, condition, message=None, logger=None): | |
| 29 """Makes all test_* methods in a given class no-ops if the given condition | |
| 30 is False. Backported from Python 3.1+'s unittest.skipIf decorator.""" | |
| 31 if not logger: | |
| 32 logger = _log | |
| 33 if not condition: | |
| 34 return klass | |
| 35 for name in dir(klass): | |
| 36 attr = getattr(klass, name) | |
| 37 if not callable(attr): | |
| 38 continue | |
| 39 if not name.startswith('test_'): | |
| 40 continue | |
| 41 setattr(klass, name, _skipped_method(attr, message, logger)) | |
| 42 klass._printed_skipped_message = False | |
| 43 return klass | |
| 44 | |
| 45 | |
| 46 def _skipped_method(method, message, logger): | |
| 47 def _skip(*args): | |
| 48 if method.im_class._printed_skipped_message: | |
| 49 return | |
| 50 method.im_class._printed_skipped_message = True | |
| 51 logger.info('Skipping %s.%s: %s' % (method.__module__, method.im_class._
_name__, message)) | |
| 52 return _skip | |
| OLD | NEW |