Add option for generating coverage data from python tests.

1 files changed, 8 insertions, 0 deletions

diff --git a/source/scripting/bin/subunitrun b/source/scripting/bin/subunitrun
index 6f1086ad376..ee2d1e11da6 100755
--- a/source/scripting/bin/subunitrun
+++ b/source/scripting/bin/subunitrun
@@ -36,6 +36,8 @@ parser.add_option_group(credopts)
 sambaopts = options.SambaOptions(parser)
 parser.add_option_group(sambaopts)
 parser.add_option_group(options.VersionOptions(parser))
+parser.add_option("--coverage", metavar="CACHE", type=str, 
+		          help="Store coverage data in CACHE")
 
 args = parser.parse_args()[1]
 
@@ -45,4 +47,10 @@ samba.tests.cmdline_credentials = credopts.get_credentials(samba.tests.cmdline_l
 param.cvar.default_config = samba.tests.cmdline_loadparm
 
 runner = SubunitTestRunner()
+if opts.coverage is not None:
+	import coverage
+	coverage.use_cache(True, opts.coverage)
+	coverage.start()
 program = TestProgram(module=None, argv=[sys.argv[0]] + args, testRunner=runner)
+if opts.coverage:
+	coverage.stop()