blob: 21eceac06d4d7d854fca7f05620656d6c761368c [file] [log] [blame]
# Copyright (C) 2012 Google, Inc.
# Copyright (C) 2010 Chris Jerdonek (cjerdonek@webkit.org)
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS BE LIABLE FOR
# ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
"""this module is responsible for finding python tests."""
import logging
import re
_log = logging.getLogger(__name__)
class _DirectoryTree(object):
def __init__(self, filesystem, top_directory, starting_subdirectory):
self.filesystem = filesystem
self.top_directory = filesystem.realpath(top_directory)
self.search_directory = self.top_directory
self.top_package = ''
if starting_subdirectory:
self.top_package = starting_subdirectory.replace(filesystem.sep, '.') + '.'
self.search_directory = filesystem.join(self.top_directory, starting_subdirectory)
def find_modules(self, suffixes, sub_directory=None):
if sub_directory:
search_directory = self.filesystem.join(self.top_directory, sub_directory)
else:
search_directory = self.search_directory
def file_filter(filesystem, dirname, basename):
return any(basename.endswith(suffix) for suffix in suffixes)
filenames = self.filesystem.files_under(search_directory, file_filter=file_filter)
return [self.to_module(filename) for filename in filenames]
def to_module(self, path):
return path.replace(self.top_directory + self.filesystem.sep, '').replace(self.filesystem.sep, '.')[:-3]
def subpath(self, path):
"""Returns the relative path from the top of the tree to the path, or None if the path is not under the top of the tree."""
realpath = self.filesystem.realpath(self.filesystem.join(self.top_directory, path))
if realpath.startswith(self.top_directory + self.filesystem.sep):
return realpath.replace(self.top_directory + self.filesystem.sep, '')
return None
def clean(self):
"""Delete all .pyc files in the tree that have no matching .py file."""
_log.debug("Cleaning orphaned *.pyc files from: %s" % self.search_directory)
filenames = self.filesystem.files_under(self.search_directory)
for filename in filenames:
if filename.endswith(".pyc") and filename[:-1] not in filenames:
_log.info("Deleting orphan *.pyc file: %s" % filename)
self.filesystem.remove(filename)
class Finder(object):
def __init__(self, filesystem):
self.filesystem = filesystem
self.trees = []
self._names_to_skip = []
def add_tree(self, top_directory, starting_subdirectory=None):
self.trees.append(_DirectoryTree(self.filesystem, top_directory, starting_subdirectory))
def skip(self, names, reason, bugid):
self._names_to_skip.append(tuple([names, reason, bugid]))
def additional_paths(self, paths):
return [tree.top_directory for tree in self.trees if tree.top_directory not in paths]
def clean_trees(self):
for tree in self.trees:
tree.clean()
def is_module(self, name):
relpath = name.replace('.', self.filesystem.sep) + '.py'
return any(self.filesystem.exists(self.filesystem.join(tree.top_directory, relpath)) for tree in self.trees)
def is_dotted_name(self, name):
return re.match(r'[a-zA-Z.][a-zA-Z0-9_.]*', name)
def to_module(self, path):
for tree in self.trees:
if path.startswith(tree.top_directory):
return tree.to_module(path)
return None
def find_names(self, args, find_all):
suffixes = ['_unittest.py', '_integrationtest.py']
if args:
names = []
for arg in args:
names.extend(self._find_names_for_arg(arg, suffixes))
return names
return self._default_names(suffixes, find_all)
def _find_names_for_arg(self, arg, suffixes):
realpath = self.filesystem.realpath(arg)
if self.filesystem.exists(realpath):
names = self._find_in_trees(realpath, suffixes)
if not names:
_log.error("%s is not in one of the test trees." % arg)
return names
# See if it's a python package in a tree (or a relative path from the top of a tree).
names = self._find_in_trees(arg.replace('.', self.filesystem.sep), suffixes)
if names:
return names
if self.is_dotted_name(arg):
# The name may not exist, but that's okay; we'll find out later.
return [arg]
_log.error("%s is not a python name or an existing file or directory." % arg)
return []
def _find_in_trees(self, path, suffixes):
for tree in self.trees:
relpath = tree.subpath(path)
if not relpath:
continue
if self.filesystem.isfile(path):
return [tree.to_module(path)]
else:
return tree.find_modules(suffixes, path)
return []
def _default_names(self, suffixes, find_all):
modules = []
for tree in self.trees:
modules.extend(tree.find_modules(suffixes))
modules.sort()
for module in modules:
_log.debug("Found: %s" % module)
if not find_all:
for (names, reason, bugid) in self._names_to_skip:
self._exclude(modules, names, reason, bugid)
return modules
def _exclude(self, modules, module_prefixes, reason, bugid):
_log.info('Skipping tests in the following modules or packages because they %s:' % reason)
for prefix in module_prefixes:
_log.info(' %s' % prefix)
modules_to_exclude = filter(lambda m: m.startswith(prefix), modules)
for m in modules_to_exclude:
if len(modules_to_exclude) > 1:
_log.debug(' %s' % m)
modules.remove(m)
_log.info(' (https://bugs.webkit.org/show_bug.cgi?id=%d; use --all to include)' % bugid)
_log.info('')