def test_downvote(self): self.client.login(username=self.user.username, password ='******') flagRanking = FlagRanking(flag = self.flag, toilet = self.toilet, up_down_vote = 1) flagRanking.save() flagvote = FlagVote(user = self.user, flag = self.flag, toilet = self.toilet, vote = 1) flagvote.save() response = json.loads(self.client.post('/api/flag/downvote/', {'toilet_pk' : self.toilet.pk, 'flag_pk' : self.flag.pk}).content)[0] self.assertEqual(response['fields']['flag'], self.flag.pk) self.assertEqual(response['fields']['up_down_vote'], 0)
def test_get_rankings(self): flagRanking = FlagRanking(flag=self.flag, toilet = self.toilet, up_down_vote = 0) flagRanking.save() response = json.loads(self.client.post('/api/FlagRanking/get/', {'filters' : json.dumps({'toilet' : self.toilet.pk}) }).content)[0] self.assertEqual(response['fields']['up_down_vote'], 0) self.assertEqual(response['fields']['toilet'], self.toilet.pk)