]>
git.ipfire.org Git - thirdparty/systemd.git/blob - test/test-shutdown.py
2 # SPDX-License-Identifier: LGPL-2.1-or-later
3 # pylint: disable=broad-except
17 logger
= logging
.getLogger("test-shutdown")
21 logger
.debug("Logging pexpect IOs to %s", args
.logfile
)
22 logfile
= open(args
.logfile
, 'w')
26 logger
.info("spawning test")
27 console
= pexpect
.spawn(args
.command
, args
.arg
, logfile
=logfile
, env
={
29 }, encoding
='utf-8', timeout
=60)
31 logger
.debug("child pid %d", console
.pid
)
34 logger
.info("waiting for login prompt")
35 console
.expect('H login: ', 10)
37 logger
.info("log in and start screen")
38 console
.sendline('root')
39 console
.expect('bash.*# ', 10)
40 console
.sendline('screen')
41 console
.expect('screen0 ', 10)
42 console
.sendcontrol('a')
44 console
.expect('screen1 ', 10)
48 console
.sendline('tty')
49 console
.expect(r
'/dev/(pts/\d+)')
50 pty
= console
.match
.group(1)
51 logger
.info("window 1 at tty %s", pty
)
53 logger
.info("schedule reboot")
54 console
.sendline('shutdown -r')
55 console
.expect("Reboot scheduled for (?P<date>.*), use 'shutdown -c' to cancel", 2)
56 date
= console
.match
.group('date')
57 logger
.info("reboot scheduled for %s", date
)
59 console
.sendcontrol('a')
61 logger
.info("verify broadcast message")
62 console
.expect(f
'Broadcast message from root@H on {pty}', 2)
63 console
.expect(f
'The system will reboot at {date}', 2)
65 logger
.info("check show output")
66 console
.sendline('shutdown --show')
67 console
.expect(f
"Reboot scheduled for {date}, use 'shutdown -c' to cancel", 2)
69 logger
.info("cancel shutdown")
70 console
.sendline('shutdown -c')
71 console
.sendcontrol('a')
73 console
.expect('System shutdown has been cancelled', 2)
75 logger
.info("call for reboot")
76 console
.sendline('sleep 10; shutdown -r now')
77 console
.sendcontrol('a')
79 console
.expect("The system will reboot now!", 12)
81 logger
.info("waiting for reboot")
83 console
.expect('H login: ', 60)
84 console
.sendline('root')
85 console
.expect('bash.*# ', 10)
87 console
.sendline('> /testok')
89 logger
.info("power off")
90 console
.sendline('poweroff')
92 logger
.info("expect termination now")
93 console
.expect(pexpect
.EOF
)
96 except Exception as e
:
98 logger
.info("killing child pid %d", console
.pid
)
100 # Ask systemd-nspawn to stop and release the container's resources properly.
101 console
.kill(signal
.SIGTERM
)
104 if not console
.isalive():
109 # We haven't exited the loop early, so check if the process is
110 # still alive - if so, force-kill it.
111 if console
.isalive():
112 console
.terminate(force
=True)
117 parser
= argparse
.ArgumentParser(description
='test logind shutdown feature')
118 parser
.add_argument("-v", "--verbose", action
="store_true", help="verbose")
119 parser
.add_argument("--logfile", metavar
='FILE', help="Save all test input/output to the given path")
120 parser
.add_argument("command", help="command to run")
121 parser
.add_argument("arg", nargs
='*', help="args for command")
123 args
= parser
.parse_args()
126 level
= logging
.DEBUG
130 logging
.basicConfig(level
=level
)
134 if __name__
== '__main__':