def test_get_track_from_album_by_title_approx(self):
data = {'id': 123,
'tracklist': [
self._make_track('SONG A', '1', '01:01'),
self._make_track('SONG B', '2', '02:02'),
self._make_track('COMPLETELY DIFFERENT', '3', '03:03'),
],
'artists': [{'name': 'ARTIST NAME', 'id': 321, 'join': ''}],
'title': 'TITLE'}
release = Bag(data=data,
title=data['title'],
artists=[Bag(data=d) for d in data['artists']])
d = DiscogsPlugin()
album_info = d.get_album_info(release)
track_list = d.get_tracks_from_album_by_title(album_info, 'SONG')
self.assertEqual(len(track_list), 2)
song_a = None
song_b = None
for track_info in track_list:
if track_info.title == 'SONG A':
song_a = track_info
elif track_info.title == 'SONG B':
song_b = track_info
self.assertIsNotNone(song_a)
self.assertIsNotNone(song_b)
self.assertEqual(song_a.artist, 'ARTIST NAME')
self.assertEqual(song_a.artist_id, 321)
self.assertEqual(song_b.artist, 'ARTIST NAME')
self.assertEqual(song_b.artist_id, 321)
def test_get_track_from_album_by_title_missing(self):
data = {'id': 123,
'tracklist': [self._make_track('SONG', '1', '01:01')],
'artists': [{'name': 'ARTIST NAME', 'id': 321, 'join': ''}],
'title': 'TITLE'}
release = Bag(data=data,
title=data['title'],
artists=[Bag(data=d) for d in data['artists']])
d = DiscogsPlugin()
album_info = d.get_album_info(release)
track_list = d.get_tracks_from_album_by_title(album_info, 'ZZZZZZZZ', 0.0)
self.assertEqual(len(track_list), 0)