Permalink
Fetching contributors…
Cannot retrieve contributors at this time
executable file 227 lines (209 sloc) 9.98 KB
#!/usr/bin/env python
import optparse
import os
import shutil
import sys
import unittest
def collect():
import tests
runtests(tests, 1)
def runtests(suite, verbosity):
results = unittest.TextTestRunner(verbosity=verbosity).run(suite)
return results.failures, results.errors
def get_option_parser():
parser = optparse.OptionParser()
basic = optparse.OptionGroup(parser, 'Basic test options')
basic.add_option(
'-e',
'--engine',
dest='engine',
help=('Database engine to test, one of '
'[sqlite, postgres, mysql, apsw, sqlcipher, berkeleydb]'))
basic.add_option('-v', '--verbosity', dest='verbosity', default=1, type='int', help='Verbosity of output')
suite = optparse.OptionGroup(parser, 'Simple test suite options')
suite.add_option('-a', '--all', dest='all', default=False, action='store_true', help='Run all tests, including extras')
suite.add_option('-x', '--extra', dest='extra', default=False, action='store_true', help='Run only extras tests')
cases = optparse.OptionGroup(parser, 'Individual test module options')
cases.add_option('--apsw', dest='apsw', default=False, action='store_true', help='apsw tests (requires apsw)')
cases.add_option('--berkeleydb', dest='berkeleydb', default=False, action='store_true', help='berkeleydb tests (requires pysqlite compiled against berkeleydb)')
cases.add_option('--csv', dest='csv', default=False, action='store_true', help='csv tests')
cases.add_option('--dataset', dest='dataset', default=False, action='store_true', help='dataset tests')
cases.add_option('--db-url', dest='db_url', default=False, action='store_true', help='db url tests')
cases.add_option('--djpeewee', dest='djpeewee', default=False, action='store_true', help='djpeewee tests')
cases.add_option('--fields', dest='fields', default=False, action='store_true', help='extra field tests')
cases.add_option('--flask', dest='flask', default=False, action='store_true', help='flask utils tests')
cases.add_option('--gfk', dest='gfk', default=False, action='store_true', help='gfk tests')
cases.add_option('--hybrid', dest='hybrid', default=False, action='store_true', help='hybrid property/method tests')
cases.add_option('--kv', dest='kv', default=False, action='store_true', help='key/value store tests')
cases.add_option('--manytomany', dest='manytomany', default=False, action='store_true', help='manytomany field tests')
cases.add_option('--migrations', dest='migrations', default=False, action='store_true', help='migration helper tests (requires psycopg2)')
cases.add_option('--pool', dest='pool', default=False, action='store_true', help='connection pool tests')
cases.add_option('--postgres-ext', dest='postgres_ext', default=False, action='store_true', help='postgres_ext tests (requires psycopg2)')
cases.add_option('--pwiz', dest='pwiz', default=False, action='store_true', help='pwiz, model code generator')
cases.add_option('--read-slave', dest='read_slave', default=False, action='store_true', help='read_slave tests')
cases.add_option('--reflection', dest='reflection', default=False, action='store_true', help='reflection schema introspector')
cases.add_option('--signals', dest='signals', default=False, action='store_true', help='signals tests')
cases.add_option('--shortcuts', dest='shortcuts', default=False, action='store_true', help='shortcuts tests')
cases.add_option('--speedups', dest='speedups', default=False, action='store_true', help='speedups c extension tests')
cases.add_option('--sqlcipher-ext', dest='sqlcipher', default=False, action='store_true', help='sqlcipher_ext tests (requires pysqlcipher)')
cases.add_option('--sqliteq', dest='sqliteq', default=False, action='store_true', help='sqliteq tests')
cases.add_option('--sqlite-c-ext', dest='sqlite_c', default=False, action='store_true', help='sqlite c extension tests')
cases.add_option('--sqlite-ext', dest='sqlite_ext', default=False, action='store_true', help='sqlite_ext tests')
cases.add_option('--sqlite-udf', dest='sqlite_udf', default=False, action='store_true', help='sqlite_udf tests')
cases.add_option('--test-utils', dest='test_utils', default=False, action='store_true', help='test_utils tests')
parser.add_option_group(basic)
parser.add_option_group(suite)
parser.add_option_group(cases)
return parser
def collect_modules(options):
modules = []
xtra = lambda op: op or options.extra or options.all
if xtra(options.apsw):
try:
from playhouse.tests import test_apsw
modules.append(test_apsw)
except ImportError:
print_('Unable to import apsw tests, skipping')
if xtra(options.berkeleydb):
try:
from playhouse.tests import test_berkeleydb
modules.append(test_berkeleydb)
except ImportError:
print_('Unable to import berkeleydb tests, skipping')
if xtra(options.csv):
from playhouse.tests import test_csv_utils
modules.append(test_csv_utils)
if xtra(options.dataset):
from playhouse.tests import test_dataset
modules.append(test_dataset)
if xtra(options.db_url):
from playhouse.tests import test_db_url
modules.append(test_db_url)
if xtra(options.djpeewee):
from playhouse.tests import test_djpeewee
modules.append(test_djpeewee)
if xtra(options.fields):
from playhouse.tests import test_extra_fields
from playhouse.tests import test_manytomany
modules.append(test_extra_fields)
if test_manytomany not in modules:
modules.append(test_manytomany)
if xtra(options.flask):
try:
import flask
except ImportError:
print_('Unable to import Flask tests, Flask is not installed.')
else:
from playhouse.tests import test_flask_utils
modules.append(test_flask_utils)
if xtra(options.gfk):
from playhouse.tests import test_gfk
modules.append(test_gfk)
if xtra(options.hybrid):
from playhouse.tests import test_hybrid
modules.append(test_hybrid)
if xtra(options.kv):
from playhouse.tests import test_kv
modules.append(test_kv)
if xtra(options.manytomany):
from playhouse.tests import test_manytomany
if test_manytomany not in modules:
modules.append(test_manytomany)
if xtra(options.migrations):
try:
from playhouse.tests import test_migrate
modules.append(test_migrate)
except ImportError:
print_('Unable to import migration tests, skipping')
if xtra(options.pool):
try:
from playhouse.tests import test_pool
modules.append(test_pool)
except ImportError:
print_('Unable to import connection pool tests, skipping')
if xtra(options.postgres_ext):
try:
from playhouse.tests import test_postgres
modules.append(test_postgres)
except ImportError:
print_('Unable to import postgres-ext tests, skipping')
if xtra(options.pwiz):
from playhouse.tests import test_pwiz
modules.append(test_pwiz)
if xtra(options.read_slave):
from playhouse.tests import test_read_slave
modules.append(test_read_slave)
if xtra(options.reflection):
from playhouse.tests import test_reflection
modules.append(test_reflection)
if xtra(options.signals):
from playhouse.tests import test_signals
modules.append(test_signals)
if xtra(options.shortcuts):
from playhouse.tests import test_shortcuts
modules.append(test_shortcuts)
if xtra(options.speedups):
try:
from playhouse.tests import test_speedups
modules.append(test_speedups)
except ImportError:
print_('Unable to import speedups tests, skipping')
if xtra(options.sqlcipher):
try:
from playhouse.tests import test_sqlcipher_ext
modules.append(test_sqlcipher_ext)
except ImportError:
print_('Unable to import pysqlcipher tests, skipping')
if xtra(options.sqliteq):
from playhouse.tests import test_sqliteq
modules.append(test_sqliteq)
if xtra(options.sqlite_c):
try:
from playhouse.tests import test_sqlite_c_ext
modules.append(test_sqlite_c_ext)
except ImportError:
print_('Unable to import SQLite C extension tests, skipping')
if xtra(options.sqlite_ext):
from playhouse.tests import test_sqlite_ext
modules.append(test_sqlite_ext)
if xtra(options.sqlite_udf):
from playhouse.tests import test_sqlite_udf
modules.append(test_sqlite_udf)
if xtra(options.test_utils):
from playhouse.tests import test_test_utils
modules.append(test_test_utils)
if not modules or options.all:
import tests
modules.insert(0, tests)
return modules
if __name__ == '__main__':
parser = get_option_parser()
options, args = parser.parse_args()
if options.engine:
os.environ['PEEWEE_TEST_BACKEND'] = options.engine
os.environ['PEEWEE_TEST_VERBOSITY'] = str(options.verbosity)
from peewee import print_
suite = unittest.TestSuite()
for module in collect_modules(options):
print_('Adding tests for "%s"' % module.__name__)
module_suite = unittest.TestLoader().loadTestsFromModule(module)
suite.addTest(module_suite)
failures, errors = runtests(suite, options.verbosity)
files_to_delete = [
'peewee_test.db',
'peewee_test',
'tmp.db',
'peewee_test.bdb.db',
'peewee_test.cipher.db']
paths_to_delete = ['peewee_test.bdb.db-journal']
for filename in files_to_delete:
if os.path.exists(filename):
os.unlink(filename)
for path in paths_to_delete:
if os.path.exists(path):
shutil.rmtree(path)
if errors:
sys.exit(2)
elif failures:
sys.exit(1)
sys.exit(0)