| Index: Tools/Scripts/webkitpy/thirdparty/unittest2/loader.py
|
| diff --git a/Tools/Scripts/webkitpy/thirdparty/unittest2/loader.py b/Tools/Scripts/webkitpy/thirdparty/unittest2/loader.py
|
| deleted file mode 100644
|
| index 690df41f8e14bc0745bd17326fd98f1de601b062..0000000000000000000000000000000000000000
|
| --- a/Tools/Scripts/webkitpy/thirdparty/unittest2/loader.py
|
| +++ /dev/null
|
| @@ -1,322 +0,0 @@
|
| -"""Loading unittests."""
|
| -
|
| -import os
|
| -import re
|
| -import sys
|
| -import traceback
|
| -import types
|
| -import unittest
|
| -
|
| -from fnmatch import fnmatch
|
| -
|
| -from unittest2 import case, suite
|
| -
|
| -try:
|
| - from os.path import relpath
|
| -except ImportError:
|
| - from unittest2.compatibility import relpath
|
| -
|
| -__unittest = True
|
| -
|
| -
|
| -def _CmpToKey(mycmp):
|
| - 'Convert a cmp= function into a key= function'
|
| - class K(object):
|
| - def __init__(self, obj):
|
| - self.obj = obj
|
| - def __lt__(self, other):
|
| - return mycmp(self.obj, other.obj) == -1
|
| - return K
|
| -
|
| -
|
| -# what about .pyc or .pyo (etc)
|
| -# we would need to avoid loading the same tests multiple times
|
| -# from '.py', '.pyc' *and* '.pyo'
|
| -VALID_MODULE_NAME = re.compile(r'[_a-z]\w*\.py$', re.IGNORECASE)
|
| -
|
| -
|
| -def _make_failed_import_test(name, suiteClass):
|
| - message = 'Failed to import test module: %s' % name
|
| - if hasattr(traceback, 'format_exc'):
|
| - # Python 2.3 compatibility
|
| - # format_exc returns two frames of discover.py as well
|
| - message += '\n%s' % traceback.format_exc()
|
| - return _make_failed_test('ModuleImportFailure', name, ImportError(message),
|
| - suiteClass)
|
| -
|
| -def _make_failed_load_tests(name, exception, suiteClass):
|
| - return _make_failed_test('LoadTestsFailure', name, exception, suiteClass)
|
| -
|
| -def _make_failed_test(classname, methodname, exception, suiteClass):
|
| - def testFailure(self):
|
| - raise exception
|
| - attrs = {methodname: testFailure}
|
| - TestClass = type(classname, (case.TestCase,), attrs)
|
| - return suiteClass((TestClass(methodname),))
|
| -
|
| -
|
| -class TestLoader(unittest.TestLoader):
|
| - """
|
| - This class is responsible for loading tests according to various criteria
|
| - and returning them wrapped in a TestSuite
|
| - """
|
| - testMethodPrefix = 'test'
|
| - sortTestMethodsUsing = cmp
|
| - suiteClass = suite.TestSuite
|
| - _top_level_dir = None
|
| -
|
| - def loadTestsFromTestCase(self, testCaseClass):
|
| - """Return a suite of all tests cases contained in testCaseClass"""
|
| - if issubclass(testCaseClass, suite.TestSuite):
|
| - raise TypeError("Test cases should not be derived from TestSuite."
|
| - " Maybe you meant to derive from TestCase?")
|
| - testCaseNames = self.getTestCaseNames(testCaseClass)
|
| - if not testCaseNames and hasattr(testCaseClass, 'runTest'):
|
| - testCaseNames = ['runTest']
|
| - loaded_suite = self.suiteClass(map(testCaseClass, testCaseNames))
|
| - return loaded_suite
|
| -
|
| - def loadTestsFromModule(self, module, use_load_tests=True):
|
| - """Return a suite of all tests cases contained in the given module"""
|
| - tests = []
|
| - for name in dir(module):
|
| - obj = getattr(module, name)
|
| - if isinstance(obj, type) and issubclass(obj, unittest.TestCase):
|
| - tests.append(self.loadTestsFromTestCase(obj))
|
| -
|
| - load_tests = getattr(module, 'load_tests', None)
|
| - tests = self.suiteClass(tests)
|
| - if use_load_tests and load_tests is not None:
|
| - try:
|
| - return load_tests(self, tests, None)
|
| - except Exception, e:
|
| - return _make_failed_load_tests(module.__name__, e,
|
| - self.suiteClass)
|
| - return tests
|
| -
|
| - def loadTestsFromName(self, name, module=None):
|
| - """Return a suite of all tests cases given a string specifier.
|
| -
|
| - The name may resolve either to a module, a test case class, a
|
| - test method within a test case class, or a callable object which
|
| - returns a TestCase or TestSuite instance.
|
| -
|
| - The method optionally resolves the names relative to a given module.
|
| - """
|
| - parts = name.split('.')
|
| - if module is None:
|
| - parts_copy = parts[:]
|
| - while parts_copy:
|
| - try:
|
| - module = __import__('.'.join(parts_copy))
|
| - break
|
| - except ImportError:
|
| - del parts_copy[-1]
|
| - if not parts_copy:
|
| - raise
|
| - parts = parts[1:]
|
| - obj = module
|
| - for part in parts:
|
| - parent, obj = obj, getattr(obj, part)
|
| -
|
| - if isinstance(obj, types.ModuleType):
|
| - return self.loadTestsFromModule(obj)
|
| - elif isinstance(obj, type) and issubclass(obj, unittest.TestCase):
|
| - return self.loadTestsFromTestCase(obj)
|
| - elif (isinstance(obj, types.UnboundMethodType) and
|
| - isinstance(parent, type) and
|
| - issubclass(parent, case.TestCase)):
|
| - return self.suiteClass([parent(obj.__name__)])
|
| - elif isinstance(obj, unittest.TestSuite):
|
| - return obj
|
| - elif hasattr(obj, '__call__'):
|
| - test = obj()
|
| - if isinstance(test, unittest.TestSuite):
|
| - return test
|
| - elif isinstance(test, unittest.TestCase):
|
| - return self.suiteClass([test])
|
| - else:
|
| - raise TypeError("calling %s returned %s, not a test" %
|
| - (obj, test))
|
| - else:
|
| - raise TypeError("don't know how to make test from: %s" % obj)
|
| -
|
| - def loadTestsFromNames(self, names, module=None):
|
| - """Return a suite of all tests cases found using the given sequence
|
| - of string specifiers. See 'loadTestsFromName()'.
|
| - """
|
| - suites = [self.loadTestsFromName(name, module) for name in names]
|
| - return self.suiteClass(suites)
|
| -
|
| - def getTestCaseNames(self, testCaseClass):
|
| - """Return a sorted sequence of method names found within testCaseClass
|
| - """
|
| - def isTestMethod(attrname, testCaseClass=testCaseClass,
|
| - prefix=self.testMethodPrefix):
|
| - return attrname.startswith(prefix) and \
|
| - hasattr(getattr(testCaseClass, attrname), '__call__')
|
| - testFnNames = filter(isTestMethod, dir(testCaseClass))
|
| - if self.sortTestMethodsUsing:
|
| - testFnNames.sort(key=_CmpToKey(self.sortTestMethodsUsing))
|
| - return testFnNames
|
| -
|
| - def discover(self, start_dir, pattern='test*.py', top_level_dir=None):
|
| - """Find and return all test modules from the specified start
|
| - directory, recursing into subdirectories to find them. Only test files
|
| - that match the pattern will be loaded. (Using shell style pattern
|
| - matching.)
|
| -
|
| - All test modules must be importable from the top level of the project.
|
| - If the start directory is not the top level directory then the top
|
| - level directory must be specified separately.
|
| -
|
| - If a test package name (directory with '__init__.py') matches the
|
| - pattern then the package will be checked for a 'load_tests' function. If
|
| - this exists then it will be called with loader, tests, pattern.
|
| -
|
| - If load_tests exists then discovery does *not* recurse into the package,
|
| - load_tests is responsible for loading all tests in the package.
|
| -
|
| - The pattern is deliberately not stored as a loader attribute so that
|
| - packages can continue discovery themselves. top_level_dir is stored so
|
| - load_tests does not need to pass this argument in to loader.discover().
|
| - """
|
| - set_implicit_top = False
|
| - if top_level_dir is None and self._top_level_dir is not None:
|
| - # make top_level_dir optional if called from load_tests in a package
|
| - top_level_dir = self._top_level_dir
|
| - elif top_level_dir is None:
|
| - set_implicit_top = True
|
| - top_level_dir = start_dir
|
| -
|
| - top_level_dir = os.path.abspath(top_level_dir)
|
| -
|
| - if not top_level_dir in sys.path:
|
| - # all test modules must be importable from the top level directory
|
| - # should we *unconditionally* put the start directory in first
|
| - # in sys.path to minimise likelihood of conflicts between installed
|
| - # modules and development versions?
|
| - sys.path.insert(0, top_level_dir)
|
| - self._top_level_dir = top_level_dir
|
| -
|
| - is_not_importable = False
|
| - if os.path.isdir(os.path.abspath(start_dir)):
|
| - start_dir = os.path.abspath(start_dir)
|
| - if start_dir != top_level_dir:
|
| - is_not_importable = not os.path.isfile(os.path.join(start_dir, '__init__.py'))
|
| - else:
|
| - # support for discovery from dotted module names
|
| - try:
|
| - __import__(start_dir)
|
| - except ImportError:
|
| - is_not_importable = True
|
| - else:
|
| - the_module = sys.modules[start_dir]
|
| - top_part = start_dir.split('.')[0]
|
| - start_dir = os.path.abspath(os.path.dirname((the_module.__file__)))
|
| - if set_implicit_top:
|
| - self._top_level_dir = os.path.abspath(os.path.dirname(os.path.dirname(sys.modules[top_part].__file__)))
|
| - sys.path.remove(top_level_dir)
|
| -
|
| - if is_not_importable:
|
| - raise ImportError('Start directory is not importable: %r' % start_dir)
|
| -
|
| - tests = list(self._find_tests(start_dir, pattern))
|
| - return self.suiteClass(tests)
|
| -
|
| - def _get_name_from_path(self, path):
|
| - path = os.path.splitext(os.path.normpath(path))[0]
|
| -
|
| - _relpath = relpath(path, self._top_level_dir)
|
| - assert not os.path.isabs(_relpath), "Path must be within the project"
|
| - assert not _relpath.startswith('..'), "Path must be within the project"
|
| -
|
| - name = _relpath.replace(os.path.sep, '.')
|
| - return name
|
| -
|
| - def _get_module_from_name(self, name):
|
| - __import__(name)
|
| - return sys.modules[name]
|
| -
|
| - def _match_path(self, path, full_path, pattern):
|
| - # override this method to use alternative matching strategy
|
| - return fnmatch(path, pattern)
|
| -
|
| - def _find_tests(self, start_dir, pattern):
|
| - """Used by discovery. Yields test suites it loads."""
|
| - paths = os.listdir(start_dir)
|
| -
|
| - for path in paths:
|
| - full_path = os.path.join(start_dir, path)
|
| - if os.path.isfile(full_path):
|
| - if not VALID_MODULE_NAME.match(path):
|
| - # valid Python identifiers only
|
| - continue
|
| - if not self._match_path(path, full_path, pattern):
|
| - continue
|
| - # if the test file matches, load it
|
| - name = self._get_name_from_path(full_path)
|
| - try:
|
| - module = self._get_module_from_name(name)
|
| - except:
|
| - yield _make_failed_import_test(name, self.suiteClass)
|
| - else:
|
| - mod_file = os.path.abspath(getattr(module, '__file__', full_path))
|
| - realpath = os.path.splitext(mod_file)[0]
|
| - fullpath_noext = os.path.splitext(full_path)[0]
|
| - if realpath.lower() != fullpath_noext.lower():
|
| - module_dir = os.path.dirname(realpath)
|
| - mod_name = os.path.splitext(os.path.basename(full_path))[0]
|
| - expected_dir = os.path.dirname(full_path)
|
| - msg = ("%r module incorrectly imported from %r. Expected %r. "
|
| - "Is this module globally installed?")
|
| - raise ImportError(msg % (mod_name, module_dir, expected_dir))
|
| - yield self.loadTestsFromModule(module)
|
| - elif os.path.isdir(full_path):
|
| - if not os.path.isfile(os.path.join(full_path, '__init__.py')):
|
| - continue
|
| -
|
| - load_tests = None
|
| - tests = None
|
| - if fnmatch(path, pattern):
|
| - # only check load_tests if the package directory itself matches the filter
|
| - name = self._get_name_from_path(full_path)
|
| - package = self._get_module_from_name(name)
|
| - load_tests = getattr(package, 'load_tests', None)
|
| - tests = self.loadTestsFromModule(package, use_load_tests=False)
|
| -
|
| - if load_tests is None:
|
| - if tests is not None:
|
| - # tests loaded from package file
|
| - yield tests
|
| - # recurse into the package
|
| - for test in self._find_tests(full_path, pattern):
|
| - yield test
|
| - else:
|
| - try:
|
| - yield load_tests(self, tests, pattern)
|
| - except Exception, e:
|
| - yield _make_failed_load_tests(package.__name__, e,
|
| - self.suiteClass)
|
| -
|
| -defaultTestLoader = TestLoader()
|
| -
|
| -
|
| -def _makeLoader(prefix, sortUsing, suiteClass=None):
|
| - loader = TestLoader()
|
| - loader.sortTestMethodsUsing = sortUsing
|
| - loader.testMethodPrefix = prefix
|
| - if suiteClass:
|
| - loader.suiteClass = suiteClass
|
| - return loader
|
| -
|
| -def getTestCaseNames(testCaseClass, prefix, sortUsing=cmp):
|
| - return _makeLoader(prefix, sortUsing).getTestCaseNames(testCaseClass)
|
| -
|
| -def makeSuite(testCaseClass, prefix='test', sortUsing=cmp,
|
| - suiteClass=suite.TestSuite):
|
| - return _makeLoader(prefix, sortUsing, suiteClass).loadTestsFromTestCase(testCaseClass)
|
| -
|
| -def findTestCases(module, prefix='test', sortUsing=cmp,
|
| - suiteClass=suite.TestSuite):
|
| - return _makeLoader(prefix, sortUsing, suiteClass).loadTestsFromModule(module)
|
|
|