def saveSong(songID): try: if (songID == "0") or (User.checkForDuplicateSavedSong(session['email'], songID, conn)): return json.dumps({'savedSong': False}) else: User.saveSong(session['email'], songID, conn) return json.dumps({'savedSong': True}) except: return json.dumps({'savedSong': True})
def test_checkForDuplicateSavedSong(): assert User.checkForDuplicateSavedSong(email1, songID1, conn) == {'email': '*****@*****.**', 'songID': '887fgf7ergf8e'} assert User.checkForDuplicateSavedSong(email1, songID4, conn) == None assert User.checkForDuplicateSavedSong(email2, songID4, conn) == None assert User.checkForDuplicateSavedSong(email3, songID4, conn) == None