def setUpTestData(cls): cls.user = UserFactory.create() cls.test_campaigns = PollCampaignFactory.create_batch(size=2) cls.options = PollOptionFactory.create_batch( size=3, campaign=cls.test_campaigns[0], ) PollFactory.create(option=cls.options[0], user=cls.user)
def test_validate_one_poll_per_user(self): campaign = PollCampaignFactory.create() user = UserFactory.create() self.assertTrue(validate_one_poll_per_user(campaign, user)) option = PollOptionFactory.create(campaign=campaign) PollFactory.create(option=option, user=user) with self.assertRaises(ValidationError): validate_one_poll_per_user(campaign, user)
def test_poll_serializer(self): # write option = PollOptionFactory.create() user = UserFactory.create() data = { 'option': str(option.id), 'user': str(user.id), } serializer = PollSerializer(data=data) serializer.is_valid(raise_exception=True) poll = serializer.save() self.assertEqual(poll.option, option) self.assertEqual(poll.user, user) # no read data = PollSerializer(poll).data self.assertEqual(data, {})
def test_retrieve_user(self): user = UserFactory.create() view_name = 'v1:user-detail' with self.assertRaises(NoReverseMatch): self.client.get(reverse(view_name, kwargs={'pk': str(user.id)}))
def setUpTestData(cls): cls.user = UserFactory.create() cls.option = PollOptionFactory.create()
def setUpTestData(cls): cls.user = UserFactory.create() cls.campaign = PollCampaignFactory.create()