diff --git a/scopes/storage/tracking.py b/scopes/storage/tracking.py index 3b50d70..54c16f8 100644 --- a/scopes/storage/tracking.py +++ b/scopes/storage/tracking.py @@ -170,8 +170,8 @@ class Container(object): return track.trackId return self.insert(track, withTrackId=True) - def remove(self, trackId): - stmt = self.table.delete().where(self.table.c.trackid == trackId) + def remove(self, track): + stmt = self.table.delete().where(self.table.c.trackid == track.trackId) n = self.session.execute(stmt).rowcount if n > 0: self.storage.mark_changed() diff --git a/scopes/tlib_storage.py b/scopes/tlib_storage.py index e6b1b7d..f19aedf 100644 --- a/scopes/tlib_storage.py +++ b/scopes/tlib_storage.py @@ -50,7 +50,7 @@ def test_tracking(self, config): tr03 = storage.getItem('rec-31') self.assertEqual(tr03.trackId, 31) - n = tracks.remove(31) + n = tracks.remove(tr03) self.assertEqual(n, 1) self.assertEqual(tracks.get(31), None)