94 lines
3.1 KiB
Python
94 lines
3.1 KiB
Python
import datetime
|
|
|
|
import pytest
|
|
|
|
from funkwhale_api.common import models, serializers, signals, tasks
|
|
|
|
|
|
def test_apply_migration(factories, mocker):
|
|
mutation = factories["common.Mutation"](payload={})
|
|
apply = mocker.patch.object(mutation.__class__, "apply")
|
|
tasks.apply_mutation(mutation_id=mutation.pk)
|
|
|
|
apply.assert_called_once_with()
|
|
|
|
|
|
def test_broadcast_mutation_created(factories, mocker):
|
|
mutation = factories["common.Mutation"](payload={})
|
|
factories["common.Mutation"](payload={}, is_approved=True)
|
|
group_send = mocker.patch("funkwhale_api.common.channels.group_send")
|
|
expected = serializers.APIMutationSerializer(mutation).data
|
|
|
|
signals.mutation_created.send(sender=None, mutation=mutation)
|
|
group_send.assert_called_with(
|
|
"instance_activity",
|
|
{
|
|
"type": "event.send",
|
|
"text": "",
|
|
"data": {
|
|
"type": "mutation.created",
|
|
"mutation": expected,
|
|
"pending_review_count": 1,
|
|
},
|
|
},
|
|
)
|
|
|
|
|
|
def test_broadcast_mutation_updated(factories, mocker):
|
|
mutation = factories["common.Mutation"](payload={}, is_approved=True)
|
|
factories["common.Mutation"](payload={})
|
|
group_send = mocker.patch("funkwhale_api.common.channels.group_send")
|
|
expected = serializers.APIMutationSerializer(mutation).data
|
|
|
|
signals.mutation_updated.send(
|
|
sender=None, mutation=mutation, old_is_approved=False, new_is_approved=True
|
|
)
|
|
group_send.assert_called_with(
|
|
"instance_activity",
|
|
{
|
|
"type": "event.send",
|
|
"text": "",
|
|
"data": {
|
|
"type": "mutation.updated",
|
|
"mutation": expected,
|
|
"old_is_approved": False,
|
|
"new_is_approved": True,
|
|
"pending_review_count": 1,
|
|
},
|
|
},
|
|
)
|
|
|
|
|
|
def test_cannot_apply_already_applied_migration(factories):
|
|
mutation = factories["common.Mutation"](payload={}, is_applied=True)
|
|
with pytest.raises(mutation.__class__.DoesNotExist):
|
|
tasks.apply_mutation(mutation_id=mutation.pk)
|
|
|
|
|
|
def test_prune_unattached_attachments(factories, settings, now):
|
|
settings.ATTACHMENTS_UNATTACHED_PRUNE_DELAY = 5
|
|
prunable_date = now - datetime.timedelta(
|
|
seconds=settings.ATTACHMENTS_UNATTACHED_PRUNE_DELAY
|
|
)
|
|
attachments = [
|
|
# attached, kept
|
|
factories["music.Album"](with_cover=True).attachment_cover,
|
|
# recent, kept
|
|
factories["common.Attachment"](),
|
|
# too old, pruned
|
|
factories["common.Attachment"](creation_date=prunable_date),
|
|
# attached to a mutation, kept even if old
|
|
models.MutationAttachment.objects.create(
|
|
mutation=factories["common.Mutation"](payload={}),
|
|
attachment=factories["common.Attachment"](creation_date=prunable_date),
|
|
).attachment,
|
|
]
|
|
|
|
tasks.prune_unattached_attachments()
|
|
|
|
attachments[0].refresh_from_db()
|
|
attachments[1].refresh_from_db()
|
|
attachments[3].refresh_from_db()
|
|
with pytest.raises(attachments[2].DoesNotExist):
|
|
attachments[2].refresh_from_db()
|