#!/usr/bin/env python

# Simple subunit testrunner for python
# Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2007
#   
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#   
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#   
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#

import sys

# Find right directory when running from source tree
sys.path.insert(0, "bin/python")

import optparse
import samba
samba.ensure_external_module("testtools", "testtools")
samba.ensure_external_module("subunit", "subunit/python")
from subunit.run import SubunitTestRunner
import samba.getopt as options
import samba.tests


usage = 'subunitrun [options] <tests>'
description = '''
This runs a Samba python test suite. The tests are typically located in
source4/scripting/python/samba/tests/*.py

To run the tests from one of those modules, specify the test as
samba.tests.MODULE. For example, to run the tests in common.py:

   subunitrun samba.tests.common

To list the tests in that module, use:

   subunitrun -l samba.tests.common
'''

def format_description(formatter):
    '''hack to prevent textwrap of the description'''
    return description

parser = optparse.OptionParser(usage=usage, description=description)
parser.format_description = format_description
credopts = options.CredentialsOptions(parser)
sambaopts = options.SambaOptions(parser)
parser.add_option_group(credopts)
parser.add_option_group(sambaopts)
parser.add_option('-D', '--directory', dest='directory', default=None,
                  help='set test directory')
try:
    from subunit.run import TestProgram
except ImportError:
    from unittest import TestProgram
else:
    parser.add_option('-l', '--list', dest='listtests', default=False,
                      help='List tests rather than running them.',
                      action="store_true")

opts, args = parser.parse_args()

lp = sambaopts.get_loadparm()
samba.tests.cmdline_credentials = credopts.get_credentials(lp)
if getattr(opts, "listtests", False):
    args.insert(0, "--list")

if opts.directory:
    sys.path.append(opts.directory)

runner = SubunitTestRunner()
program = TestProgram(module=None, argv=[sys.argv[0]] + args, testRunner=runner)