funkwhale/api/tests/playlists/test_filters.py

29 wiersze
1018 B
Python

from funkwhale_api.playlists import filters, models
def test_playlist_filter_track(factories, queryset_equal_list):
plt = factories["playlists.PlaylistTrack"]()
factories["playlists.PlaylistTrack"]()
qs = models.Playlist.objects.all()
filterset = filters.PlaylistFilter({"track": plt.track.pk}, queryset=qs)
assert filterset.qs == [plt.playlist]
def test_playlist_filter_album(factories, queryset_equal_list):
plt = factories["playlists.PlaylistTrack"]()
factories["playlists.PlaylistTrack"]()
qs = models.Playlist.objects.all()
filterset = filters.PlaylistFilter({"album": plt.track.album.pk}, queryset=qs)
assert filterset.qs == [plt.playlist]
def test_playlist_filter_artist(factories, queryset_equal_list):
plt = factories["playlists.PlaylistTrack"]()
factories["playlists.PlaylistTrack"]()
qs = models.Playlist.objects.all()
filterset = filters.PlaylistFilter({"artist": plt.track.artist.pk}, queryset=qs)
assert filterset.qs == [plt.playlist]