funkwhale/api/tests/common/test_models.py

105 wiersze
3.3 KiB
Python

import pytest
from django.urls import reverse
from funkwhale_api.federation import utils as federation_utils
@pytest.mark.parametrize(
"model,factory_args,namespace",
[("common.Mutation", {"created_by__local": True}, "federation:edits-detail")],
)
def test_mutation_fid_is_populated(factories, model, factory_args, namespace):
instance = factories[model](**factory_args, fid=None, payload={})
assert instance.fid == federation_utils.full_url(
reverse(namespace, kwargs={"uuid": instance.uuid})
)
@pytest.mark.parametrize(
"factory_name, expected",
[
("music.Artist", "/library/artists/{obj.pk}"),
("music.Album", "/library/albums/{obj.pk}"),
("music.Track", "/library/tracks/{obj.pk}"),
("playlists.Playlist", "/library/playlists/{obj.pk}"),
],
)
def test_get_absolute_url(factory_name, factories, expected):
obj = factories[factory_name]()
assert obj.get_absolute_url() == expected.format(obj=obj)
@pytest.mark.parametrize(
"factory_name, expected",
[
("music.Artist", "/manage/library/artists/{obj.pk}"),
("music.Album", "/manage/library/albums/{obj.pk}"),
("music.Track", "/manage/library/tracks/{obj.pk}"),
("music.Library", "/manage/library/libraries/{obj.uuid}"),
("federation.Actor", "/manage/moderation/accounts/{obj.full_username}"),
],
)
def test_get_moderation_url(factory_name, factories, expected):
obj = factories[factory_name]()
assert obj.get_moderation_url() == expected.format(obj=obj)
def test_attachment(factories, now):
attachment = factories["common.Attachment"]()
assert attachment.uuid is not None
assert attachment.mimetype == "image/jpeg"
assert attachment.file is not None
assert attachment.url is not None
assert attachment.actor is not None
assert attachment.creation_date > now
assert attachment.last_fetch_date is None
assert attachment.size > 0
@pytest.mark.parametrize("args, expected", [([], [0]), ([True], [0]), ([False], [1])])
def test_attachment_queryset_attached(args, expected, factories, queryset_equal_list):
attachments = [
factories["music.Album"](
with_cover=True, artist__attachment_cover=None
).attachment_cover,
factories["common.Attachment"](),
]
queryset = attachments[0].__class__.objects.attached(*args).order_by("id")
expected_objs = [attachments[i] for i in expected]
assert queryset == expected_objs
def test_removing_obj_removes_content(factories):
kept_content = factories["common.Content"]()
removed_content = factories["common.Content"]()
track1 = factories["music.Track"](description=removed_content)
factories["music.Track"](description=kept_content)
track1.delete()
with pytest.raises(removed_content.DoesNotExist):
removed_content.refresh_from_db()
kept_content.refresh_from_db()
def test_content_as_plain_text(factories):
content = factories["common.Content"](
content_type="text/html", text="<b>hello world</b>"
)
assert content.as_plain_text == "hello world"
def test_content_truncate(factories):
content = factories["common.Content"](
content_type="text/html", text="<b>hello world</b>"
)
assert content.truncate(5) == "hello…"