yetangitu/funkwhale
1
0
Fork 0
mirror of https://code.eliotberriot.com/funkwhale/funkwhale.git synced 2025-10-04 18:59:17 +02:00
Code Issues Releases Wiki Activity

Implement tag models

Browse source
This commit is contained in:
Eliot Berriot 2019-07-08 15:26:14 +02:00
parent c170ee9394
commit 6dde4b73cd
28 changed files with 1034 additions and 141 deletions
Download patch file Download diff file Expand all files Collapse all files

104
api/tests/common/test_commands.py Normal file
View file

@ -0,0 +1,104 @@
import pytest
from django.core.management import call_command
from funkwhale_api.federation import models as federation_models
from funkwhale_api.music import models as music_models
from funkwhale_api.tags import models as tags_models
from funkwhale_api.users import models as users_models
def test_load_test_data_dry_run(factories, mocker):
call_command("load_test_data", artists=10)
assert music_models.Artist.objects.count() == 0
@pytest.mark.parametrize(
"kwargs, expected_counts",
[
(
{"create_dependencies": True, "artists": 10},
[(music_models.Artist.objects.all(), 10)],
),
(
{"create_dependencies": True, "albums": 10, "artists": 1},
[
(music_models.Album.objects.all(), 10),
(music_models.Artist.objects.all(), 1),
],
),
(
{"create_dependencies": True, "tracks": 20, "albums": 10, "artists": 1},
[
(music_models.Track.objects.all(), 20),
(music_models.Album.objects.all(), 10),
(music_models.Artist.objects.all(), 1),
],
),
(
{"create_dependencies": True, "albums": 10, "albums_artist_factor": 0.5},
[
(music_models.Album.objects.all(), 10),
(music_models.Artist.objects.all(), 5),
],
),
(
{"create_dependencies": True, "albums": 3},
[
(music_models.Album.objects.all(), 3),
(music_models.Artist.objects.all(), 1),
],
),
(
{"create_dependencies": True, "local_accounts": 3},
[
(users_models.User.objects.all(), 3),
(federation_models.Actor.objects.all(), 3),
],
),
(
{"create_dependencies": True, "local_libraries": 3},
[
(users_models.User.objects.all(), 3),
(federation_models.Actor.objects.all(), 3),
(music_models.Library.objects.all(), 3),
],
),
(
{"create_dependencies": True, "local_uploads": 3},
[
(users_models.User.objects.all(), 1),
(federation_models.Actor.objects.all(), 1),
(music_models.Library.objects.all(), 1),
(music_models.Upload.objects.filter(import_status="finished"), 3),
(music_models.Track.objects.all(), 3),
],
),
(
{"create_dependencies": True, "tags": 3},
[(tags_models.Tag.objects.all(), 3)],
),
(
{"create_dependencies": True, "track_tags": 3},
[
(tags_models.Tag.objects.all(), 1),
(tags_models.TaggedItem.objects.all(), 3),
(music_models.Track.objects.all(), 3),
],
),
],
)
def test_load_test_data_args(factories, kwargs, expected_counts, mocker):
call_command("load_test_data", dry_run=False, **kwargs)
for qs, expected_count in expected_counts:
assert qs.count() == expected_count
def test_load_test_data_skip_dependencies(factories):
factories["music.Artist"].create_batch(size=5)
call_command("load_test_data", dry_run=False, albums=10, create_dependencies=False)
assert music_models.Artist.objects.count() == 5
assert music_models.Album.objects.count() == 10