configurationengine/source/runtests.py
changeset 0 2e8eeb919028
child 3 e7e0ae78773e
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/configurationengine/source/runtests.py	Thu Mar 11 17:04:37 2010 +0200
@@ -0,0 +1,158 @@
+#
+# 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: 
+# Script for running all ConE unit tests (cone, plug-ins and scripts).
+#
+
+import os, sys, re, imp
+import unittest
+
+# Path to the directory where this file is located
+ROOT_PATH = os.path.dirname(os.path.abspath(__file__))
+
+# For module 'testautomation'
+sys.path.append(os.path.join(ROOT_PATH, 'testautomation'))
+
+from testautomation import testcli
+import cone
+import cone.public.plugin
+
+def _load_module(path):
+    if not path.endswith('.py'):
+        raise ValueError("Given parameter ('%s') is not a .py file" % path)
+    
+    dir = os.path.dirname(path)
+    sys.path.insert(0, dir)
+    try:
+        modname = path.replace('.', '_')
+        return imp.load_source(modname, path)
+    finally:
+        del sys.path[0]
+        # Since the module name __init__ is needed in many places,
+        # but its contents may differ, remove it from sys.modules
+        # in order to force reloading every time
+        if '__init__' in sys.modules:
+            del sys.modules['__init__']
+
+def _collect_unittest_suite_from_file(file_path):
+    suite = unittest.TestSuite()
+    module = _load_module(file_path)
+    suite.addTests(unittest.TestLoader().loadTestsFromModule(module))
+    return suite
+
+
+def _find_unittest_files(path, recursive=True):
+    """
+    Find all unittest_*.py files under the given directory.
+    """
+    pattern = re.compile(r'^unittest_.*\.py$')
+    unittest_files = []
+    if recursive:
+        for root, dirs, files in os.walk(path, topdown=True):
+            for filename in files:
+                if pattern.match(filename) != None:
+                    filepath = os.path.abspath(os.path.join(root, filename))
+                    unittest_files.append(filepath)
+    else:
+        for filename in os.listdir(path):
+            if pattern.match(filename) != None:
+                filepath = os.path.abspath(os.path.join(path, filename))
+                unittest_files.append(filepath)
+    
+    return unittest_files
+
+def _collect_unittest_suite_from_path(path, recursive=True):
+    """
+    Collect a test suite containing all test cases loaded from
+    unittest_*.py files in the given directory.
+    """
+    path = os.path.abspath(path)
+    
+    # Collect the list of .py files containing unit tests
+    unittest_files = _find_unittest_files(path, recursive)
+    
+    # Load the files as modules and load tests from them
+    suite = unittest.TestSuite()
+    for file in unittest_files:
+        module = _load_module(file)
+        suite.addTests(unittest.TestLoader().loadTestsFromModule(module))
+    return suite
+
+def _collect_suite():
+    def get_suite(path):
+        return _collect_unittest_suite_from_path(os.path.join(ROOT_PATH, path))
+    def get_suite_from_file(file_path):
+        return _collect_unittest_suite_from_file(os.path.join(ROOT_PATH, file_path))
+    
+    suite = unittest.TestSuite()
+    suite.addTests(get_suite('cone'))
+    suite.addTests(get_suite('scripts/tests'))
+    suite.addTests(get_suite('plugins'))
+    suite.addTests(get_suite('testautomation'))
+    
+    # Tests can also be loaded from a file:
+    #suite.addTests(get_suite_from_file('cone/public/tests/unittest_rules_on_configuration.py'))
+    
+    
+    # Force-reload all ConE plug-in reader classes, since the __init__.py
+    # files in the imported test cases have added plug-in sources to sys.path
+    cone.public.plugin.ImplFactory.force_reload_reader_classes()
+    
+    return suite
+
+def _run_without_nose():
+    suite = _collect_suite()
+    unittest.TextTestRunner(verbosity=2).run(suite)
+
+def _run_with_nose():
+    # Call plugin_utils.init_all() so that all plug-ins are loaded
+    # (otherwise script tests, plug-in unit tests and plug-in integration
+    # tests would not work)
+    PLUGIN_SOURCE_ROOT = os.path.normpath(os.path.join(ROOT_PATH, 'plugins'))
+    assert os.path.isdir(PLUGIN_SOURCE_ROOT)
+    sys.path.append(PLUGIN_SOURCE_ROOT)
+    import plugin_utils
+    plugin_utils.init_all()
+    
+    # Find all unittest_*.py files
+    test_files = []
+    def add_tests(path):
+        test_files.extend(_find_unittest_files(os.path.join(ROOT_PATH, path)))
+    add_tests('cone')
+    add_tests('scripts/tests')
+    add_tests('plugins')
+    add_tests('testautomation')
+    
+    
+    # Configure nose
+    import nose
+    plugins = nose.plugins.manager.DefaultPluginManager()
+    conf = nose.config.Config(plugins=plugins, testNames=test_files)
+    
+    # Run the tests
+    args = ['--verbosity=3',
+            '--with-xunit',
+            '--xunit-file=cone-alltests.xml',
+            #'--collect-only',
+            ]
+    nose.run(config=conf, argv=[sys.argv[0]] + args)
+
+def main():
+    if '--with-nose' in sys.argv:
+        _run_with_nose()
+    else:
+        _run_without_nose()
+
+if __name__ == '__main__':
+    main()