def allocate_ids(self, project, key_pbs): from gcloud.datastore.test_connection import _KeyProto self._alloc_cw.append((project, key_pbs)) num_pbs = len(key_pbs) return [_KeyProto(i) for i in list(range(num_pbs))]