@SuppressWarnings("unchecked") @Test void testForEachBatched() { Entity permittedEntity = mock(Entity.class); Entity notPermittedEntity = mock(Entity.class); when(permissionChecker.isReadAllowed(permittedEntity)).thenReturn(true); Fetch fetch = mock(Fetch.class); doAnswer( invocation -> { ((Consumer<List<Entity>>) invocation.getArgument(1)) .accept(asList(permittedEntity, notPermittedEntity)); return null; }) .when(delegateRepository) .forEachBatched(eq(fetch), any(), eq(1000)); List<Entity> actualEntities = new ArrayList<>(); permissionCheckingDecorator.forEachBatched(fetch, actualEntities::addAll, 1000); assertEquals(singletonList(permittedEntity), actualEntities); }
@Override public void forEachBatched(Fetch fetch, Consumer<List<E>> consumer, int batchSize) { delegate() .forEachBatched( fetch, entities -> consumer.accept( entities.stream().filter(permissionChecker::isReadAllowed).collect(toList())), batchSize); }
PermissionCheckingDecorator extends AbstractRepositoryDecorator<E> { @Override public void forEachBatched(Fetch fetch, Consumer<List<E>> consumer, int batchSize) { delegate() .forEachBatched( fetch, entities -> consumer.accept( entities.stream().filter(permissionChecker::isReadAllowed).collect(toList())), batchSize); } }
PermissionCheckingDecorator extends AbstractRepositoryDecorator<E> { @Override public void forEachBatched(Fetch fetch, Consumer<List<E>> consumer, int batchSize) { delegate() .forEachBatched( fetch, entities -> consumer.accept( entities.stream().filter(permissionChecker::isReadAllowed).collect(toList())), batchSize); } PermissionCheckingDecorator(
      Repository<E> delegateRepository, PermissionChecker<E> permissionChecker); }
PermissionCheckingDecorator extends AbstractRepositoryDecorator<E> { @Override public void forEachBatched(Fetch fetch, Consumer<List<E>> consumer, int batchSize) { delegate() .forEachBatched( fetch, entities -> consumer.accept( entities.stream().filter(permissionChecker::isReadAllowed).collect(toList())), batchSize); } PermissionCheckingDecorator(
      Repository<E> delegateRepository, PermissionChecker<E> permissionChecker); @Override @Nonnull Iterator<E> iterator(); @Override void forEachBatched(Fetch fetch, Consumer<List<E>> consumer, int batchSize); @Override long count(); @Override long count(Query<E> q); @Override Stream<E> findAll(Query<E> q); @Override E findOne(Query<E> q); @Override E findOneById(Object id); @Override E findOneById(Object id, Fetch fetch); @Override Stream<E> findAll(Stream<Object> ids); @Override Stream<E> findAll(Stream<Object> ids, Fetch fetch); @Override void update(E entity); @Override void update(Stream<E> entities); @Override void delete(E entity); @Override void delete(Stream<E> entities); @Override void deleteById(Object id); @Transactional @Override void deleteAll(Stream<Object> ids); @Transactional @Override void deleteAll(); @Override void add(E entity); @Override Integer add(Stream<E> entities); @Override AggregateResult aggregate(AggregateQuery aggregateQuery); }
PermissionCheckingDecorator extends AbstractRepositoryDecorator<E> { @Override public void forEachBatched(Fetch fetch, Consumer<List<E>> consumer, int batchSize) { delegate() .forEachBatched( fetch, entities -> consumer.accept( entities.stream().filter(permissionChecker::isReadAllowed).collect(toList())), batchSize); } PermissionCheckingDecorator(
      Repository<E> delegateRepository, PermissionChecker<E> permissionChecker); @Override @Nonnull Iterator<E> iterator(); @Override void forEachBatched(Fetch fetch, Consumer<List<E>> consumer, int batchSize); @Override long count(); @Override long count(Query<E> q); @Override Stream<E> findAll(Query<E> q); @Override E findOne(Query<E> q); @Override E findOneById(Object id); @Override E findOneById(Object id, Fetch fetch); @Override Stream<E> findAll(Stream<Object> ids); @Override Stream<E> findAll(Stream<Object> ids, Fetch fetch); @Override void update(E entity); @Override void update(Stream<E> entities); @Override void delete(E entity); @Override void delete(Stream<E> entities); @Override void deleteById(Object id); @Transactional @Override void deleteAll(Stream<Object> ids); @Transactional @Override void deleteAll(); @Override void add(E entity); @Override Integer add(Stream<E> entities); @Override AggregateResult aggregate(AggregateQuery aggregateQuery);  }