def main():
vote_repository = InMemoryVoteRepository()
Vote().save(vote_repository)
Vote().save(vote_repository)
print(vote_repository.all())
print(f'Total votes: {vote_repository.total()}')
def test_vote_existing_vote_id():
vote_id = str(uuid.uuid4())
assert Vote(vote_id).vote_id == vote_id
def test_vote_defaults():
assert uuid.UUID(Vote().vote_id)
def test_vote_save():
vote = Vote()
vote_repository = InMemoryVoteRepository()
assert vote.save(vote_repository).vote_id == vote.vote_id
def test_vote_repository_total():
vote_repository = InMemoryVoteRepository()
Vote().save(vote_repository)
Vote().save(vote_repository)
assert vote_repository.total() == 2
def test_vote_repository_all():
vote_repository = InMemoryVoteRepository()
vote1 = Vote().save(vote_repository)
vote2 = Vote().save(vote_repository)
assert set(vote_repository.all()) == {vote1, vote2}
def vote() -> Vote:
return Vote().save(vote_repository)