def test_manage_upload_action_relaunch_import(factories, mocker):
m = mocker.patch("funkwhale_api.common.utils.on_commit")
# this one is finished and should stay as is
finished = factories["music.Upload"](import_status="finished")
draft = factories["music.Upload"](import_status="draft")
to_relaunch = [
factories["music.Upload"](import_status="pending"),
factories["music.Upload"](import_status="skipped"),
factories["music.Upload"](import_status="errored"),
]
s = serializers.UploadActionSerializer(queryset=None)
s.handle_relaunch_import(models.Upload.objects.all())
for obj in to_relaunch:
obj.refresh_from_db()
assert obj.import_status == "pending"
m.assert_any_call(tasks.process_upload.delay, upload_id=obj.pk)
finished.refresh_from_db()
assert finished.import_status == "finished"
draft.refresh_from_db()
assert draft.import_status == "draft"
assert m.call_count == 3
def test_manage_upload_action_delete(factories, queryset_equal_list, mocker):
dispatch = mocker.patch("funkwhale_api.federation.routes.outbox.dispatch")
library1 = factories["music.Library"]()
library2 = factories["music.Library"]()
library1_uploads = factories["music.Upload"].create_batch(size=3,
library=library1)
library2_uploads = factories["music.Upload"].create_batch(size=3,
library=library2)
s = serializers.UploadActionSerializer(queryset=None)
s.handle_delete(library1_uploads[0].__class__.objects.all())
assert library1_uploads[0].__class__.objects.count() == 0
dispatch.assert_any_call(
{
"type": "Delete",
"object": {
"type": "Audio"
}
},
context={"uploads": library1_uploads},
)
dispatch.assert_any_call(
{
"type": "Delete",
"object": {
"type": "Audio"
}
},
context={"uploads": library2_uploads},
)
def test_manage_upload_action_publish(factories, mocker):
m = mocker.patch("funkwhale_api.common.utils.on_commit")
draft = factories["music.Upload"](import_status="draft")
s = serializers.UploadActionSerializer(queryset=None)
s.handle_publish(models.Upload.objects.all())
draft.refresh_from_db()
assert draft.import_status == "pending"
m.assert_any_call(tasks.process_upload.delay, upload_id=draft.pk)