skip.py 2.2 KB

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