Test multiple admins on install

This commit is contained in:
GeorgianaElena
2019-07-15 12:02:18 +03:00
parent 989eda8906
commit 55e9aa2c29

View File

@@ -4,6 +4,7 @@ Unit test functions in installer.py
import os
from tljh import installer
from tljh.yaml import yaml
def test_ensure_node():
@@ -19,3 +20,16 @@ def test_ensure_config_yaml(tljh_dir):
assert os.path.isdir(os.path.join(installer.CONFIG_DIR, 'jupyterhub_config.d'))
# verify that old config doesn't exist
assert not os.path.exists(os.path.join(tljh_dir, 'config.yaml'))
def test_ensure_admins(tljh_dir):
# --admin option called multiple times on the installer
# creates a list of argument lists.
admins = [['a1'], ['a2'], ['a3']]
installer.ensure_admins(admins)
config_path = installer.CONFIG_FILE
with open(config_path, 'r') as f:
config = yaml.load(f)
# verify the list was flattened
assert config['users']['admin'] == ['a1', 'a2', 'a3']