class ChoiceRepositoryTestCase(TestCase, Utility): sut = None; poll = None; def setUp(self): self.sut = ChoiceRepository() self.poll = Poll(uniqueId = self.getUniqueId()) self.poll.save() def test_when_voting_for_a_choice_then_the_votes_count_is_incremented(self): choice = self._createTestChoice(self.getUniqueId(), 1, self.poll) self.sut.voteForChoices([choice.uniqueId]) self.assertEqual(Choice.objects.get(pk=choice.id).votes, 2) def test_when_voting_for_multiple_choices_then_votes_for_every_choice_is_incremented(self): choices = [ self._createTestChoice(self.getUniqueId(), 3, self.poll), self._createTestChoice(self.getUniqueId(), 4, self.poll) ] self.sut.voteForChoices([choices[0].uniqueId, choices[1].uniqueId]) self.assertEqual(Choice.objects.get(pk=choices[0].id).votes, 4) self.assertEqual(Choice.objects.get(pk=choices[1].id).votes, 5) def test_when_creating_a_choice_then_choice_is_persisted(self): expected = Choice(text='test', isSelected=True, uniqueId='1234', poll=self.poll) self.sut.create(expected) actual = Choice.objects.get(pk=expected.id) self.assertEqual(expected.text, actual.text) self.assertEqual(expected.isSelected, actual.isSelected) self.assertEqual(expected.uniqueId, actual.uniqueId) def _createTestChoice(self, uniqueId, votes, poll): choice = Choice(uniqueId = uniqueId, votes = votes, poll = poll) choice.save() return choice def test_when_retrieving_choice_by_unique_id_then_the_correct_choice_is_returned(self): uniqueId = self.getUniqueId() choice = Choice() poll = Poll() poll.save() choice.poll = poll choice.uniqueId = uniqueId choice.save(); result = self.sut.getChoiceByUniqueId(uniqueId) self.assertEqual(choice.id, result.id) def test_when_retrieving_choice_by_unique_id_then_none_is_returned_when_there_is_no_match(self): choice = self.sut.getChoiceByUniqueId(-1) self.assertEqual(choice, None)
class ChoiceFacade:
_uniqueIdGenerator = None
_choiceRepository = None
def __init__(self):
self._uniqueIdGenerator = UniqueIdGenerator()
self._choiceRepository = ChoiceRepository()
def create(self, choice, poll):
choice.uniqueId = self._getUnusedChoiceUuid()
choice.poll = poll
self._choiceRepository.create(choice)
def _getUnusedChoiceUuid(self):
for i in range(100):
candidateUuid = self._uniqueIdGenerator.createUniqueId()
existingChoice = self._choiceRepository.getChoiceByUniqueId(candidateUuid)
if existingChoice is None:
break
return candidateUuid
def setUp(self): self.sut = ChoiceRepository() self.poll = Poll(uniqueId = self.getUniqueId()) self.poll.save()
def __init__(self):
self._uniqueIdGenerator = UniqueIdGenerator()
self._choiceRepository = ChoiceRepository()