kitty_tests/ssh.py: skip login shell detection if getpwuid() fails

This commit is contained in:
Samuel Tam 2023-03-19 18:45:47 +08:00 committed by GitHub
parent fda2646dd3
commit 31319f0b65
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -173,7 +173,11 @@ env COLORTERM
methods.append('using_passwd')
self.assertTrue(methods)
import pwd
expected_login_shell = pwd.getpwuid(os.geteuid()).pw_shell
try:
expected_login_shell = pwd.getpwuid(os.geteuid()).pw_shell
except KeyError:
with suppress(Exception):
self.skipTest('Skipping login shell detection as getpwuid() failed to read login shell')
if os.path.basename(expected_login_shell) == 'nologin':
self.skipTest('Skipping login shell detection as login shell is set to nologin')
for m in methods: