DATACMNS-1764 - Allow registration of RepositoryFactory customizers.

Original Pull Request: #455
This commit is contained in:
Mark Paluch
2020-07-08 15:55:40 +02:00
committed by Christoph Strobl
parent e77947f0a5
commit 35a3d3cde9
3 changed files with 82 additions and 1 deletions

View File

@@ -18,6 +18,11 @@ package org.springframework.data.repository.core.support;
import static org.assertj.core.api.Assertions.*;
import static org.mockito.Mockito.*;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import org.aopalliance.intercept.MethodInterceptor;
import org.aopalliance.intercept.MethodInvocation;
import org.junit.jupiter.api.Test;
import org.springframework.data.querydsl.QuerydslPredicateExecutor;
@@ -69,6 +74,29 @@ class RepositoryFactoryBeanSupportUnitTests {
assertThat(information.getQueryMethods()).isEmpty();
}
@Test // DATACMNS-1688
void customizesRepositoryFactory() {
RepositoryFactoryBeanSupport<SampleRepository, Object, Long> factoryBean = //
new DummyRepositoryFactoryBean<>(SampleRepository.class);
factoryBean.addRepositoryFactoryCustomizer(repositoryFactory -> repositoryFactory
.addRepositoryProxyPostProcessor((factory, repositoryInformation) -> factory.addAdvice(new MethodInterceptor() {
@Nullable
@Override
public Object invoke(@Nonnull MethodInvocation invocation) {
throw new UnsupportedOperationException();
}
})));
factoryBean.afterPropertiesSet();
try {
factoryBean.getObject().someMethod();
fail("Missing UnsupportedOperationException");
} catch (UnsupportedOperationException e) {
// all good
}
}
@Test // DATACMNS-1345
void reportsMappingContextUnavailableForPersistentEntityLookup() {
@@ -87,7 +115,11 @@ class RepositoryFactoryBeanSupportUnitTests {
.isThrownBy(() -> bean.getPersistentEntity());
}
interface SampleRepository extends Repository<Object, Long> {}
interface SampleRepository extends Repository<Object, Long> {
void someMethod();
}
interface SampleWithQuerydslRepository extends Repository<Object, Long>, QuerydslPredicateExecutor<Object> {}
}