Files
yunohost/tests/test_user-group.py
machine-a-tsoins edb4c397df
Some checks failed
Check for new n releases / updater (push) Has been cancelled
CodeQL / Analyze (python) (push) Has been cancelled
🍽 Fork YunoHost — snapshot mangé par la machine à tsoins
Upstream: https://github.com/YunoHost/yunohost @ 3a5f8bac8301c450897b96cbd43a4c7d3ba750fb
But (José) : transformer tout le code en bions + ploxions du xerboxion.
La carte de digestion vit au labo : /yunohost-digest.json
2026-07-03 17:20:06 +00:00

387 lines
11 KiB
Python

#!/usr/bin/env python3
#
# Copyright (c) 2024 YunoHost Contributors
#
# This file is part of YunoHost (see https://yunohost.org)
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
import pytest
from yunohost.domain import _get_maindomain
from yunohost.user import (
FIELDS_FOR_IMPORT,
user_create,
user_delete,
user_export,
user_group_create,
user_group_delete,
user_group_info,
user_group_list,
user_group_update,
user_import,
user_info,
user_list,
user_update,
)
from .conftest import message, raiseYunohostError
from .test_permission import check_LDAP_db_integrity
# Get main domain
maindomain = ""
def clean_user_groups():
for u in user_list()["users"]:
user_delete(u, purge=True, force=True)
for g in user_group_list()["groups"]:
if g not in ["all_users", "visitors", "admins"]:
user_group_delete(g)
def setup_function(function):
clean_user_groups()
global maindomain
maindomain = _get_maindomain()
user_create("alice", maindomain, "test123Ynh", admin=True, fullname="Alice White")
user_create("bob", maindomain, "test123Ynh", fullname="Bob Snow")
user_create("jack", maindomain, "test123Ynh", fullname="Jack Black")
user_group_create("dev")
user_group_create("apps")
user_group_update("dev", add=["alice"])
user_group_update("apps", add=["bob"])
def teardown_function(function):
clean_user_groups()
@pytest.fixture(autouse=True)
def check_LDAP_db_integrity_call():
check_LDAP_db_integrity()
yield
check_LDAP_db_integrity()
#
# List functions
#
def test_list_users():
res = user_list()["users"]
assert "alice" in res
assert "bob" in res
assert "jack" in res
def test_list_groups():
res = user_group_list()["groups"]
assert "all_users" in res
assert "alice" in res
assert "bob" in res
assert "jack" in res
assert "alice" in res["admins"]["members"]
for u in ["alice", "bob", "jack"]:
assert u in res
assert u in res[u]["members"]
assert u in res["all_users"]["members"]
#
# Create - Remove functions
#
def test_create_user():
with message("user_created"):
user_create(
"morgan-claude.good_7",
maindomain,
"test123Ynh",
fullname="Morgan-Claude Good",
)
group_res = user_group_list()["groups"]
assert "morgan-claude.good_7" in user_list()["users"]
assert "morgan-claude.good_7" in group_res
assert "morgan-claude.good_7" in group_res["morgan-claude.good_7"]["members"]
assert "morgan-claude.good_7" in group_res["all_users"]["members"]
def test_del_user():
with message("user_deleted"):
user_delete("alice", force=True)
group_res = user_group_list()["groups"]
assert "alice" not in user_list()
assert "alice" not in group_res
assert "alice" not in group_res["all_users"]["members"]
def test_import_user():
import csv
from io import StringIO
fieldnames = [
"username",
"firstname",
"lastname",
"password",
"mailbox-quota",
"mail",
"mail-alias",
"mail-forward",
"groups",
]
with StringIO() as csv_io:
writer = csv.DictWriter(csv_io, fieldnames, delimiter=";", quotechar='"')
writer.writeheader()
writer.writerow(
{
"username": "morgan-claude.good_7",
"firstname": "Morgan-Claude",
"lastname": "Good",
"password": "",
"mailbox-quota": "1G",
"mail": "morgan-claude.good_7@" + maindomain,
"mail-alias": "morgan-claude.good_72@" + maindomain,
"mail-forward": "morgan-claude.good_7@example.com",
"groups": "dev",
}
)
writer.writerow(
{
"username": "sam",
"firstname": "Sam",
"lastname": "White",
"password": "",
"mailbox-quota": "1G",
"mail": "sam@" + maindomain,
"mail-alias": "sam1@" + maindomain + ",sam2@" + maindomain,
"mail-forward": "",
"groups": "apps",
}
)
writer.writerow(
{
"username": "alice",
"firstname": "Alice",
"lastname": "White",
"password": "",
"mailbox-quota": "1G",
"mail": "alice@" + maindomain,
"mail-alias": "",
"mail-forward": "",
"groups": "apps",
}
)
csv_io.seek(0)
with message("user_import_success"):
user_import(csv_io, update=True, delete=True)
group_res = user_group_list()["groups"]
user_res = user_list(list(FIELDS_FOR_IMPORT.keys()))["users"]
assert "morgan-claude.good_7" in user_res
assert "sam" in user_res
assert "alice" in user_res
assert "bob" not in user_res
assert len(user_res["sam"]["mail-alias"]) == 2
assert "morgan-claude.good_7" in group_res["dev"]["members"]
assert "sam" in group_res["apps"]["members"]
assert "sam" not in group_res["dev"]["members"]
assert "alice" in group_res["admins"]["members"]
assert "alice" not in group_res["apps"]["members"]
def test_export_user():
result = user_export()
should_be = (
"username;firstname;lastname;password;mail;mail-alias;mail-forward;mailbox-quota;groups\r\n"
f"alice;Alice;White;;alice@{maindomain};;;0;admins,dev\r\n"
f"bob;Bob;Snow;;bob@{maindomain};;;0;apps\r\n"
f"jack;Jack;Black;;jack@{maindomain};;;0;"
)
assert result == should_be
def test_create_group():
with message("group_created", group="volunteer-stand.2026_02"):
user_group_create("volunteer-stand.2026_02")
group_res = user_group_list()["groups"]
assert "volunteer-stand.2026_02" in group_res
assert "members" in group_res["volunteer-stand.2026_02"].keys()
assert group_res["volunteer-stand.2026_02"]["members"] == []
def test_del_group():
with message("group_deleted", group="dev"):
user_group_delete("dev")
group_res = user_group_list()["groups"]
assert "dev" not in group_res
#
# Error on create / remove function
#
def test_create_user_with_password_too_simple(mocker):
with raiseYunohostError(mocker, "password_listed"):
user_create("other", maindomain, "12", fullname="Alice White")
def test_create_user_already_exists(mocker):
with raiseYunohostError(mocker, "user_already_exists"):
user_create("alice", maindomain, "test123Ynh", fullname="Alice White")
def test_create_user_with_domain_that_doesnt_exists(mocker):
with raiseYunohostError(mocker, "domain_unknown"):
user_create("alice", "doesnt.exists", "test123Ynh", fullname="Alice White")
def test_update_user_with_mail_address_already_taken(mocker):
with raiseYunohostError(mocker, "user_update_failed"):
user_update("bob", add_mailalias="alice@" + maindomain)
def test_update_user_with_mail_address_with_unknown_domain(mocker):
with raiseYunohostError(mocker, "mail_domain_unknown"):
user_update("alice", add_mailalias="alice@doesnt.exists")
def test_del_user_that_does_not_exist(mocker):
with raiseYunohostError(mocker, "user_unknown"):
user_delete("doesnt_exist", force=True)
def test_create_group_all_users(mocker):
# Check groups already exist with special group "all_users"
with raiseYunohostError(mocker, "group_already_exist"):
user_group_create("all_users")
def test_create_group_already_exists(mocker):
# Check groups already exist (regular groups)
with raiseYunohostError(mocker, "group_already_exist"):
user_group_create("dev")
def test_del_group_all_users(mocker):
with raiseYunohostError(mocker, "group_cannot_be_deleted"):
user_group_delete("all_users")
def test_del_group_that_does_not_exist(mocker):
with raiseYunohostError(mocker, "group_unknown"):
user_group_delete("doesnt_exist")
#
# Update function
#
def test_update_user():
with message("user_updated"):
user_update("alice", fullname="New2Name New2Last")
info = user_info("alice")
assert info["fullname"] == "New2Name New2Last"
def test_update_group_add_user():
with message("group_updated", group="dev"):
user_group_update("dev", add=["bob"])
group_res = user_group_list()["groups"]
assert set(group_res["dev"]["members"]) == {"alice", "bob"}
def test_update_group_add_user_already_in():
with message("group_user_already_in_group", user="bob", group="apps"):
user_group_update("apps", add=["bob"])
group_res = user_group_list()["groups"]
assert group_res["apps"]["members"] == ["bob"]
def test_update_group_remove_user():
with message("group_updated", group="apps"):
user_group_update("apps", remove=["bob"])
group_res = user_group_list()["groups"]
assert group_res["apps"]["members"] == []
def test_update_group_remove_user_not_already_in():
with message("group_user_not_in_group", user="jack", group="apps"):
user_group_update("apps", remove=["jack"])
group_res = user_group_list()["groups"]
assert group_res["apps"]["members"] == ["bob"]
#
# Error on update functions
#
def test_update_user_that_doesnt_exist(mocker):
with raiseYunohostError(mocker, "user_unknown"):
user_update("doesnt_exist", fullname="Foo Bar")
def test_update_group_that_doesnt_exist(mocker):
with raiseYunohostError(mocker, "group_unknown"):
user_group_update("doesnt_exist", add=["alice"])
def test_update_group_all_users_manually(mocker):
with raiseYunohostError(mocker, "group_cannot_edit_all_users"):
user_group_update("all_users", remove=["alice"])
assert "alice" in user_group_list()["groups"]["all_users"]["members"]
def test_update_group_primary_manually(mocker):
with raiseYunohostError(mocker, "group_cannot_edit_primary_group"):
user_group_update("alice", remove=["alice"])
assert "alice" in user_group_list()["groups"]["alice"]["members"]
def test_update_group_add_user_that_doesnt_exist(mocker):
with raiseYunohostError(mocker, "user_unknown"):
user_group_update("dev", add=["doesnt_exist"])
assert "doesnt_exist" not in user_group_list()["groups"]["dev"]["members"]
def test_update_group_remove_last_admin(mocker):
with raiseYunohostError(mocker, "group_cannot_remove_last_admin"):
user_group_update("admins", remove=["alice"])
assert "alice" in user_group_info("admins")["members"]