def testGetVotes(self): self.assertLen(self.blockable_1.GetVotes(), 0) self.assertLen(self.blockable_2.GetVotes(), 0) test_utils.CreateVotes(self.blockable_1, 3) test_utils.CreateVotes(self.blockable_2, 2) self.assertLen(self.blockable_1.GetVotes(), 3) self.assertLen(self.blockable_2.GetVotes(), 2)
def testGetVotes(self): self.assertEqual(0, len(self.blockable_1.GetVotes())) self.assertEqual(0, len(self.blockable_2.GetVotes())) test_utils.CreateVotes(self.blockable_1, 3) test_utils.CreateVotes(self.blockable_2, 2) self.assertEqual(3, len(self.blockable_1.GetVotes())) self.assertEqual(2, len(self.blockable_2.GetVotes()))
def testGetVotes_Inactive(self): self.assertLen(self.blockable_1.GetVotes(), 0) test_utils.CreateVotes(self.blockable_1, 2) self.assertLen(self.blockable_1.GetVotes(), 2) votes = vote_models.Vote.query().fetch() new_votes = [] for vote in votes: new_key = ndb.Key(flat=vote.key.flat()[:-1] + (None,)) new_votes.append(datastore_utils.CopyEntity(vote, new_key=new_key)) ndb.delete_multi(vote.key for vote in votes) ndb.put_multi(new_votes) self.assertLen(self.blockable_1.GetVotes(), 0)