test: updated test case for returned playlist

klepp0 · Jul 4, 2024 · 8ab1ca9 · 8ab1ca9
1 parent b4d1ddd
commit 8ab1ca9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/tests/integrationtests/spotify/test_session.py b/tests/integrationtests/spotify/test_session.py
@@ -90,10 +90,8 @@ def test_find_track(spotify_session, tidal_track):
 def test_add_to_playlist(spotify_session, create_playlist, spotify_track_list):
     playlist = create_playlist
     n_tracks_before = playlist.n_tracks
-    spotify_session.add_to_playlist(playlist, spotify_track_list)
-
-    spotify_response = spotify_session._client.playlist(playlist.playlist_id)
-    updated_playlist = Playlist.from_spotify(spotify_response)
+    updated_playlist = spotify_session.add_to_playlist(playlist, spotify_track_list)
     n_tracks_after = updated_playlist.n_tracks
 
+    assert playlist.playlist_id == updated_playlist.playlist_id
     assert n_tracks_before < n_tracks_after
diff --git a/tests/integrationtests/tidal/test_session.py b/tests/integrationtests/tidal/test_session.py
@@ -87,10 +87,8 @@ def test_search_track(tidal_session, spotify_track):
 def test_add_to_playlist(tidal_session, create_playlist, tidal_track_list):
     playlist = create_playlist
     n_tracks_before = playlist.n_tracks
-    tidal_session.add_to_playlist(playlist, tidal_track_list)
-
-    tidal_response = tidal_session._client.playlist(playlist.playlist_id)
-    updated_playlist = Playlist.from_tidal(tidal_response)
+    updated_playlist = tidal_session.add_to_playlist(playlist, tidal_track_list)
     n_tracks_after = updated_playlist.n_tracks
 
+    assert playlist.playlist_id == updated_playlist.playlist_id
     assert n_tracks_before < n_tracks_after