canaille-globuzma/tests/backends/test_models.py

from canaille.core.models import Group
from canaille.core.models import User


def test_model_comparison(testclient, slapd_connection):
    foo1 = User(
        user_name="foo",
        family_name="foo",
        formatted_name="foo",
    )
    foo2 = User(
        user_name="foo",
        family_name="foo",
        formatted_name="foo",
    )
    bar = User(
        user_name="bar",
        family_name="bar",
        formatted_name="foo",
    )

    assert foo1 == foo2
    assert foo1 != bar


def test_model_lifecycle(testclient, slapd_connection, slapd_server):
    user = User(
        user_name="user_name",
        family_name="family_name",
        formatted_name="formatted_name",
    )

    assert not User.query()
    assert not User.query(id=user.id)
    assert not User.query(id="invalid")
    assert not User.get(id=user.id)

    user.save()

    assert User.query() == [user]
    assert User.query(id=user.id) == [user]
    assert not User.query(id="invalid")
    assert User.get(id=user.id) == user

    user.family_name = "new_family_name"

    assert user.family_name == ["new_family_name"]

    user.reload()

    assert user.family_name == ["family_name"]

    user.delete()

    assert not User.query(id=user.id)
    assert not User.get(id=user.id)


def test_model_attribute_edition(testclient, slapd_connection):
    user = User(
        user_name="user_name",
        family_name="family_name",
        formatted_name="formatted_name",
        display_name="display_name",
        email=["email1@user.com", "email2@user.com"],
    )
    user.save()

    assert user.user_name == ["user_name"]
    assert user.family_name == ["family_name"]
    assert user.email == ["email1@user.com", "email2@user.com"]

    user = User.get(id=user.id)
    assert user.user_name == ["user_name"]
    assert user.family_name == ["family_name"]
    assert user.email == ["email1@user.com", "email2@user.com"]

    user.family_name = ["new_family_name"]
    user.email = ["email1@user.com"]
    user.save()

    assert user.family_name == ["new_family_name"]
    assert user.email == ["email1@user.com"]

    user = User.get(id=user.id)
    assert user.family_name == ["new_family_name"]
    assert user.email == ["email1@user.com"]

    user.display_name = [""]
    user.save()

    assert not user.display_name

    user.delete()


def test_model_indexation(testclient, slapd_connection):
    user = User(
        user_name="user_name",
        family_name="family_name",
        formatted_name="formatted_name",
        email=["email1@user.com", "email2@user.com"],
    )
    user.save()

    assert User.get(family_name="family_name") == user
    assert not User.get(family_name="new_family_name")
    assert User.get(email="email1@user.com") == user
    assert User.get(email="email2@user.com") == user
    assert not User.get(email="email3@user.com")

    user.family_name = "new_family_name"
    user.email = ["email2@user.com"]

    assert User.get(family_name="family_name") != user
    assert not User.get(family_name="new_family_name")
    assert User.get(email="email1@user.com") != user
    assert User.get(email="email2@user.com") != user
    assert not User.get(email="email3@user.com")

    user.save()

    assert not User.get(family_name="family_name")
    assert User.get(family_name="new_family_name") == user
    assert not User.get(email="email1@user.com")
    assert User.get(email="email2@user.com") == user
    assert not User.get(email="email3@user.com")

    user.delete()

    assert not User.get(family_name="family_name")
    assert not User.get(family_name="new_family_name")
    assert not User.get(email="email1@user.com")
    assert not User.get(email="email2@user.com")
    assert not User.get(email="email3@user.com")


def test_fuzzy(user, moderator, admin, slapd_connection):
    assert set(User.query()) == {user, moderator, admin}
    assert set(User.fuzzy("Jack")) == {moderator}
    assert set(User.fuzzy("Jack", ["formatted_name"])) == {moderator}
    assert set(User.fuzzy("Jack", ["user_name"])) == set()
    assert set(User.fuzzy("Jack", ["user_name", "formatted_name"])) == {moderator}
    assert set(User.fuzzy("moderator")) == {moderator}
    assert set(User.fuzzy("oderat")) == {moderator}
    assert set(User.fuzzy("oDeRat")) == {moderator}
    assert set(User.fuzzy("ack")) == {moderator}


# def test_model_references(user, admin, foo_group, bar_group):
def test_model_references(
    testclient, user, foo_group, admin, bar_group, slapd_connection
):
    assert foo_group.members == [user]
    assert user.groups == [foo_group]
    assert foo_group in Group.query(members=user)
    assert user in User.query(groups=foo_group)

    assert user not in bar_group.members
    assert bar_group not in user.groups
    user.groups = user.groups + [bar_group]
    user.save()
    bar_group.reload()

    assert user in bar_group.members
    assert bar_group in user.groups

    bar_group.members = [admin]
    bar_group.save()
    user.reload()

    assert user not in bar_group.members
    assert bar_group not in user.groups


def test_model_references_set_unsaved_object(
    testclient, logged_moderator, user, slapd_connection
):
    group = Group(members=[user], display_name="foo")
    group.save()
    user.reload()  # an LDAP group can be inconsistent by containing members which doesn't exist

    non_existent_user = User(formatted_name="foo", family_name="bar")
    group.members = group.members + [non_existent_user]
    assert group.members == [user, non_existent_user]

    group.save()
    assert group.members == [user, non_existent_user]

    group.reload()
    assert group.members == [user]

    testclient.get("/groups/foo", status=200)

    group.delete()