optimising utils.py with prefect_related

This commit is contained in:
Petitminion 2021-07-15 10:48:59 +02:00
parent dc19381cd2
commit 7af11d3fdb
2 changed files with 5 additions and 7 deletions

View file

@ -21,10 +21,10 @@ def test_get_track_id_from_xspf(factories, tmp_path):
def test_generate_xspf_from_playlist(factories):
playlist = factories["playlists.PlaylistTrack"]()
playlist_factory = models.Playlist.objects.get()
xspf_test = utils.generate_xspf_from_playlist(playlist.id)
tree = etree.fromstring(xspf_test)
playlist_factory = models.Playlist.objects.get()
track1 = playlist_factory.playlist_tracks.get(id=1)
track1_name = track1.track
assert playlist_factory.name == tree.findtext("./title")
assert track1_name.title == tree.findtext("./trackList/track/title")
assert track1_name.title == tree.findtext("./trackList/track/title")