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_saveSong(): User.saveSong(email1, songID1, conn) User.saveSong(email2, songID2, conn) User.saveSong(email1, songID3, conn) assert User.getSavedSongs(email1, conn) == [{'email': '*****@*****.**', 'songID': '887fgf7ergf8e'}, {'email': '*****@*****.**', 'songID': 'djhgds7678wev'}] assert User.getSavedSongs(email2, conn) == [{'email': '*****@*****.**', 'songID': 'chds763ndsuds'}] assert User.getSavedSongs(email3, conn) == ()