mirror of
https://github.com/wagtail/wagtail.git
synced 2024-11-24 19:17:48 +01:00
e5a4208774
This commit implements a '--bench' argument for runtests.py. When specified, this runs a couple of special "benchmark" tests that are specifically created for testing performance of the explorer page. It's very rudimentary at the moment as I needed something quickly to implement the performance improvement in this PR. It gets the job done though.
84 lines
2.9 KiB
Python
Executable File
84 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import argparse
|
|
import os
|
|
import shutil
|
|
import sys
|
|
import warnings
|
|
|
|
from django.core.management import execute_from_command_line
|
|
|
|
os.environ['DJANGO_SETTINGS_MODULE'] = 'wagtail.tests.settings'
|
|
|
|
|
|
def make_parser():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--deprecation', choices=['all', 'pending', 'imminent', 'none'], default='imminent')
|
|
parser.add_argument('--postgres', action='store_true')
|
|
parser.add_argument('--elasticsearch2', action='store_true')
|
|
parser.add_argument('--elasticsearch5', action='store_true')
|
|
parser.add_argument('--elasticsearch6', action='store_true')
|
|
parser.add_argument('--bench', action='store_true')
|
|
return parser
|
|
|
|
|
|
def parse_args(args=None):
|
|
return make_parser().parse_known_args(args)
|
|
|
|
|
|
def runtests():
|
|
args, rest = parse_args()
|
|
|
|
only_wagtail = r'^wagtail(\.|$)'
|
|
if args.deprecation == 'all':
|
|
# Show all deprecation warnings from all packages
|
|
warnings.simplefilter('default', DeprecationWarning)
|
|
warnings.simplefilter('default', PendingDeprecationWarning)
|
|
elif args.deprecation == 'pending':
|
|
# Show all deprecation warnings from wagtail
|
|
warnings.filterwarnings('default', category=DeprecationWarning, module=only_wagtail)
|
|
warnings.filterwarnings('default', category=PendingDeprecationWarning, module=only_wagtail)
|
|
elif args.deprecation == 'imminent':
|
|
# Show only imminent deprecation warnings from wagtail
|
|
warnings.filterwarnings('default', category=DeprecationWarning, module=only_wagtail)
|
|
elif args.deprecation == 'none':
|
|
# Deprecation warnings are ignored by default
|
|
pass
|
|
|
|
if args.postgres:
|
|
os.environ['DATABASE_ENGINE'] = 'django.db.backends.postgresql'
|
|
|
|
if args.elasticsearch2:
|
|
os.environ.setdefault('ELASTICSEARCH_URL', 'http://localhost:9200')
|
|
os.environ.setdefault('ELASTICSEARCH_VERSION', '2')
|
|
elif args.elasticsearch5:
|
|
os.environ.setdefault('ELASTICSEARCH_URL', 'http://localhost:9200')
|
|
os.environ.setdefault('ELASTICSEARCH_VERSION', '5')
|
|
elif args.elasticsearch6:
|
|
os.environ.setdefault('ELASTICSEARCH_URL', 'http://localhost:9200')
|
|
os.environ.setdefault('ELASTICSEARCH_VERSION', '6')
|
|
|
|
elif 'ELASTICSEARCH_URL' in os.environ:
|
|
# forcibly delete the ELASTICSEARCH_URL setting to skip those tests
|
|
del os.environ['ELASTICSEARCH_URL']
|
|
|
|
if args.bench:
|
|
benchmarks = [
|
|
'wagtail.admin.tests.benches',
|
|
]
|
|
|
|
argv = [sys.argv[0], 'test', '-v2'] + benchmarks + rest
|
|
else:
|
|
argv = [sys.argv[0], 'test'] + rest
|
|
|
|
try:
|
|
execute_from_command_line(argv)
|
|
finally:
|
|
from wagtail.tests.settings import STATIC_ROOT, MEDIA_ROOT
|
|
shutil.rmtree(STATIC_ROOT, ignore_errors=True)
|
|
shutil.rmtree(MEDIA_ROOT, ignore_errors=True)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
runtests()
|