130 lines
3.6 KiB
Python
130 lines
3.6 KiB
Python
#
|
|
# Gramps - a GTK+/GNOME based genealogy program
|
|
#
|
|
# Copyright (C) 2000-2005 Donald N. Allingham
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
#
|
|
|
|
# Written by Richard Taylor
|
|
|
|
"""
|
|
Testing framework for performing a variety of unttests for GRAMPS.
|
|
"""
|
|
|
|
import logging
|
|
|
|
import os
|
|
import sys
|
|
import unittest
|
|
from optparse import OptionParser
|
|
|
|
sys.path.append(os.path.join(os.path.dirname(__file__), ".."))
|
|
os.environ["GRAMPS_RESOURCES"] = os.path.join(
|
|
os.path.dirname(os.path.abspath(__file__)), ".."
|
|
)
|
|
|
|
|
|
def make_parser():
|
|
usage = "usage: %prog [options]"
|
|
parser = OptionParser(usage)
|
|
parser.add_option(
|
|
"-v",
|
|
"--verbosity",
|
|
type="int",
|
|
dest="verbose_level",
|
|
default=0,
|
|
help="Level of verboseness",
|
|
)
|
|
parser.add_option(
|
|
"-p",
|
|
"--performance",
|
|
action="store_true",
|
|
dest="performance",
|
|
default=False,
|
|
help="Run the performance tests.",
|
|
)
|
|
return parser
|
|
|
|
|
|
def getTestSuites():
|
|
# Sorry about this line, but it is the easiest way of doing it.
|
|
# It just walks the filetree from '.' downwards and returns
|
|
# a tuple per directory of (dirpath,filelist) if the directory
|
|
# contains any test files.
|
|
|
|
paths = [
|
|
(f[0], f[2])
|
|
for f in os.walk(".")
|
|
if len([i for i in f[2] if i[-8:] == "_Test.py"])
|
|
]
|
|
|
|
test_suites = []
|
|
perf_suites = []
|
|
|
|
for dir, test_modules in paths:
|
|
sys.path.append(dir)
|
|
|
|
for module in test_modules:
|
|
if module[-8:] != "_Test.py":
|
|
continue
|
|
mod = __import__(module[:-3])
|
|
test_suites.append(mod.testSuite())
|
|
try:
|
|
perf_suites.append(mod.perfSuite())
|
|
except:
|
|
pass
|
|
|
|
return (test_suites, perf_suites)
|
|
|
|
|
|
def allTheTests():
|
|
return unittest.TestSuite(getTestSuites()[0])
|
|
|
|
|
|
def perfTests():
|
|
return unittest.TestSuite(getTestSuites()[1])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
console = logging.StreamHandler()
|
|
console.setLevel(logging.INFO)
|
|
console.setFormatter(logging.Formatter("%(name)-12s: %(levelname)-8s %(message)s"))
|
|
|
|
logger = logging.getLogger("Gramps")
|
|
logger.addHandler(console)
|
|
|
|
(options, args) = make_parser().parse_args()
|
|
|
|
if options.verbose_level == 1:
|
|
logger.setLevel(logging.INFO)
|
|
console.setLevel(logging.INFO)
|
|
elif options.verbose_level == 2:
|
|
logger.setLevel(logging.DEBUG)
|
|
console.setLevel(logging.DEBUG)
|
|
elif options.verbose_level == 3:
|
|
logger.setLevel(logging.NOTSET)
|
|
console.setLevel(logging.NOTSET)
|
|
elif options.verbose_level >= 4:
|
|
logger.setLevel(logging.NOTSET)
|
|
console.setLevel(logging.NOTSET)
|
|
os.environ["GRAMPS_SIGNAL"] = "1"
|
|
else:
|
|
logger.setLevel(logging.ERROR)
|
|
console.setLevel(logging.ERROR)
|
|
|
|
tests = perfTests() if options.performance else allTheTests()
|
|
unittest.TextTestRunner(verbosity=options.verbose_level).run(tests)
|