Fix #564: Added a script to prune pre 0.17 federated tracks

This commit is contained in:
Eliot Berriot 2019-01-14 17:11:24 +01:00
parent b453769a32
commit 6aaa51e8e7
No known key found for this signature in database
GPG key ID: DD6965E2476E5C27
5 changed files with 56 additions and 1 deletions

View file

@ -12,7 +12,12 @@ def command():
@pytest.mark.parametrize(
"script_name", ["django_permissions_to_user_permissions", "test"]
"script_name",
[
"django_permissions_to_user_permissions",
"test",
"delete_pre_017_federated_uploads",
],
)
def test_script_command_list(command, script_name, mocker):
mocked = mocker.patch("funkwhale_api.common.scripts.{}.main".format(script_name))
@ -235,3 +240,17 @@ def test_migrate_to_users_libraries_command(
for part in ["followers", "following"]:
generate_actor_urls.assert_any_call(part, command.stdout)
def test_delete_pre_017_federated_uploads(factories, command):
to_delete = factories["music.Upload"](
source="https://test.com/federation/music/file/1"
)
to_keep = factories["music.Upload"](source="https://hello.world")
scripts.delete_pre_017_federated_uploads.main(command)
to_keep.refresh_from_db()
with pytest.raises(to_delete.__class__.DoesNotExist):
to_delete.refresh_from_db()