tests: add a --help option to make runtest more friendly
authorMike Frysinger <vapier@gentoo.org>
Sun, 9 Oct 2011 18:42:01 +0000 (14:42 -0400)
committerMike Frysinger <vapier@gentoo.org>
Sun, 9 Oct 2011 20:40:43 +0000 (16:40 -0400)
At the moment, trying to do `./runtests.h -h` just produces an ugly and
indecipherable traceback.  Make it a bit more friendly.

Signed-off-by: Mike Frysinger <vapier@gentoo.org>
pym/portage/tests/__init__.py

index 7f1ed3f70cef277c4958d2e564a8de7675c08c12..fa297b5bbdc6e8a6c02b46ed2c6cf5d9fdd85117 100644 (file)
@@ -5,6 +5,7 @@
 import sys
 import time
 import unittest
+from optparse import OptionParser, OptionValueError
 
 try:
        from unittest.runner import _TextTestResult # new in python-2.7
@@ -23,8 +24,12 @@ def main():
        basedir = os.path.dirname(os.path.realpath(__file__))
        testDirs = []
 
-       if len(sys.argv) > 1:
-               suite.addTests(getTestFromCommandLine(sys.argv[1:], basedir))
+       usage = "usage: %s [options] [tests to run]" % os.path.basename(sys.argv[0])
+       parser = OptionParser(usage=usage)
+       (options, args) = parser.parse_args(args=sys.argv)
+
+       if len(args) > 1:
+               suite.addTests(getTestFromCommandLine(args[1:], basedir))
                return TextTestRunner(verbosity=2).run(suite)
 
        # the os.walk help mentions relative paths as being quirky