def setUp(self):
"""Initial data for the tests."""
self.nominee1, self.nominee2, self.nominee3 = UserFactory.create_batch(3, groups=['Rep'])
self.rep = UserFactory.create(groups=['Rep'])
self.admin = UserFactory.create(groups=['Admin', 'Rep'])
self.mozillian = UserFactory.create(groups=['Mozillians'])
poll_start = now() + timedelta(days=5)
self.admin_group = Group.objects.get(name='Admin')
self.rep_group = Group.objects.get(name='Rep')
self.poll = PollFactory.create(valid_groups=self.admin_group,
start=poll_start,
end=poll_start + timedelta(days=10),
comments_allowed=False)
self.range_poll = RangePollFactory(poll=self.poll)
self.range_poll_choice1 = RangePollChoiceFactory(range_poll=self.range_poll,
nominee=self.nominee1)
self.range_poll_choice2 = RangePollChoiceFactory(range_poll=self.range_poll,
nominee=self.nominee2)
self.radio_poll = RadioPollFactory(poll=self.poll)
self.radio_poll_choice1, self.radio_poll_choice2 = RadioPollChoiceFactory.create_batch(
2, radio_poll=self.radio_poll)
self.post_data = {'range_poll__1': 1,
'range_poll__2': 2,
'radio_poll__1': 2}
self.edit_future_data = {
'name': u'Test edit voting',
'description': u'This is a description.',
'created_by': self.poll.created_by.id,
'valid_groups': self.admin_group.id,
'start_form_0_year': now().year + 1,
'start_form_0_month': 10,
'start_form_0_day': 1,
'start_form_1_hour': 7,
'start_form_1_minute': 00,
'end_form_0_year': now().year + 1,
'end_form_0_month': 10,
'end_form_0_day': 4,
'end_form_1_hour': 7,
'end_form_1_minute': 00,
'range_polls-TOTAL_FORMS': u'1',
'range_polls-INITIAL_FORMS': u'1',
'range_polls-MAX_NUM_FORMS': u'1000',
'range_polls-0-id': self.range_poll.id,
'range_polls-0-name': u'Current Range Poll 1',
'{0}_range_choices-0-id'.format(self.range_poll.id): self.range_poll_choice1.id,
'{0}_range_choices-0-nominee'.format(self.range_poll.id): self.nominee1.id,
'{0}_range_choices-0-DELETE'.format(self.range_poll.id): False,
'{0}_range_choices-1-id'.format(self.range_poll.id): self.range_poll_choice2.id,
'{0}_range_choices-1-nominee'.format(self.range_poll.id): self.nominee2.id,
'{0}_range_choices-1-DELETE'.format(self.range_poll.id): False,
'{0}_range_choices-2-id'.format(self.range_poll.id): u'',
'{0}_range_choices-2-nominee'.format(self.range_poll.id): self.nominee3.id,
'{0}_range_choices-2-DELETE'.format(self.range_poll.id): False,
'{0}_range_choices-TOTAL_FORMS'.format(self.range_poll.id): u'3',
'{0}_range_choices-INITIAL_FORMS'.format(self.range_poll.id): u'2',
'{0}_range_choices-TOTAL_FORMS'.format(self.range_poll.id): u'1000',
'radio_polls-0-id': self.radio_poll.id,
'radio_polls-0-question': u'Radio Poll - Question 1',
'radio_polls-TOTAL_FORMS': u'1',
'radio_polls-INITIAL_FORMS': u'1',
'radio_polls-MAX_NUM_FORMS': u'1000',
'{0}_radio_choices-TOTAL_FORMS'.format(self.radio_poll.id): u'2',
'{0}_radio_choices-INITIAL_FORMS'.format(self.radio_poll.id): u'2',
'{0}_radio_choices-MAX_NUM_FORMS'.format(self.radio_poll.id): u'1000',
'{0}_radio_choices-0-id'.format(self.radio_poll.id): self.radio_poll_choice1.id,
'{0}_radio_choices-0-answer'.format(self.radio_poll.id): u'Radio Poll - Answer 1',
'{0}_radio_choices-0-DELETE'.format(self.radio_poll.id): False,
'{0}_radio_choices-1-id'.format(self.radio_poll.id): self.radio_poll_choice2.id,
'{0}_radio_choices-1-answer'.format(self.radio_poll.id): u'Radio Poll - Answer 2',
'{0}_radio_choices-1-DELETE'.format(self.radio_poll.id): False}
self.edit_current_data = {
'name': u'Test edit voting',
'description': u'This is a description.',
'created_by': self.nominee1.id,
'valid_groups': self.admin_group.id,
'start_form_0_year': 2011,
'end_form_0_year': now().year,
'end_form_0_month': 10,
'end_form_0_day': 4,
'end_form_1_hour': 7,
'end_form_1_minute': 00}
# Give permissions to admin group
group = Group.objects.get(name='Admin')
permissions = Permission.objects.filter(name__icontains='poll')
for perm in permissions:
group.permissions.add(perm)
