mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-09-20 19:09:03 +03:00
52 lines
1.5 KiB
Python
Executable File
52 lines
1.5 KiB
Python
Executable File
#! /usr/bin/env python3
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
|
|
def split_class_and_function(string):
|
|
class_string, function_string = string.split('.', 1)
|
|
return "%s and %s" % (class_string, function_string)
|
|
|
|
|
|
def is_function(string):
|
|
# `True` if it looks like a test function is included in the string.
|
|
return string.startswith('test_') or '.test_' in string
|
|
|
|
|
|
def is_class(string):
|
|
# `True` if first character is uppercase - assume it's a class name.
|
|
return string[0] == string[0].upper()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) > 1:
|
|
pytest_args = sys.argv[1:]
|
|
first_arg = pytest_args[0]
|
|
|
|
try:
|
|
pytest_args.remove('--coverage')
|
|
except ValueError:
|
|
pass
|
|
else:
|
|
pytest_args = [
|
|
'--cov', '.',
|
|
'--cov-report', 'xml',
|
|
] + pytest_args
|
|
|
|
if first_arg.startswith('-'):
|
|
# `runtests.py [flags]`
|
|
pytest_args = ['tests'] + pytest_args
|
|
elif is_class(first_arg) and is_function(first_arg):
|
|
# `runtests.py TestCase.test_function [flags]`
|
|
expression = split_class_and_function(first_arg)
|
|
pytest_args = ['tests', '-k', expression] + pytest_args[1:]
|
|
elif is_class(first_arg) or is_function(first_arg):
|
|
# `runtests.py TestCase [flags]`
|
|
# `runtests.py test_function [flags]`
|
|
pytest_args = ['tests', '-k', pytest_args[0]] + pytest_args[1:]
|
|
else:
|
|
pytest_args = []
|
|
|
|
sys.exit(pytest.main(pytest_args))
|