diff options
| -rwxr-xr-x | test/test-shutdown.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/test/test-shutdown.py b/test/test-shutdown.py index e491f1e1a9..f496122f80 100755 --- a/test/test-shutdown.py +++ b/test/test-shutdown.py @@ -21,15 +21,18 @@ def run(args): "TERM": "linux", }, encoding='utf-8', timeout=60) - if args.verbose: - console.logfile = sys.stdout - logger.debug("child pid %d", console.pid) try: logger.info("waiting for login prompt") console.expect('H login: ', 10) + if args.logfile: + logger.debug("Logging pexpect IOs to %s", args.logfile) + console.logfile = open(args.logfile, 'w') + elif args.verbose: + console.logfile = sys.stdout + logger.info("log in and start screen") console.sendline('root') console.expect('bash.*# ', 10) @@ -44,7 +47,7 @@ def run(args): console.sendline('tty') console.expect(r'/dev/(pts/\d+)') pty = console.match.group(1) - logger.info("window 1 at line %s", pty) + logger.info("window 1 at tty %s", pty) logger.info("schedule reboot") console.sendline('shutdown -r') @@ -112,6 +115,7 @@ def run(args): def main(): parser = argparse.ArgumentParser(description='test logind shutdown feature') parser.add_argument("-v", "--verbose", action="store_true", help="verbose") + parser.add_argument("--logfile", metavar='FILE', help="Save all test input/output to the given path") parser.add_argument("command", help="command to run") parser.add_argument("arg", nargs='*', help="args for command") |