1  # Copyright (C) 2005-2012 Aaron Bentley <abentley@panoramicfeedback.com>
  2  #                         Chris Ball <cjb@laptop.org>
  3  #                         Marien Zwart <marien.zwart@gmail.com>
  4  #                         W. Trevor King <wking@tremily.us>
  5  #
  6  # This file is part of Bugs Everywhere.
  7  #
  8  # Bugs Everywhere is free software: you can redistribute it and/or modify it
  9  # under the terms of the GNU General Public License as published by the Free
 10  # Software Foundation, either version 2 of the License, or (at your option) any
 11  # later version.
 12  #
 13  # Bugs Everywhere is distributed in the hope that it will be useful, but
 14  # WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 15  # FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 16  # more details.
 17  #
 18  # You should have received a copy of the GNU General Public License along with
 19  # Bugs Everywhere.  If not, see <http://www.gnu.org/licenses/>.
 20  
 21  import doctest
 22  import os
 23  import os.path
 24  import sys
 25  import unittest
 26  
 27  import libbe
 28  libbe.TESTING = True
 29  from libbe.util.tree import Tree
 30  from libbe.util.plugin import import_by_name
 31  from libbe.version import version
 32  
 33  def python_tree(root_path='libbe', root_modname='libbe'):
 34      tree = Tree()
 35      tree.path = root_path
 36      tree.parent = None
 37      stack = [tree]
 38      while len(stack) > 0:
 39          f = stack.pop(0)
 40          if f.path.endswith('.py'):
 41              f.name = os.path.basename(f.path)[:-len('.py')]
 42          elif os.path.isdir(f.path) \
 43                  and os.path.exists(os.path.join(f.path, '__init__.py')):
 44              f.name = os.path.basename(f.path)
 45              f.is_module = True
 46              for child in os.listdir(f.path):
 47                  if child == '__init__.py':
 48                      continue
 49                  c = Tree()
 50                  c.path = os.path.join(f.path, child)
 51                  c.parent = f
 52                  stack.append(c)
 53          else:
 54              continue
 55          if f.parent == None:
 56              f.modname = root_modname
 57          else:
 58              f.modname = f.parent.modname + '.' + f.name
 59              f.parent.append(f)
 60      return tree
 61  
 62  def add_module_tests(suite, modname):
 63      try:
 64          mod = import_by_name(modname)
 65      except ValueError as e:
 66          sys.stderr.write('Failed to import "{}"\n'.format(modname))
 67          raise e
 68      if hasattr(mod, 'suite'):
 69          s = mod.suite
 70      else:
 71          s = unittest.TestLoader().loadTestsFromModule(mod)
 72          try:
 73              sdoc = doctest.DocTestSuite(mod)
 74              suite.addTest(sdoc)
 75          except ValueError:
 76              pass
 77      suite.addTest(s)
 78  
 79  if __name__ == '__main__':
 80      import optparse
 81      parser = optparse.OptionParser(usage='%prog [options] [modules ...]',
 82                                     description=
 83  """When called without optional module names, run the test suites for
 84  *all* modules.  This may raise lots of errors if you haven't installed
 85  one of the versioning control systems.
 86  
 87  When called with module name arguments, only run the test suites from
 88  those modules and their submodules.  For example::
 89  
 90      $ python test.py libbe.bugdir libbe.storage
 91  """)
 92      parser.add_option('-q', '--quiet', action='store_true', default=False,
 93                        help='Run unittests in quiet mode (verbosity 1).')
 94      options,args = parser.parse_args()
 95      sys.stderr.write('Testing BE\n{}\n'.format(version(verbose=True)))
 96  
 97      verbosity = 2
 98      if options.quiet == True:
 99          verbosity = 1
100  
101      suite = unittest.TestSuite()
102      tree = python_tree()
103      if len(args) == 0:
104          for node in tree.traverse():
105              add_module_tests(suite, node.modname)
106      else:
107          added = []
108          for modname in args:
109              for node in tree.traverse():
110                  if node.modname == modname:
111                      for n in node.traverse():
112                          if n.modname not in added:
113                              add_module_tests(suite, n.modname)
114                              added.append(n.modname)
115                      break
116      
117      result = unittest.TextTestRunner(verbosity=verbosity).run(suite)
118      
119      numErrors = len(result.errors)
120      numFailures = len(result.failures)
121      numBad = numErrors + numFailures
122      if numBad > 126:
123          numBad = 1
124      sys.exit(numBad)