Skip to content

Commit

Permalink
Add test for dict merging
Browse files Browse the repository at this point in the history
  • Loading branch information
yuvipanda committed May 24, 2024
1 parent fb85cfb commit 8564ff0
Showing 1 changed file with 32 additions and 0 deletions.
32 changes: 32 additions & 0 deletions jupyterhub/tests/test_spawner.py
Original file line number Diff line number Diff line change
Expand Up @@ -656,6 +656,38 @@ async def test_group_override_lexical_ordering(app):
assert s.start_timeout == 300


async def test_group_override_dict_merging(app):
app.load_groups = {
"admin": {"users": ["admin"]},
"user": {"users": ["admin", "user"]},
}
await app.init_groups()

group_overrides = {
"01-admin-env-add": {
"groups": ["admin"],
"spawner_override": {"environment": {"AM_I_ADMIN": "yes"}},
},
"02-user-env-add": {
"groups": ["user"],
"spawner_override": {"environment": {"AM_I_USER": "yes"}},
},
}

admin_user = find_user(app.db, "admin")
s = Spawner(user=admin_user)
s.group_overrides = group_overrides
await s.apply_group_overrides()
assert s.environment["AM_I_ADMIN"] == "yes"
assert s.environment["AM_I_USER"] == "yes"

admin_user = find_user(app.db, "user")
s = Spawner(user=admin_user)
s.group_overrides = group_overrides
await s.apply_group_overrides()
assert s.environment["AM_I_USER"] == "yes"
assert "AM_I_ADMIN" not in s.environment

async def test_group_override_callable(app):
app.load_groups = {
"admin": {"users": ["admin"]},
Expand Down

0 comments on commit 8564ff0

Please sign in to comment.