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)
