72 lines
3.1 KiB
Java
72 lines
3.1 KiB
Java
|
package com.njcn.middle.stream.autoconfig;
|
|||
|
|
|
|||
|
|
import com.njcn.middle.stream.container.StreamAutoClaimer;
|
|||
|
|
import com.njcn.middle.stream.container.StreamListenerContainer;
|
|||
|
|
import com.njcn.middle.stream.support.StreamKeyBuilder;
|
|||
|
|
import com.njcn.middle.stream.support.StreamTrimScheduler;
|
|||
|
|
import com.njcn.middle.stream.template.RedisStreamEnhanceTemplate;
|
|||
|
|
import org.junit.jupiter.api.Test;
|
|||
|
|
import org.springframework.boot.autoconfigure.AutoConfigurations;
|
|||
|
|
import org.springframework.boot.autoconfigure.data.redis.RedisAutoConfiguration;
|
|||
|
|
import org.springframework.boot.test.context.runner.ApplicationContextRunner;
|
|||
|
|
import org.springframework.context.annotation.Bean;
|
|||
|
|
import org.springframework.context.annotation.Configuration;
|
|||
|
|
import org.springframework.data.redis.connection.RedisConnectionFactory;
|
|||
|
|
import org.springframework.data.redis.core.StringRedisTemplate;
|
|||
|
|
|
|||
|
|
import static org.assertj.core.api.Assertions.assertThat;
|
|||
|
|
import static org.mockito.Mockito.mock;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 自动装配测试:验证 stream starter 的 Bean 装配、用户自定义回退、以及缺 {@link StringRedisTemplate}
|
|||
|
|
* 时整类优雅退场（与 cache 装配口径对齐,不因缺依赖启动失败）。
|
|||
|
|
*/
|
|||
|
|
class RedisStreamAutoConfigurationTest {
|
|||
|
|
|
|||
|
|
private final ApplicationContextRunner runner = new ApplicationContextRunner()
|
|||
|
|
.withConfiguration(AutoConfigurations.of(
|
|||
|
|
RedisAutoConfiguration.class, RedisStreamAutoConfiguration.class))
|
|||
|
|
.withPropertyValues("spring.redis.host=127.0.0.1", "spring.redis.port=6379");
|
|||
|
|
|
|||
|
|
@Test
|
|||
|
|
void registersStreamBeans() {
|
|||
|
|
runner.run(ctx -> {
|
|||
|
|
assertThat(ctx).hasSingleBean(StreamKeyBuilder.class);
|
|||
|
|
assertThat(ctx).hasSingleBean(RedisStreamEnhanceTemplate.class);
|
|||
|
|
assertThat(ctx).hasSingleBean(StreamListenerContainer.class);
|
|||
|
|
assertThat(ctx).hasSingleBean(StreamAutoClaimer.class);
|
|||
|
|
assertThat(ctx).hasSingleBean(StreamTrimScheduler.class);
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@Test
|
|||
|
|
void backsOffWhenUserDefinesOwnTemplate() {
|
|||
|
|
runner.withUserConfiguration(CustomConfig.class).run(ctx -> {
|
|||
|
|
assertThat(ctx).hasSingleBean(RedisStreamEnhanceTemplate.class);
|
|||
|
|
assertThat(ctx.getBean(RedisStreamEnhanceTemplate.class))
|
|||
|
|
.isSameAs(ctx.getBean("myTemplate"));
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@Test
|
|||
|
|
void backsOffWhenNoStringRedisTemplateBean() {
|
|||
|
|
// 不装配 RedisAutoConfiguration,上下文内无 StringRedisTemplate Bean:
|
|||
|
|
// 应整类 back off 优雅退场,而非因缺依赖启动失败（与 cache 装配口径一致）。
|
|||
|
|
new ApplicationContextRunner()
|
|||
|
|
.withConfiguration(AutoConfigurations.of(RedisStreamAutoConfiguration.class))
|
|||
|
|
.run(ctx -> assertThat(ctx)
|
|||
|
|
.hasNotFailed()
|
|||
|
|
.doesNotHaveBean(RedisStreamEnhanceTemplate.class));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@Configuration
|
|||
|
|
static class CustomConfig {
|
|||
|
|
@Bean
|
|||
|
|
RedisStreamEnhanceTemplate myTemplate() {
|
|||
|
|
return new RedisStreamEnhanceTemplate(
|
|||
|
|
new StringRedisTemplate(mock(RedisConnectionFactory.class)),
|
|||
|
|
new StreamKeyBuilder(false, ""));
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|