package com.tcm.training.admin.bootstrap;
|
|
import com.baomidou.dynamic.datasource.DynamicRoutingDataSource;
|
import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DataSourceProperty;
|
import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceProperties;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.boot.DefaultApplicationArguments;
|
import org.springframework.beans.factory.ObjectProvider;
|
|
import java.util.LinkedHashMap;
|
import java.util.Map;
|
|
import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.junit.jupiter.api.Assertions.assertThrows;
|
import static org.mockito.Mockito.mock;
|
import static org.mockito.Mockito.when;
|
|
/**
|
* 海诊通动态数据源启动校验测试。
|
*/
|
class HaizhentongDataSourceStartupValidatorTest {
|
|
/**
|
* 缺少海诊通数据源配置时应在启动阶段直接失败。
|
*/
|
@Test
|
void runShouldFailWhenHaizhentongConfigMissing() {
|
DynamicDataSourceProperties properties = buildProperties(false);
|
HaizhentongDataSourceStartupValidator validator = new HaizhentongDataSourceStartupValidator(
|
properties,
|
emptyProvider()
|
);
|
|
IllegalStateException exception = assertThrows(IllegalStateException.class,
|
() -> validator.run(new DefaultApplicationArguments(new String[0])));
|
|
assertEquals("海诊通数据源 haizhentong 未配置，请检查当前激活环境的配置文件", exception.getMessage());
|
}
|
|
/**
|
* 配置存在但运行期未注册海诊通数据源时也应直接失败。
|
*/
|
@Test
|
void runShouldFailWhenHaizhentongDatasourceNotRegistered() {
|
DynamicDataSourceProperties properties = buildProperties(true);
|
DynamicRoutingDataSource routingDataSource = mock(DynamicRoutingDataSource.class);
|
when(routingDataSource.getDataSources()).thenReturn(new LinkedHashMap<String, javax.sql.DataSource>());
|
HaizhentongDataSourceStartupValidator validator = new HaizhentongDataSourceStartupValidator(
|
properties,
|
fixedProvider(routingDataSource)
|
);
|
|
IllegalStateException exception = assertThrows(IllegalStateException.class,
|
() -> validator.run(new DefaultApplicationArguments(new String[0])));
|
|
assertEquals("海诊通数据源 haizhentong 未注册成功，请检查多数据源配置", exception.getMessage());
|
}
|
|
/**
|
* 配置与运行期注册都完整时应通过启动校验。
|
*/
|
@Test
|
void runShouldPassWhenHaizhentongDatasourceConfiguredAndRegistered() {
|
DynamicDataSourceProperties properties = buildProperties(true);
|
DynamicRoutingDataSource routingDataSource = mock(DynamicRoutingDataSource.class);
|
Map<String, javax.sql.DataSource> dataSourceMap = new LinkedHashMap<String, javax.sql.DataSource>();
|
dataSourceMap.put("master", mock(javax.sql.DataSource.class));
|
dataSourceMap.put("haizhentong", mock(javax.sql.DataSource.class));
|
when(routingDataSource.getDataSources()).thenReturn(dataSourceMap);
|
HaizhentongDataSourceStartupValidator validator = new HaizhentongDataSourceStartupValidator(
|
properties,
|
fixedProvider(routingDataSource)
|
);
|
|
assertDoesNotThrow(() -> validator.run(new DefaultApplicationArguments(new String[0])));
|
}
|
|
/**
|
* 存在海诊通配置但当前上下文拿不到动态路由 Bean 时，应跳过运行期校验而不是阻断启动。
|
*/
|
@Test
|
void runShouldPassWhenRoutingDatasourceBeanUnavailable() {
|
DynamicDataSourceProperties properties = buildProperties(true);
|
HaizhentongDataSourceStartupValidator validator = new HaizhentongDataSourceStartupValidator(
|
properties,
|
emptyProvider()
|
);
|
|
assertDoesNotThrow(() -> validator.run(new DefaultApplicationArguments(new String[0])));
|
}
|
|
/**
|
* 构造动态数据源配置对象，模拟 application.yml 中的多数据源声明。
|
*/
|
private DynamicDataSourceProperties buildProperties(boolean includeHaizhentong) {
|
DynamicDataSourceProperties properties = new DynamicDataSourceProperties();
|
Map<String, DataSourceProperty> datasource = new LinkedHashMap<String, DataSourceProperty>();
|
datasource.put("master", new DataSourceProperty());
|
if (includeHaizhentong) {
|
datasource.put("haizhentong", new DataSourceProperty());
|
}
|
properties.setDatasource(datasource);
|
return properties;
|
}
|
|
/**
|
* 构造返回固定 Bean 的 ObjectProvider 测试桩。
|
*/
|
private ObjectProvider<DynamicRoutingDataSource> fixedProvider(DynamicRoutingDataSource routingDataSource) {
|
return new SimpleObjectProvider(routingDataSource);
|
}
|
|
/**
|
* 构造返回空值的 ObjectProvider 测试桩。
|
*/
|
private ObjectProvider<DynamicRoutingDataSource> emptyProvider() {
|
return new SimpleObjectProvider(null);
|
}
|
|
/**
|
* 以最小实现满足测试对 ObjectProvider 的使用。
|
*/
|
private static class SimpleObjectProvider implements ObjectProvider<DynamicRoutingDataSource> {
|
|
private final DynamicRoutingDataSource value;
|
|
/**
|
* 注入当前测试需要暴露的动态路由 Bean。
|
*/
|
private SimpleObjectProvider(DynamicRoutingDataSource value) {
|
this.value = value;
|
}
|
|
/**
|
* 按固定值返回 Bean，模拟 Spring 容器中的直接获取。
|
*/
|
@Override
|
public DynamicRoutingDataSource getObject() {
|
return value;
|
}
|
|
/**
|
* 按固定值返回 Bean，模拟 Spring 容器中的可选注入。
|
*/
|
@Override
|
public DynamicRoutingDataSource getObject(Object... args) {
|
return value;
|
}
|
|
/**
|
* 按固定值返回 Bean，模拟 Spring 容器中的可选注入。
|
*/
|
@Override
|
public DynamicRoutingDataSource getIfAvailable() {
|
return value;
|
}
|
|
/**
|
* 按固定值返回 Bean，模拟 Spring 容器中的唯一候选。
|
*/
|
@Override
|
public DynamicRoutingDataSource getIfUnique() {
|
return value;
|
}
|
|
/**
|
* 返回简单迭代结果，满足 ObjectProvider 的接口约束。
|
*/
|
@Override
|
public java.util.Iterator<DynamicRoutingDataSource> iterator() {
|
return value == null
|
? java.util.Collections.<DynamicRoutingDataSource>emptyList().iterator()
|
: java.util.Collections.singletonList(value).iterator();
|
}
|
}
|
}
|
