From: James Westby <james.westby@canonical.com>
Date: Mon, 7 Jun 2010 18:37:52 +0000 (+0100)
Subject: Add --discover-pattern.
X-Git-Url: http://git.samba.org/samba.git/?p=third_party%2Fsubunit;a=commitdiff_plain;h=4700d73f0ef336c6a51159bae023aa15746a1e71

Add --discover-pattern.
---

diff --git a/python/subunit/run.py b/python/subunit/run.py
index 5c440eb..7925512 100755
--- a/python/subunit/run.py
+++ b/python/subunit/run.py
@@ -49,6 +49,9 @@ if __name__ == '__main__':
     if has_discover:
         parser.add_option("--discover", dest="discover", action="store_true",
                 help="Use test discovery on the given testspec.")
+        parser.add_option("--discover-pattern", dest="discover_pattern",
+                help="The pattern that discovery should match. Default "
+                "'test*.py'", default="test*.py")
     options, args = parser.parse_args()
     stream = get_default_formatter()
     runner = SubunitTestRunner(stream)
@@ -56,7 +59,8 @@ if __name__ == '__main__':
         loader = discover.DiscoveringTestLoader()
         test = TestSuite()
         for arg in args:
-            test.addTest(loader.discover(arg))
+            test.addTest(loader.discover(
+                    arg, pattern=options.discover_pattern))
         result = runner.run(test)
         sys.exit(not result.wasSuccessful())
     program = TestProgram(module=None, argv=[sys.argv[0]] + args,