fix tracks.remove(): use track as parameter, not trackId
This commit is contained in:
parent
5dac623011
commit
5044978acc
2 changed files with 3 additions and 3 deletions
|
@ -171,8 +171,8 @@ class Container(object):
|
||||||
return track.trackId
|
return track.trackId
|
||||||
return self.insert(track, withTrackId=True)
|
return self.insert(track, withTrackId=True)
|
||||||
|
|
||||||
def remove(self, trackId):
|
def remove(self, track):
|
||||||
stmt = self.table.delete().where(self.table.c.trackid == trackId)
|
stmt = self.table.delete().where(self.table.c.trackid == track.trackId)
|
||||||
n = self.session.execute(stmt).rowcount
|
n = self.session.execute(stmt).rowcount
|
||||||
if n > 0:
|
if n > 0:
|
||||||
self.storage.mark_changed()
|
self.storage.mark_changed()
|
||||||
|
|
|
@ -50,7 +50,7 @@ def test_tracking(self, config):
|
||||||
tr03 = storage.getItem('rec-31')
|
tr03 = storage.getItem('rec-31')
|
||||||
self.assertEqual(tr03.trackId, 31)
|
self.assertEqual(tr03.trackId, 31)
|
||||||
|
|
||||||
n = tracks.remove(31)
|
n = tracks.remove(tr03)
|
||||||
self.assertEqual(n, 1)
|
self.assertEqual(n, 1)
|
||||||
self.assertEqual(tracks.get(31), None)
|
self.assertEqual(tracks.get(31), None)
|
||||||
|
|
||||||
|
|
Loading…
Add table
Reference in a new issue