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)