def testGetSlotTransferEntitiesForOrg(self): """Tests if all the slot transfer entities for an organization is returned. """ #Every organization has a single slot transfer entity. expected = self.slot_transfer_entities[0] actual = slot_transfer_logic.getSlotTransferEntitiesForOrg( self.org_entities[0]) self.assertEqual(actual[0].key(), expected.key()) self.assertNotEqual(actual[0].key(), self.slot_transfer_entities[1].key()) #Multiple slot transfer entities for an organization. All the entities #must be returned. expected = [ entity.key() for entity in self.gsoc_organization_slot_transfer_entities] slot_transfer_entities = slot_transfer_logic.getSlotTransferEntitiesForOrg( self.gsoc_organization) actual = [entity.key() for entity in slot_transfer_entities] self.assertEqual(expected, actual) #An organization has no slot transer entity expected = [] organization = seeder_logic.seed(GSoCOrganization, {'scope': self.gsoc_program}) actual = slot_transfer_logic.getSlotTransferEntitiesForOrg(organization) self.assertEqual(expected, actual)
def slotTransferEntities(self): assert access_checker.isSet(self.data.organization) self.data.slot_transfer_entities = \ slot_transfer_logic.getSlotTransferEntitiesForOrg( self.data.organization)
def slotTransferEntities(self): self.data.slot_transfer_entities = ( slot_transfer_logic.getSlotTransferEntitiesForOrg( self.data.url_ndb_org.key))