#
# Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).
# All rights reserved.
# This component and the accompanying materials are made available
# under the terms of "Eclipse Public License v1.0"
# which accompanies this distribution, and is available
# at the URL "http://www.eclipse.org/legal/epl-v10.html".
#
# Initial Contributors:
# Nokia Corporation - initial contribution.
#
# Contributors:
#
# Description: 
#

import sys, os, re, unittest

ROOT_PATH = os.path.abspath(os.path.dirname(__file__))

CONE_PATH = os.path.join(ROOT_PATH, 'cone')
EGGS_PATH = os.path.join(ROOT_PATH, 'eggs')

def add_eggs_to_path(egg_path):
    if not os.path.isdir(egg_path):
        return
    
    for name in os.listdir(egg_path):
        if name.endswith('.egg'):
            path = os.path.normpath(os.path.join(egg_path, name))
            if path not in sys.path:
                print "Adding '%s' to path" % name
                sys.path.append(path)


def _collect_test_suite(base_dir, module_subdir):
    # Check that the directory exists
    module_dir = os.path.join(base_dir, module_subdir)
    if not os.path.exists(module_dir):
        raise RuntimeError("'%s' does not exist!" % module_dir)
    
    # Collect the names of all test modules (of the form "unittest_*.py")
    test_modules_names = []
    for name in os.listdir(module_dir):
        if re.match(r'^unittest_.*\.py$', name) != None:
            test_modules_names.append(name[:-3])
    
    # Import the modules
    sys.path.insert(0, base_dir)
    try:
        suite = unittest.TestSuite()
        for modname in test_modules_names:
            # Load the test module dynamically and add it to the test suite
            top_module = __import__(module_subdir + '.' + modname)
            
            # top_module contains now actually e.g. mymodule.unittest_sometest,
            # so get the actual unit test module
            module = getattr(top_module, modname)
            
            suite.addTests(unittest.TestLoader().loadTestsFromModule(module))
        return suite
    finally:
        del sys.path[0]

def _run_without_nose():
    suite = unittest.TestSuite()
    
    # Collect script test suite
    suite.addTest(_collect_test_suite(ROOT_PATH, 'tests'))
    
    # Collect test suites from the plugin-tests/ directory
    PLUGIN_TEST_DIR = os.path.join(ROOT_PATH, 'plugin-tests')
    if os.path.exists(PLUGIN_TEST_DIR):
        for name in os.listdir(PLUGIN_TEST_DIR):
            plugin_suite = _collect_test_suite(PLUGIN_TEST_DIR, name)
            suite.addTest(plugin_suite)
    
    # Run the tests
    unittest.TextTestRunner(verbosity=2).run(suite)

def _run_with_nose():
    import nose
    import nose.plugins.xunit
    
    # Configure nose
    plugins = nose.plugins.manager.DefaultPluginManager()
    allfiles = nose.config.all_config_files() + ['nose_bat.cfg']
    conf = nose.config.Config(files=allfiles, plugins=plugins)
    
    # Run the tests
    nose.run(config=conf, argv=sys.argv[:1])

def main():
    if 'CONE_PATH' not in os.environ:
        cone_path = os.path.join(ROOT_PATH, 'cone')
        print "Setting CONE_PATH to '%s'" % cone_path
        os.environ['CONE_PATH'] = cone_path
    
    add_eggs_to_path(EGGS_PATH)
    
    if '--with-nose' in sys.argv:
        _run_with_nose()
    else:
        _run_without_nose()
    return 0

if __name__ == "__main__":
    sys.exit(main())