def test_save_wrong_array(self): vote_key = 'vote_key' + str(random.randint(0, 50000)) self.assertFalse( vote_dao.save_vote(self.__votation__.votation_id, vote_key, [1, 2])) self.assertFalse( vote_dao.save_vote(self.__votation__.votation_id, vote_key, [1, 2, 3, 4]))
def test_save_ok2(self): vote_key = 'vote_key' + str(random.randint(0, 50000)) self.assertTrue( vote_dao.save_vote(self.__votation__.votation_id, vote_key, [1, 2, 2])) ar = vote_dao.load_vote_by_key(vote_key) self.assertEqual(3, len(ar)) check = True for v in ar: if (v.option_id == self.__option1.option_id and \ v.jud_value == 1) or \ (v.option_id == self.__option2.option_id and \ v.jud_value == 2) or \ (v.option_id == self.__option3.option_id and \ v.jud_value == 2): check = True else: check = False self.assertTrue(check)
def save_votes(user_id, vote_key, votation_id, vote_array): """ Save a vote, one rank number per option id. The vote in vote_array[0] is for the first option. The vote in vote_array[1] is for the second option, and so on. Every option need a vote. Also, the voter is set as voted. """ u = user.load_user_by_id(user_id) if not u: return False if voter_dao.has_voted(user_id, votation_id): ar = vote_dao.load_vote_by_key(vote_key) if len(ar) == 0: return False vote_dao.delete_votes_by_key(vote_key) if vote_dao.save_vote(votation_id=votation_id, vote_key=vote_key, array_judgements=vote_array): if voter_dao.set_voted(user_id, votation_id): db.session.commit() return True db.session.rollback() return False
def test_save_wrong_votation_id(self): vote_key = 'vote_key' + str(random.randint(0, 50000)) self.assertFalse(vote_dao.save_vote(333333, vote_key, [3, 2, 1]))
def test_save_null(self): vote_key = 'vote_key' + str(random.randint(0, 50000)) self.assertFalse( vote_dao.save_vote(self.__votation__.votation_id, vote_key, [1, 2, None]))