zhangyiss/the-littlest-jupyterhub
1
0
Fork 0
mirror of https://github.com/jupyterhub/the-littlest-jupyterhub.git synced 2025-12-18 21:54:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

add test

Browse Source
This commit is contained in:
Matthias Bussonnier
2018-08-31 18:04:06 -07:00
parent 552db9f74d
commit 5bbba2bd7c
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/test_user.py
View File

@@ -3,6 +3,8 @@ Test wrappers in tljw.user module
""" """
from tljh import user from tljh import user
import os import os
import os.path
import stat
import uuid import uuid
import pwd import pwd
import grp import grp
@@ -23,9 +25,16 @@ def test_ensure_user():
# Create user! # Create user!
user.ensure_user(username) user.ensure_user(username)
# This raises exception if user doesn't exist # This raises exception if user doesn't exist
ent = pwd.getpwnam(username) entry = pwd.getpwnam(username)
# Home directory must also exist # Home directory must also exist
assert os.path.exists(ent.pw_dir) home_dir = entry.pw_dir
assert os.path.exists(home_dir)
# Ensure not word readable/writable especially in teaching context
homedir_stats = os.stat(home_dir).st_mode
assert not (homedir_stats & stat.S_IROTH), "Everyone should not be able to read users home directory"
assert not (homedir_stats & stat.S_IWOTH), "Everyone should not be able to write users home directory"
assert not (homedir_stats & stat.S_IXOTH), "Everyone should not be able to list what is in users home directory"
# Run ensure_user again, should be a noop # Run ensure_user again, should be a noop
user.ensure_user(username) user.ensure_user(username)
# User still exists, after our second ensure_user call # User still exists, after our second ensure_user call