X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/83c835b2eb41598eec5f17726a4cb15316604d27..e612178e5914b6c6cb9ab947e5a433f9a498ca22:/tools/tesh/tesh.py

diff --git a/tools/tesh/tesh.py b/tools/tesh/tesh.py
index 5f011b40f0..7f3aca4df6 100755
--- a/tools/tesh/tesh.py
+++ b/tools/tesh/tesh.py
@@ -48,11 +48,8 @@ import argparse
 if sys.version_info[0] == 3:
     import subprocess
     import _thread
-elif sys.version_info[0] < 3:
-    import subprocess32 as subprocess
-    import thread as _thread
 else:
-    raise "This program has not been made to exist this long"
+    raise "This program is expected to run with Python3 only"
 
 
 
@@ -165,6 +162,7 @@ class TeshState(Singleton):
         self.threads = []
         self.args_suffix = ""
         self.ignore_regexps_common = []
+        self.wrapper = None
     
     def add_thread(self, thread):
         self.threads.append(thread)
@@ -174,9 +172,6 @@ class TeshState(Singleton):
             t.acquire()
             t.release()
 
-
-
-
 #Command line object
 class Cmd(object):
     def __init__(self):
@@ -227,6 +222,7 @@ class Cmd(object):
         if file is None:
             fatal_error("Unable to create file "+filename)
         file.write("\n".join(self.input_pipe))
+        file.write("\n")
         file.close()
 
     def _cmd_cd(self, argline):
@@ -289,10 +285,14 @@ class Cmd(object):
             if lock is not None: lock.release()
             return
         
+        if TeshState().wrapper is not None:
+            self.timeout *= 20
+            self.args = TeshState().wrapper + self.args
+            
         self.args += TeshState().args_suffix
         
         print("["+FileReader().filename+":"+str(self.linenumber)+"] "+self.args)
-        
+                
         args = shlex.split(self.args)
         #print (args)
         try:
@@ -365,12 +365,6 @@ class Cmd(object):
 
 
 
-
-
-
-
-
-
 ##############
 #
 # Main
@@ -389,6 +383,7 @@ if __name__ == '__main__':
     group1.add_argument('--cfg', metavar='arg', help='add parameter --cfg=arg to each command line')
     group1.add_argument('--log', metavar='arg', help='add parameter --log=arg to each command line')
     group1.add_argument('--enable-coverage', action='store_true', help='ignore output lines starting with "profiling:"')
+    group1.add_argument('--wrapper', metavar='arg', help='Run each command in the provided wrapper (eg valgrind)')
 
     try:
         options = parser.parse_args()
@@ -417,7 +412,9 @@ if __name__ == '__main__':
         TeshState().args_suffix += " --cfg="+options.cfg
     if options.log is not None:
         TeshState().args_suffix += " --log="+options.log
-    
+
+    if options.wrapper is not None:
+        TeshState().wrapper = options.wrapper
     
     #cmd holds the current command line
     # tech commands will add some parameters to it
@@ -508,8 +505,3 @@ if __name__ == '__main__':
         print("Test suite from stdin OK")
     else:
         print("Test suite `"+f.filename+"' OK")
-
-
-
-
-