Update remove test and method to use track instead of position
This commit is contained in:
parent
2c064059d4
commit
151321d200
@ -81,9 +81,8 @@ class BaseCurrentPlaylistController(object):
|
|||||||
|
|
||||||
self.playlist = Playlist(tracks=new_tracks)
|
self.playlist = Playlist(tracks=new_tracks)
|
||||||
|
|
||||||
def remove(self, position):
|
def remove(self,track):
|
||||||
tracks = self.playlist.tracks
|
tracks = filter(lambda t: t != track, self.playlist.tracks)
|
||||||
del tracks[position]
|
|
||||||
|
|
||||||
self.playlist = Playlist(tracks=tracks)
|
self.playlist = Playlist(tracks=tracks)
|
||||||
|
|
||||||
|
|||||||
@ -147,7 +147,7 @@ class BaseCurrentPlaylistControllerTest(object):
|
|||||||
def test_remove(self):
|
def test_remove(self):
|
||||||
track1 = self.controller.playlist.tracks[1]
|
track1 = self.controller.playlist.tracks[1]
|
||||||
track2 = self.controller.playlist.tracks[2]
|
track2 = self.controller.playlist.tracks[2]
|
||||||
self.controller.remove(1)
|
self.controller.remove(track1)
|
||||||
self.assert_(track1 not in self.controller.playlist.tracks)
|
self.assert_(track1 not in self.controller.playlist.tracks)
|
||||||
self.assertEqual(track2, self.controller.playlist.tracks[1])
|
self.assertEqual(track2, self.controller.playlist.tracks[1])
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user