Spring/spring-pulsar
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add producer interceptor cache key tests

Browse Source 
This commit adds a test to PulsarTemplateTests that ensures producers
are properly cached when the template is configured with one or more
producer interceptors.

(cherry picked from commit cffffa6496)
This commit is contained in:
Chris Bono
2024-03-01 22:54:43 -06:00
parent 739e89b175
commit 9b35311b30
1 changed files with 30 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
spring-pulsar/src/test/java/org/springframework/pulsar/core/PulsarTemplateTests.java
View File

@@ -198,7 +198,7 @@ class PulsarTemplateTests implements PulsarTestContainerSupport {
@ParameterizedTest(name = "{0}")
@MethodSource("interceptorInvocationTestProvider")
void interceptorInvocationTest(String topic, List<ProducerInterceptor> interceptors) {
void interceptorInvocationTest(String topic, List<ProducerInterceptor> interceptors) throws PulsarClientException {
PulsarProducerFactory<String> producerFactory = new DefaultPulsarProducerFactory<>(client, topic);
PulsarTemplate<String> pulsarTemplate = new PulsarTemplate<>(producerFactory, interceptors);
pulsarTemplate.send("test-interceptor");
@@ -215,6 +215,35 @@ class PulsarTemplateTests implements PulsarTestContainerSupport {
List.of(mock(ProducerInterceptor.class), mock(ProducerInterceptor.class))));
}
@Test
void interceptorUsedAsCacheKeyProperly() throws PulsarClientException {
var producerFactory = new CachingPulsarProducerFactory<String>(client, null, null, new DefaultTopicResolver(),
Duration.ofSeconds(10L), 10L, 10);
try {
var interceptors = List.of(mock(ProducerInterceptor.class));
var pulsarTemplate = new PulsarTemplate<>(producerFactory, interceptors);
assertCacheSize(producerFactory, 0);
for (int i = 0; i < 3; i++) {
pulsarTemplate.send("test-intercept-topic", "test-interceptor-" + i);
assertCacheSize(producerFactory, 1);
}
assertCacheSize(producerFactory, 1);
}
finally {
// The CPPF returns producers that do not actually close when the template
// calls close on them - destroy does close the producers though
if (producerFactory != null) {
producerFactory.destroy();
}
}
}
private <T> void assertCacheSize(CachingPulsarProducerFactory<T> producerFactory, int expectedSize) {
assertThat(producerFactory).extracting("producerCache.cache.cache")
.asInstanceOf(InstanceOfAssertFactories.MAP)
.hasSize(expectedSize);
}
@ParameterizedTest
@ValueSource(booleans = { true, false })
void sendMessageWithTopicInferredByTypeMappings(boolean producerFactoryHasDefaultTopic) throws Exception {