From 66b472321623e2ae9d88be46a1f232a900a7352a Mon Sep 17 00:00:00 2001 From: James Prestwood Date: Tue, 2 Mar 2021 08:30:29 -0800 Subject: [PATCH] test-runner: fix duplicate process output Process output was being duplicated when -v was used. This was due to both stderr and stdout being appended to the write_fd list as well as stderr being set to stdout in the Popen call. To fix this only stdout should be appended to the write_fd list, but then there comes a problem with closing the streams. stdout cannot be closed, so instead it is special cased. A new verbose boolean was added to Process which, if True, will cause any output to be written to stdout explicitly. --- tools/test-runner | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/tools/test-runner b/tools/test-runner index eab0f0ab..a0c4022e 100755 --- a/tools/test-runner +++ b/tools/test-runner @@ -174,6 +174,7 @@ class Process: self.write_fds = [] self.io_watch = None self.cleanup = cleanup + self.verbose = False if not namespace: self.output_name = '/tmp/%s-out' % self.name @@ -195,8 +196,7 @@ class Process: if ctx: # Verbose requested, add stdout/stderr to write FD list if self.name in ctx.args.verbose: - self.write_fds.append(sys.__stdout__) - self.write_fds.append(sys.__stderr__) + self.verbose = True # Add output file to FD list if outfile: @@ -230,11 +230,11 @@ class Process: # the process is being waited for, the log/outfile bits # will be handled after the process exists. # - if self.write_fds != [] and not wait and not check: + if self.write_fds != [] and not wait and not check or self.verbose: self.io_watch = GLib.io_add_watch(self.stdout, GLib.IO_IN, self.io_callback) - self.pid = subprocess.Popen(self.args, stdout=self.stdout, stderr=subprocess.STDOUT, + self.pid = subprocess.Popen(self.args, stdout=self.stdout, stderr=self.stdout, env=env, cwd=os.getcwd()) print("Starting process {}".format(self.pid.args)) @@ -261,6 +261,9 @@ class Process: self.write_fds = [] + if self.verbose: + sys.__stdout__.write(self.out) + print("%s returned %d" % (args[0], self.ret)) if check and self.ret != 0: raise subprocess.CalledProcessError(returncode=self.ret, cmd=self.args) @@ -284,6 +287,9 @@ class Process: for f in self.write_fds: f.write(data) + if self.verbose: + sys.__stdout__.write(data) + return True def __del__(self):