Made testify completion complete method names.

This commit is contained in:
Ivan Malison 2013-03-19 09:47:46 -07:00
parent 6bbd27029c
commit 7f26363ccb

View File

@ -3,7 +3,7 @@
_testify() { _testify() {
typeset -A opt_args typeset -A opt_args
local context state line local context state line
BASE_TEST_PATH='tests' BASE_TEST_PATH='tests'
_arguments "1:testify_file:->file" "2:testify_:->test_case" _arguments "1:testify_file:->file" "2:testify_:->test_case"
@ -24,9 +24,13 @@ _print_test_methods() {
python - "$@" <<ENDPYTHON python - "$@" <<ENDPYTHON
import sys import sys
from testify import test_discovery from testify import test_discovery
if __name__ == '__main__': if __name__ == '__main__':
for test_case in test_discovery.discover(sys.argv[1]): for test_case in test_discovery.discover(sys.argv[1]):
print test_case.__name__ print test_case.__name__
for member in dir(test_case):
if member.startswith("test"):
print "%s.%s" % (test_case.__name__, member)
ENDPYTHON ENDPYTHON
} }