def set_album_cover(track_info): if track_info['album'].image: # we already have a cover return track_info['album'].image if not os.path.exists(COVERSDIR): os.mkdir(COVERSDIR) filename = hashlib.sha1(track_info['album'].name).hexdigest() + '.png' path = os.path.join(COVERSDIR, filename) result = fetch_cover(track_info['artist'].name, track_info['album'].name, path) if not result: return None track_info['album'].image = path return path
def test_bad_album(): assert fetch_cover('Arcade Fire', 'they have no album like this', '/tmp/test-cover-3.png') == None
def test_success(): assert fetch_cover('Arcade Fire', 'Neon Bible', '/tmp/test-cover-1.png') == '/tmp/test-cover-1.png'
def test_bad_artist(): assert fetch_cover('abcdefghijklmnopqrstuvwxyz', 'Neon Bible', '/tmp/test-cover-2.png') == None