Step 2: Move test folder to root

[integration/test.git] / tools / CSIT_Test / base / run.py

diff --git a/tools/CSIT_Test/base/run.py b/tools/CSIT_Test/base/run.py
new file mode 100644 (file)
index 0000000..28aab41
--- /dev/null
+++ b/tools/CSIT_Test/base/run.py
@@ -0,0 +1,48 @@
+#!/usr/bin/env python
+"""
+CSIT test tools.
+Authors: Baohua Yang@IBM, Denghui Huang@IBM
+Updated: 2013-11-07
+
+Usage: Before running the test tool, should
+ 1. Start 2-layer tree topology network. e.g., in Mininet, run
+    'sudo mn --controller=remote,ip=127.0.0.1 --mac --topo tree,2'.
+ 2. Configure gateway in the controller web GUI, name = 'gateway', subnet = '10.0.0.254/24'.
+ 3. In Mininet, run 'h1 ping h2' to make sure the network is connected.
+"""
+import doctest
+import os
+from restlib import *  # noqa
+
+
+def test_module(module_name):
+    '''
+    Run single test on given module.
+    '''
+    print "#Test case: " + module_name.replace('_', ' ')
+    cmd = 'python -m doctest ' + module_name + '.py'
+    os.system(cmd)
+
+
+def run(modules=None):
+    '''
+    Run test cases according to the given modules.
+    If no parameter is given, then will scan the case directory,
+     and try to run all cases.
+    '''
+    backup_dir = os.getcwd()
+    if not modules:
+        modules = [e[:-3] for e in os.listdir(MODULES_DIR) if e.endswith('.py')]
+    os.chdir(backup_dir + '/' + MODULES_DIR)
+    for name in modules:
+        test_module(name)
+    os.chdir(backup_dir)
+
+
+if __name__ == '__main__':
+    doctest.testmod()
+    test_modules = ['switch_manager', 'topology_manager', 'forwarding_rule_manager', 'statistics_manager',
+                    'host_tracker', 'arp_handler', 'forwarding_manager', 'container_manager']
+    # test_modules = ['topology_manager']
+    run(test_modules)
+    # run()