Changes in tests/pybin/tools.py [d65f92c:143e6f3]

File:

• tests/pybin/tools.py (modified) (13 diffs)

tests/pybin/tools.py

@@ -2 +2 @@
 import argparse
 import contextlib
+import datetime
 import fileinput
 import multiprocessing
@@ -22 +23 @@
 
 # helper functions to run terminal commands
-def sh(*cmd, timeout = False, output_file = None, input_file = None, input_text = None, error = subprocess.STDOUT):
+def sh(*cmd, timeout = False, output = None, input = None, error = subprocess.STDOUT):
         cmd = list(cmd)
-
-        if input_file and input_text:
-                return 401, "Cannot use both text and file inputs"
 
         # if this is a dry_run, only print the commands that would be ran
         if settings.dry_run :
                 cmd = "{} cmd: {}".format(os.getcwd(), ' '.join(cmd))
-                if output_file and not isinstance(output_file, int):
+                if output and not isinstance(output, int):
                         cmd += " > "
-                        cmd += output_file
+                        cmd += output
 
                 if error and not isinstance(error, int):
@@ -39 +37 @@
                         cmd += error
 
-                if input_file and not isinstance(input_file, int) and os.path.isfile(input_file):
+                if input and not isinstance(input, int) and os.path.isfile(input):
                         cmd += " < "
-                        cmd += input_file
+                        cmd += input
 
                 print(cmd)
@@ -48 +46 @@
         with contextlib.ExitStack() as onexit:
                 # add input redirection if needed
-                input_file = openfd(input_file, 'r', onexit, True)
+                input = openfd(input, 'r', onexit, True)
 
                 # add output redirection if needed
-                output_file = openfd(output_file, 'w', onexit, False)
+                output = openfd(output, 'w', onexit, False)
 
                 # add error redirection if needed
@@ -60 +58 @@
                         proc = subprocess.run(
                                 cmd,
-                                **({'input' : bytes(input_text, encoding='utf-8')} if input_text else {'stdin' : input_file}),
-                                stdout  = output_file,
-                                stderr  = error,
-                                timeout = settings.timeout.single if timeout else None
+                                stdin =input,
+                                stdout=output,
+                                stderr=error,
+                                timeout=settings.timeout.single if timeout else None
                         )
-
                         return proc.returncode, proc.stdout.decode("utf-8") if proc.stdout else None
                 except subprocess.TimeoutExpired:
@@ -78 +75 @@
                 return False
 
-        code, out = sh("file %s" % fname, output_file=subprocess.PIPE)
+        code, out = sh("file %s" % fname, output=subprocess.PIPE)
         if code != 0:
                 return False
@@ -110 +107 @@
         if isinstance(files, str ): files = [ files ]
         for file in files:
-                sh( 'rm', '-f', file, output_file=subprocess.DEVNULL, error=subprocess.DEVNULL )
+                sh( 'rm', '-f', file, output=subprocess.DEVNULL, error=subprocess.DEVNULL )
 
 # Create 1 or more directory
@@ -118 +115 @@
                 p = os.path.normpath( file )
                 d = os.path.dirname ( p )
-                sh( 'mkdir', '-p', d, output_file=subprocess.DEVNULL, error=subprocess.DEVNULL )
+                sh( 'mkdir', '-p', d, output=subprocess.DEVNULL, error=subprocess.DEVNULL )
 
 
@@ -141 +138 @@
                 lhs,
                 rhs,
-                output_file=subprocess.PIPE
+                output=subprocess.PIPE
         )
 
 # call make
-def make(target, *, flags = '', output_file = None, error = None, error_file = None, silent = False):
+def make(target, *, flags = '', output = None, error = None, error_file = None, silent = False):
         test_param = """test="%s" """ % (error_file) if error_file else None
         cmd = [
@@ -154 +151 @@
                 settings.debug.flags,
                 settings.install.flags,
-                settings.distcc if settings.distribute else None,
                 flags,
                 target
         ]
         cmd = [s for s in cmd if s]
-        return sh(*cmd, output_file=output_file, error=error)
+        return sh(*cmd, output=output, error=error)
 
 def which(program):
@@ -205 +201 @@
 # cat one file into the other
 def cat(source, dest):
-        ret, _ = sh("cat", source, output_file=dest)
+        ret, _ = sh("cat", source, output=dest)
         return ret
 
@@ -278 +274 @@
 ################################################################################
 
-# get hash for given configuration
-def config_hash():
-        path = os.path.normpath(os.path.join(
-                settings.SRCDIR,
-        ))
-
-        distcc_hash = os.path.join(settings.SRCDIR, '../tools/build/distcc_hash')
-        config = "%s-%s" % (settings.arch.target, settings.debug.path)
-        _, out = sh(distcc_hash, config, output_file=subprocess.PIPE)
-        return out.strip()
+def pretty_now():
+        ts = time.time()
+        print(ts, file=sys.stderr)
+        return datetime.datetime.fromtimestamp(ts).strftime('%Y-%m-%d_%H:%M:%S')
 
 # check if arguments is yes or no
@@ -318 +308 @@
                 return 1, "ERR No core dump"
 
-        return sh('gdb', '-n', path, core, '-batch', '-x', cmd, output_file=subprocess.PIPE)
+        return sh('gdb', '-n', path, core, '-batch', '-x', cmd, output=subprocess.PIPE)
+
+def core_archive(dst, name, exe):
+        # Get the core dump
+        core = os.path.join(os.getcwd(), "core" )
+
+        # update the path for this test
+        dst  = os.path.join(dst, name)
+
+        # make a directory for this test
+        # mkdir makes the parent directory only so add a dummy
+        mkdir(os.path.join(dst, "dir"))
+
+        # moves the files
+        mv( core, os.path.join(dst, "core" ) )
+        mv( exe , os.path.join(dst, name   ) )
+
+        # return explanatory test
+        return "Archiving %s (executable and core) to %s" % (os.path.relpath(exe, settings.BUILDDIR), os.path.relpath(dst, settings.original_path))
 
 class Timed: