#!/usr/bin/env python
#
# Twisted, the Framework of Your Internet
# Copyright (C) 2001 Matthew W. Lefkowitz
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of version 2.1 of the GNU Lesser General Public
# License as published by the Free Software Foundation.
#
# This library 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
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

# FIXME
# - Hangs.

from twisted.python import usage, reflect
from twisted.trial import unittest
import sys, os, types

class Options(usage.Options):
    optFlags = [["help", "h"],
                ["text", "t", "Text mode (ignored)"],
                ["verbose", "v", "Verbose output"],
                ["debug", "b", "Run tests in the Python debugger"]]
    optParameters = [["reactor", "r", None,
                      "The Twisted reactor to install before running the tests (looked up as a module contained in twisted.internet)"],
                     ["logfile", "l", "test.log", "log file name"]]

    def __init__(self):
        usage.Options.__init__(self)
        self['modules'] = []
        self['packages'] = []
        self['testcases'] = []

    def opt_module(self, module):
        "Module to test"
        self['modules'].append(module)

    def opt_package(self, package):
        "Package to test"
        self['packages'].append(package)

    def opt_testcase(self, case):
        "TestCase to test"
        self['testcases'].append(case)

    def opt_file(self, filename):
        "Filename of module to test"
        from twisted.python import reflect
        self['modules'].append(reflect.filenameToModuleName(filename))

    opt_m = opt_module
    opt_p = opt_package
    opt_c = opt_testcase
    opt_f = opt_file


def run():
    if len(sys.argv) == 1:
        sys.argv.append("--help")

    config = Options()
    try:
        config.parseOptions()
    except usage.error, ue:
        print "%s: %s" % (sys.argv[0], ue)
        os._exit(1)

    if config['reactor']:
        mod = 'twisted.internet.' + config['reactor']
        print "Using %s reactor" % mod
        reflect.namedModule(mod).install()

    suite = unittest.TestSuite()
    for package in config['packages']:
        suite.addPackage(package)
    for module in config['modules']:
        suite.addModule(module)
    for testcase in config['testcases']:
        case = reflect.namedObject(testcase)
        if type(case) is types.ClassType and unittest.isTestClass(case):
            suite.addTestClass(case)
    
    testdir = "_trial_temp"
    if os.path.exists(testdir):
       import shutil
       shutil.rmtree(testdir)
    os.mkdir(testdir)
    os.chdir(testdir)

    if config['logfile']:
       from twisted.python import log
       log.startLogging(open(config['logfile'], 'a'), 0)

    if config['verbose']:
        reporter = unittest.TreeReporter(sys.stdout)
    else:
        reporter = unittest.TextReporter(sys.stdout)

    if config['debug']:
        import pdb
        pdb.run("suite.run(reporter)", globals(), locals())
    else:
        suite.run(reporter)
        sys.exit(not reporter.allPassed())

if __name__ == '__main__':
    run()