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)