package cn.iocoder.yudao.module.bi.service.config;
|
|
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
import cn.iocoder.yudao.module.bi.controller.admin.config.BiChartConfigController;
|
import cn.iocoder.yudao.module.bi.dal.dataobject.BiChartConfigDO;
|
import cn.iocoder.yudao.module.bi.dal.mysql.BiChartConfigMapper;
|
import com.baomidou.mybatisplus.annotation.DbType;
|
import com.baomidou.mybatisplus.core.MybatisConfiguration;
|
import com.baomidou.mybatisplus.core.MybatisSqlSessionFactoryBuilder;
|
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
|
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
|
import org.apache.ibatis.datasource.unpooled.UnpooledDataSource;
|
import org.apache.ibatis.mapping.Environment;
|
import org.apache.ibatis.session.SqlSession;
|
import org.apache.ibatis.transaction.jdbc.JdbcTransactionFactory;
|
import org.junit.jupiter.api.AfterEach;
|
import org.junit.jupiter.api.BeforeEach;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.test.util.ReflectionTestUtils;
|
|
import java.sql.Statement;
|
import java.util.List;
|
import java.util.UUID;
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
|
class BiChartConfigServiceImplTest {
|
|
private SqlSession session;
|
private BiChartConfigServiceImpl service;
|
|
@BeforeEach
|
void setUp() throws Exception {
|
UnpooledDataSource dataSource = new UnpooledDataSource("org.h2.Driver",
|
"jdbc:h2:mem:chart_" + UUID.randomUUID() + ";MODE=MySQL", "sa", "");
|
MybatisConfiguration configuration = new MybatisConfiguration();
|
configuration.setMapUnderscoreToCamelCase(true);
|
configuration.setEnvironment(new Environment("test", new JdbcTransactionFactory(), dataSource));
|
MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
|
interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.H2));
|
configuration.addInterceptor(interceptor);
|
configuration.addMapper(BiChartConfigMapper.class);
|
session = new MybatisSqlSessionFactoryBuilder().build(configuration).openSession(true);
|
try (Statement statement = session.getConnection().createStatement()) {
|
statement.execute("""
|
CREATE TABLE bi_chart_config (
|
id BIGINT PRIMARY KEY, dashboard_id BIGINT, name VARCHAR(255),
|
chart_type VARCHAR(50), data_source_id BIGINT, query_sql VARCHAR(2000),
|
position VARCHAR(500), chart_options VARCHAR(2000), refresh_interval INT,
|
sort INT, status INT, creator VARCHAR(64), create_time TIMESTAMP,
|
updater VARCHAR(64), update_time TIMESTAMP, deleted INT DEFAULT 0
|
)
|
""");
|
statement.execute("""
|
INSERT INTO bi_chart_config (id,dashboard_id,name,sort,deleted) VALUES
|
(1,10,'月度销售趋势',2,0),
|
(2,10,'销售金额统计',1,0),
|
(3,10,'采购统计',3,0),
|
(4,20,'销售排名',1,0),
|
(5,10,'已删除销售统计',0,1)
|
""");
|
}
|
service = new BiChartConfigServiceImpl();
|
ReflectionTestUtils.setField(service, "mapper", session.getMapper(BiChartConfigMapper.class));
|
}
|
|
@AfterEach
|
void tearDown() {
|
if (session != null) {
|
session.close();
|
}
|
}
|
|
@Test
|
void controllerSearchesPartialNameTogetherWithDashboardAndPaginates() {
|
BiChartConfigController controller = new BiChartConfigController();
|
ReflectionTestUtils.setField(controller, "chartConfigService", service);
|
|
PageResult<BiChartConfigDO> first = controller.getPage(10L, " 销售 ", 1, 1).getData();
|
assertEquals(2L, first.getTotal());
|
assertEquals(List.of(2L), ids(first));
|
PageResult<BiChartConfigDO> second = controller.getPage(10L, "销售", 2, 1).getData();
|
assertEquals(2L, second.getTotal());
|
assertEquals(List.of(1L), ids(second));
|
}
|
|
@Test
|
void searchesPartialNameAcrossDashboardsWhenDashboardIsAbsent() {
|
PageResult<BiChartConfigDO> page = service.getPage(null, "销售", 1, 10);
|
assertEquals(3L, page.getTotal());
|
assertEquals(List.of(1L, 2L, 4L), ids(page).stream().sorted().toList());
|
assertEquals(0L, service.getPage(null, "不存在的图表", 1, 10).getTotal());
|
}
|
|
@Test
|
void omittedOrClearedNameDoesNotFilterTheDashboard() {
|
for (String name : new String[]{null, "", " "}) {
|
PageResult<BiChartConfigDO> page = service.getPage(10L, name, 1, 10);
|
assertEquals(3L, page.getTotal());
|
assertEquals(List.of(2L, 1L, 3L), ids(page));
|
}
|
}
|
|
private List<Long> ids(PageResult<BiChartConfigDO> page) {
|
return page.getList().stream().map(BiChartConfigDO::getId).toList();
|
}
|
}
|
