def test_update_vote():
value = 3
for voter_id, _, _ in EXISTING_VOTES:
Vote.update(voter_id, voted_id=value)
votes = Vote.find_all()
for vote in votes:
assert vote.voted_id == value
def count_vote(room_id):
votes = Vote.find_all(room_id=room_id)
tally = Counter()
for vote in votes:
tally[vote.voted_id] += 1
return tally, len(votes)
def test_find_all_votes():
votes = Vote.find_all(room_id=1)
assert len(votes) == 2
for vote in votes:
print(vote.voter_id, vote.room_id, vote.voted_id)