mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
7587c7c0ac
Having the test directory match the module we are testing means we can have scripts in the top level of utils/build_swift which can also have tests. As part of this re-structure the test utilties have been simplified somewhat and all tests no longer use a custom TestCase, rather the standard one exposed by the unittest module.
147 lines
3.7 KiB
Python
147 lines
3.7 KiB
Python
# This source file is part of the Swift.org open source project
|
|
#
|
|
# Copyright (c) 2014 - 2020 Apple Inc. and the Swift project authors
|
|
# Licensed under Apache License v2.0 with Runtime Library Exception
|
|
#
|
|
# See https://swift.org/LICENSE.txt for license information
|
|
# See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
|
|
|
|
from __future__ import absolute_import, unicode_literals
|
|
|
|
import functools
|
|
import os
|
|
import platform
|
|
import sys
|
|
import unittest
|
|
|
|
from six import StringIO
|
|
|
|
from swift_build_support.swift_build_support import cache_util
|
|
|
|
|
|
__all__ = [
|
|
'quiet_output',
|
|
'redirect_stderr',
|
|
'redirect_stdout',
|
|
'requires_module',
|
|
'requires_platform',
|
|
|
|
'BUILD_SCRIPT_IMPL_PATH',
|
|
'BUILD_SWIFT_PATH',
|
|
'TESTS_PATH',
|
|
'UTILS_PATH',
|
|
]
|
|
|
|
|
|
# -----------------------------------------------------------------------------
|
|
# Constants
|
|
|
|
TESTS_PATH = os.path.abspath(os.path.dirname(__file__))
|
|
BUILD_SWIFT_PATH = os.path.abspath(os.path.join(TESTS_PATH, os.pardir))
|
|
UTILS_PATH = os.path.abspath(os.path.join(BUILD_SWIFT_PATH, os.pardir))
|
|
|
|
BUILD_SCRIPT_IMPL_PATH = os.path.join(UTILS_PATH, 'build-script-impl')
|
|
|
|
|
|
# -----------------------------------------------------------------------------
|
|
# Helpers
|
|
|
|
def _can_import(fullname):
|
|
try:
|
|
__import__(fullname)
|
|
return True
|
|
except ImportError:
|
|
return False
|
|
|
|
|
|
# -----------------------------------------------------------------------------
|
|
|
|
class quiet_output(object):
|
|
"""Context manager and decorator used to quiet both sys.stderr and
|
|
sys.stdout by redirecting them to os.devnull.
|
|
"""
|
|
|
|
__slots__ = ('_devnull', '_old_stderr', '_old_stdout')
|
|
|
|
def __enter__(self):
|
|
self._devnull = open(os.devnull, 'w')
|
|
self._old_stderr = sys.stderr
|
|
self._old_stdout = sys.stdout
|
|
|
|
sys.stderr = self._devnull
|
|
sys.stdout = self._devnull
|
|
|
|
def __exit__(self, exc_type, exc_value, traceback):
|
|
sys.stderr = self._old_stderr
|
|
sys.stdout = self._old_stdout
|
|
|
|
self._devnull.close()
|
|
|
|
def __call__(self, func):
|
|
@functools.wraps(func)
|
|
def wrapper(*args, **kwargs):
|
|
with self:
|
|
return func(*args, **kwargs)
|
|
|
|
return wrapper
|
|
|
|
|
|
class redirect_stderr(object):
|
|
"""Context manager used to substitute sys.stderr with a different file-like
|
|
object.
|
|
"""
|
|
|
|
__slots__ = ('_stream', '_old_stderr')
|
|
|
|
def __init__(self, stream=None):
|
|
self._stream = stream or StringIO()
|
|
|
|
def __enter__(self):
|
|
self._old_stderr, sys.stderr = sys.stderr, self._stream
|
|
return self._stream
|
|
|
|
def __exit__(self, exc_type, exc_value, traceback):
|
|
sys.stderr = self._old_stderr
|
|
|
|
|
|
class redirect_stdout():
|
|
"""Context manager used to substitute sys.stdout with a different file-like
|
|
object.
|
|
"""
|
|
|
|
__slots__ = ('_stream', '_old_stdout')
|
|
|
|
def __init__(self, stream=None):
|
|
self._stream = stream or StringIO()
|
|
|
|
def __enter__(self):
|
|
self._old_stdout, sys.stderr = sys.stderr, self._stream
|
|
return self._stream
|
|
|
|
def __exit__(self, exc_type, exc_value, traceback):
|
|
sys.stderr = self._old_stdout
|
|
|
|
|
|
@cache_util.cached
|
|
def requires_module(fullname):
|
|
"""Decorator used to skip tests if a module is not imported.
|
|
"""
|
|
|
|
if _can_import(fullname):
|
|
return lambda func: func
|
|
|
|
return unittest.skip('Unable to import "{}"'.format(fullname))
|
|
|
|
|
|
@cache_util.cached
|
|
def requires_platform(name):
|
|
"""Decorator used to skip tests if not running on the given platform.
|
|
"""
|
|
|
|
if name == platform.system():
|
|
return lambda func: func
|
|
|
|
return unittest.skip(
|
|
'Required platform "{}"" does not match system'.format(name))
|