def test_has_user_voted_for_song(self): vote_service = VoteService() vote = Vote() vote.user_id = 1 vote.song_id = 1 vote.created_date = datetime.now() vote.like = True vote.save() user = User() user.id = 1 song = Song song.id = 1 self.assertTrue(vote_service.has_user_voted_for_song(song, user))
def clean(self): vote_service = VoteService() if vote_service.has_user_voted_for_song(self.song, self.user): raise ValidationError("Vote already exists") else: return self