package com.battery.business.mapper;
|
|
import static org.junit.jupiter.api.Assertions.*;
|
|
import java.util.List;
|
import java.util.UUID;
|
import com.github.pagehelper.PageHelper;
|
import com.github.pagehelper.PageInfo;
|
import com.github.pagehelper.PageInterceptor;
|
import org.apache.ibatis.builder.xml.XMLMapperBuilder;
|
import org.apache.ibatis.datasource.unpooled.UnpooledDataSource;
|
import org.apache.ibatis.mapping.Environment;
|
import org.apache.ibatis.session.Configuration;
|
import org.apache.ibatis.session.SqlSession;
|
import org.apache.ibatis.session.SqlSessionFactoryBuilder;
|
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 com.battery.business.domain.TCollector;
|
import com.battery.business.domain.TDetectionCabinet;
|
import com.battery.business.domain.vo.DeviceRelatedTestVo;
|
|
/** 使用真实映射和内存数据库验证绑定查询，不连接共享开发库。 */
|
class DeviceRelatedTestMapperTest
|
{
|
private SqlSession session;
|
|
@BeforeEach
|
void setUp() throws Exception
|
{
|
var source = new UnpooledDataSource("org.h2.Driver", "jdbc:h2:mem:" + UUID.randomUUID() + ";MODE=MySQL", "sa", "");
|
var configuration = new Configuration(new Environment("test", new JdbcTransactionFactory(), source));
|
var pagination = new PageInterceptor();
|
var properties = new java.util.Properties();
|
properties.setProperty("helperDialect", "mysql");
|
pagination.setProperties(properties);
|
configuration.addInterceptor(pagination);
|
configuration.getTypeAliasRegistry().registerAliases("com.battery.business.domain");
|
for (String name : List.of("DeviceRelatedTestMapper", "TDetectionCabinetMapper", "TCollectorMapper"))
|
{
|
String resource = "mapper/battery/" + name + ".xml";
|
try (var input = getClass().getClassLoader().getResourceAsStream(resource))
|
{
|
new XMLMapperBuilder(input, configuration, resource, configuration.getSqlFragments()).parse();
|
}
|
}
|
session = new SqlSessionFactoryBuilder().build(configuration).openSession();
|
try (var sql = session.getConnection().createStatement())
|
{
|
sql.execute("create table t_test(test_id bigint primary key, test_name varchar(100), test_status varchar(30), del_flag int)");
|
sql.execute("insert into t_test values (1,'Waiting','WAITING',0),(2,'Running','RUNNING',0),(3,'Paused','PAUSED',0),(4,'Completed','COMPLETED',0),(5,'Deleted','RUNNING',1),(6,'Other device','WAITING',0)");
|
sql.execute("create table t_detection_channel(channel_id bigint primary key, cabinet_id bigint, subdevid int, chlid int, del_flag int)");
|
sql.execute("insert into t_detection_channel values (11,1,1,1,0),(12,1,1,2,0),(13,1,2,1,0),(21,2,1,1,0)");
|
sql.execute("create table t_test_channel_binding(test_id bigint, channel_id bigint)");
|
sql.execute("insert into t_test_channel_binding values (1,11),(1,12),(1,13),(2,11),(3,12),(4,11),(5,11),(6,21),(1,21)");
|
sql.execute("create table t_test_collector_channel(test_id bigint, collector_id bigint, channel_code varchar(100))");
|
sql.execute("insert into t_test_collector_channel values (1,1,'S-01'),(1,1,'S-02'),(2,1,'S-01'),(3,1,'S-02'),(4,1,'S-01'),(5,1,'S-01'),(6,2,'S-03'),(1,2,'S-03')");
|
String common = "adapter_id bigint, ip_address varchar(45), mac_address varchar(20), manufacturer varchar(100), model varchar(100), channel_count int, online_status varchar(10), status varchar(1), create_by varchar(64), create_time timestamp, update_by varchar(64), update_time timestamp, remark varchar(200), del_flag int default 0";
|
sql.execute("create table t_detection_cabinet(cabinet_id bigint primary key, adapter_code varchar(64), cabinet_code varchar(64), cabinet_name varchar(100), capacity decimal, registration_status varchar(20) default 'REGISTERED', " + common + ")");
|
sql.execute("insert into t_detection_cabinet(cabinet_id,cabinet_name,online_status) values (1,'Cabinet','OFFLINE'),(3,'Unused','ONLINE')");
|
sql.execute("create table t_collector(collector_id bigint primary key, collector_code varchar(64), collector_name varchar(100), last_seen_at_utc timestamp, " + common + ")");
|
sql.execute("insert into t_collector(collector_id,collector_name,online_status) values (1,'Collector','ONLINE'),(3,'Unused','OFFLINE')");
|
}
|
}
|
|
@AfterEach
|
void tearDown()
|
{
|
PageHelper.clearPage();
|
if (session != null) session.close();
|
}
|
|
@Test
|
void cabinetIncludesAllTestStatesOnceAndOnlyItsOwnChannels()
|
{
|
var tests = session.getMapper(DeviceRelatedTestMapper.class).selectCabinetRelatedTests(1L);
|
assertStates(tests);
|
assertEquals(List.of("1-1", "1-2", "2-1"), tests.get(3).getChannels());
|
assertEquals("Waiting", tests.get(3).getTestName());
|
assertTrue(session.getMapper(DeviceRelatedTestMapper.class).selectCabinetRelatedTests(3L).isEmpty());
|
}
|
|
@Test
|
void collectorIncludesAllTestStatesOnceAndOnlyItsOwnChannels()
|
{
|
var tests = session.getMapper(DeviceRelatedTestMapper.class).selectCollectorRelatedTests(1L);
|
assertStates(tests);
|
assertEquals(List.of("S-01", "S-02"), tests.get(3).getChannels());
|
assertTrue(session.getMapper(DeviceRelatedTestMapper.class).selectCollectorRelatedTests(3L).isEmpty());
|
}
|
|
@Test
|
void deviceListAndDetailAttachTheSameRelationsWithoutMultiplyingDeviceRows()
|
{
|
var cabinets = session.getMapper(TDetectionCabinetMapper.class);
|
assertEquals(2, cabinets.selectDetectionCabinetList(new TDetectionCabinet()).size());
|
assertStates(cabinets.selectDetectionCabinetById(1L).getRelatedTests());
|
assertTrue(cabinets.selectDetectionCabinetById(3L).getRelatedTests().isEmpty());
|
var collectors = session.getMapper(TCollectorMapper.class);
|
assertEquals(2, collectors.selectCollectorList(new TCollector()).size());
|
assertStates(collectors.selectCollectorById(1L).getRelatedTests());
|
assertTrue(collectors.selectCollectorById(3L).getRelatedTests().isEmpty());
|
}
|
|
@Test
|
void devicePaginationKeepsTotalsAndCompleteRelatedTestCollections()
|
{
|
PageHelper.startPage(2, 1);
|
var cabinets = session.getMapper(TDetectionCabinetMapper.class).selectDetectionCabinetList(new TDetectionCabinet());
|
assertEquals(2, new PageInfo<>(cabinets).getTotal());
|
assertEquals(1, cabinets.size());
|
assertEquals(1L, cabinets.get(0).getCabinetId());
|
assertStates(cabinets.get(0).getRelatedTests());
|
assertEquals(List.of("1-1", "1-2", "2-1"), cabinets.get(0).getRelatedTests().get(3).getChannels());
|
|
PageHelper.startPage(2, 1);
|
var collectors = session.getMapper(TCollectorMapper.class).selectCollectorList(new TCollector());
|
assertEquals(2, new PageInfo<>(collectors).getTotal());
|
assertEquals(1, collectors.size());
|
assertEquals(1L, collectors.get(0).getCollectorId());
|
assertStates(collectors.get(0).getRelatedTests());
|
assertEquals(List.of("S-01", "S-02"), collectors.get(0).getRelatedTests().get(3).getChannels());
|
}
|
|
@Test
|
void erasedTestDisappearsFromBothDevices() throws Exception
|
{
|
try (var sql = session.getConnection().createStatement())
|
{
|
sql.execute("delete from t_test where test_id = 1");
|
}
|
var mapper = session.getMapper(DeviceRelatedTestMapper.class);
|
assertEquals(List.of(4L, 3L, 2L), mapper.selectCabinetRelatedTests(1L).stream().map(DeviceRelatedTestVo::getTestId).toList());
|
assertEquals(List.of(4L, 3L, 2L), mapper.selectCollectorRelatedTests(1L).stream().map(DeviceRelatedTestVo::getTestId).toList());
|
}
|
|
private void assertStates(List<DeviceRelatedTestVo> tests)
|
{
|
assertEquals(List.of(4L, 3L, 2L, 1L), tests.stream().map(DeviceRelatedTestVo::getTestId).toList());
|
assertEquals(List.of("COMPLETED", "PAUSED", "RUNNING", "WAITING"), tests.stream().map(DeviceRelatedTestVo::getTestStatus).toList());
|
}
|
}
|
