forked from Github-Mirrors/canaille
222 lines
7.3 KiB
Python
222 lines
7.3 KiB
Python
import datetime
|
|
|
|
import pytest
|
|
import time_machine
|
|
|
|
from canaille.app import models
|
|
|
|
|
|
def test_model_comparison(testclient, backend):
|
|
foo1 = models.User(
|
|
user_name="foo",
|
|
family_name="foo",
|
|
formatted_name="foo",
|
|
)
|
|
backend.save(foo1)
|
|
bar = models.User(
|
|
user_name="bar",
|
|
family_name="bar",
|
|
formatted_name="bar",
|
|
)
|
|
backend.save(bar)
|
|
foo2 = backend.get(models.User, id=foo1.id)
|
|
|
|
assert foo1 == foo2
|
|
assert foo1 != bar
|
|
|
|
backend.delete(foo1)
|
|
backend.delete(bar)
|
|
|
|
|
|
def test_model_lifecycle(testclient, backend):
|
|
user = models.User(
|
|
user_name="user_name",
|
|
family_name="family_name",
|
|
formatted_name="formatted_name",
|
|
)
|
|
|
|
assert not user.id
|
|
assert not backend.query(models.User)
|
|
assert not backend.query(models.User, id=user.id)
|
|
assert not backend.query(models.User, id="invalid")
|
|
assert not backend.get(models.User, id=user.id)
|
|
|
|
backend.save(user)
|
|
|
|
assert backend.query(models.User) == [user]
|
|
assert backend.query(models.User, id=user.id) == [user]
|
|
assert not backend.query(models.User, id="invalid")
|
|
assert backend.get(models.User, id=user.id) == user
|
|
|
|
user.family_name = "new_family_name"
|
|
|
|
assert user.family_name == "new_family_name"
|
|
|
|
backend.reload(user)
|
|
|
|
assert user.family_name == "family_name"
|
|
|
|
backend.delete(user)
|
|
|
|
assert not backend.query(models.User, id=user.id)
|
|
assert not backend.get(models.User, id=user.id)
|
|
|
|
backend.delete(user)
|
|
|
|
|
|
def test_model_attribute_edition(testclient, backend):
|
|
user = models.User(
|
|
user_name="user_name",
|
|
family_name="family_name",
|
|
formatted_name="formatted_name",
|
|
display_name="display_name",
|
|
emails=["email1@user.test", "email2@user.test"],
|
|
)
|
|
backend.save(user)
|
|
|
|
assert user.user_name == "user_name"
|
|
assert user.family_name == "family_name"
|
|
assert user.emails == ["email1@user.test", "email2@user.test"]
|
|
|
|
user = backend.get(models.User, id=user.id)
|
|
assert user.user_name == "user_name"
|
|
assert user.family_name == "family_name"
|
|
assert user.emails == ["email1@user.test", "email2@user.test"]
|
|
|
|
user.family_name = "new_family_name"
|
|
user.emails = ["email1@user.test"]
|
|
backend.save(user)
|
|
|
|
assert user.family_name == "new_family_name"
|
|
assert user.emails == ["email1@user.test"]
|
|
|
|
user = backend.get(models.User, id=user.id)
|
|
assert user.family_name == "new_family_name"
|
|
assert user.emails == ["email1@user.test"]
|
|
|
|
user.display_name = ""
|
|
assert not user.display_name
|
|
|
|
backend.save(user)
|
|
assert not user.display_name
|
|
|
|
backend.delete(user)
|
|
|
|
|
|
def test_model_indexation(testclient, backend):
|
|
user = models.User(
|
|
user_name="user_name",
|
|
family_name="family_name",
|
|
formatted_name="formatted_name",
|
|
emails=["email1@user.test", "email2@user.test"],
|
|
)
|
|
backend.save(user)
|
|
|
|
assert backend.get(models.User, family_name="family_name") == user
|
|
assert not backend.get(models.User, family_name="new_family_name")
|
|
assert backend.get(models.User, emails=["email1@user.test"]) == user
|
|
assert backend.get(models.User, emails=["email2@user.test"]) == user
|
|
assert not backend.get(models.User, emails=["email3@user.test"])
|
|
|
|
user.family_name = "new_family_name"
|
|
user.emails = ["email2@user.test"]
|
|
|
|
assert backend.get(models.User, family_name="family_name") != user
|
|
assert backend.get(models.User, emails=["email1@user.test"]) != user
|
|
assert not backend.get(models.User, emails=["email3@user.test"])
|
|
|
|
backend.save(user)
|
|
|
|
assert not backend.get(models.User, family_name="family_name")
|
|
assert backend.get(models.User, family_name="new_family_name") == user
|
|
assert not backend.get(models.User, emails=["email1@user.test"])
|
|
assert backend.get(models.User, emails=["email2@user.test"]) == user
|
|
assert not backend.get(models.User, emails=["email3@user.test"])
|
|
|
|
backend.delete(user)
|
|
|
|
assert not backend.get(models.User, family_name="family_name")
|
|
assert not backend.get(models.User, family_name="new_family_name")
|
|
assert not backend.get(models.User, emails=["email1@user.test"])
|
|
assert not backend.get(models.User, emails=["email2@user.test"])
|
|
assert not backend.get(models.User, emails=["email3@user.test"])
|
|
|
|
|
|
def test_fuzzy_unique_attribute(user, moderator, admin, backend):
|
|
assert set(backend.query(models.User)) == {user, moderator, admin}
|
|
assert set(backend.fuzzy(models.User, "Jack")) == {moderator}
|
|
assert set(backend.fuzzy(models.User, "Jack", ["formatted_name"])) == {moderator}
|
|
assert set(backend.fuzzy(models.User, "Jack", ["user_name"])) == set()
|
|
assert set(backend.fuzzy(models.User, "Jack", ["user_name", "formatted_name"])) == {
|
|
moderator
|
|
}
|
|
assert set(backend.fuzzy(models.User, "moderator")) == {moderator}
|
|
assert set(backend.fuzzy(models.User, "oderat")) == {moderator}
|
|
assert set(backend.fuzzy(models.User, "oDeRat")) == {moderator}
|
|
assert set(backend.fuzzy(models.User, "ack")) == {moderator}
|
|
|
|
|
|
def test_fuzzy_multiple_attribute(user, moderator, admin, backend):
|
|
assert set(backend.query(models.User)) == {user, moderator, admin}
|
|
assert set(backend.fuzzy(models.User, "jack@doe.test")) == {moderator}
|
|
assert set(backend.fuzzy(models.User, "jack@doe.test", ["emails"])) == {moderator}
|
|
assert set(backend.fuzzy(models.User, "jack@doe.test", ["formatted_name"])) == set()
|
|
assert set(
|
|
backend.fuzzy(models.User, "jack@doe.test", ["emails", "formatted_name"])
|
|
) == {moderator}
|
|
assert set(backend.fuzzy(models.User, "ack@doe.te")) == {moderator}
|
|
assert set(backend.fuzzy(models.User, "doe.test")) == {user, moderator, admin}
|
|
|
|
|
|
def test_model_references(testclient, user, foo_group, admin, bar_group, backend):
|
|
assert foo_group.members == [user]
|
|
assert user.groups == [foo_group]
|
|
assert foo_group in backend.query(models.Group, members=user)
|
|
assert user in backend.query(models.User, groups=foo_group)
|
|
|
|
assert user not in bar_group.members
|
|
assert bar_group not in user.groups
|
|
user.groups = user.groups + [bar_group]
|
|
backend.save(user)
|
|
backend.reload(bar_group)
|
|
|
|
assert user in bar_group.members
|
|
assert bar_group in user.groups
|
|
|
|
bar_group.members = [admin]
|
|
backend.save(bar_group)
|
|
backend.reload(user)
|
|
|
|
assert user not in bar_group.members
|
|
assert bar_group not in user.groups
|
|
|
|
|
|
def test_model_creation_edition_datetime(testclient, backend):
|
|
if "ldap" in backend.__class__.__module__:
|
|
pytest.skip()
|
|
|
|
with time_machine.travel("2020-01-01 02:00:00+00:00", tick=False):
|
|
user = models.User(
|
|
user_name="foo",
|
|
family_name="foo",
|
|
formatted_name="foo",
|
|
)
|
|
backend.save(user)
|
|
assert user.created == datetime.datetime(
|
|
2020, 1, 1, 2, tzinfo=datetime.timezone.utc
|
|
)
|
|
assert user.last_modified == datetime.datetime(
|
|
2020, 1, 1, 2, tzinfo=datetime.timezone.utc
|
|
)
|
|
|
|
with time_machine.travel("2021-01-01 02:00:00+00:00", tick=False):
|
|
user.family_name = "bar"
|
|
backend.save(user)
|
|
assert user.created == datetime.datetime(
|
|
2020, 1, 1, 2, tzinfo=datetime.timezone.utc
|
|
)
|
|
assert user.last_modified == datetime.datetime(
|
|
2021, 1, 1, 2, tzinfo=datetime.timezone.utc
|
|
)
|
|
|
|
backend.delete(user)
|