Update remove test and method to use track instead of position

This commit is contained in:
Thomas Adamcik 2010-02-13 23:12:53 +01:00
parent 2c064059d4
commit 151321d200
2 changed files with 3 additions and 4 deletions

View File

@ -81,9 +81,8 @@ class BaseCurrentPlaylistController(object):
self.playlist = Playlist(tracks=new_tracks)
def remove(self, position):
tracks = self.playlist.tracks
del tracks[position]
def remove(self,track):
tracks = filter(lambda t: t != track, self.playlist.tracks)
self.playlist = Playlist(tracks=tracks)

View File

@ -147,7 +147,7 @@ class BaseCurrentPlaylistControllerTest(object):
def test_remove(self):
track1 = self.controller.playlist.tracks[1]
track2 = self.controller.playlist.tracks[2]
self.controller.remove(1)
self.controller.remove(track1)
self.assert_(track1 not in self.controller.playlist.tracks)
self.assertEqual(track2, self.controller.playlist.tracks[1])