package cn.iocoder.yudao.module.mes.service.pro.scan;
|
|
import cn.iocoder.yudao.module.mes.dal.mysql.pro.bagcode.MesProBagCodeMapper;
|
import cn.iocoder.yudao.module.mes.dal.mysql.pro.batch.MesProBatchMapper;
|
import cn.iocoder.yudao.module.mes.dal.mysql.pro.pallet.MesProPalletMapper;
|
import org.apache.ibatis.annotations.Update;
|
import org.h2.jdbcx.JdbcDataSource;
|
import org.junit.jupiter.api.BeforeEach;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.jdbc.core.JdbcTemplate;
|
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
|
import java.util.HashMap;
|
import java.util.Map;
|
import java.util.UUID;
|
import static org.junit.jupiter.api.Assertions.*;
|
|
/** 执行 Mapper 中的实际 SQL，验证混扫时状态与袋数的结果。 */
|
class InboundCompletionSqlTest {
|
private JdbcTemplate db;
|
private NamedParameterJdbcTemplate named;
|
@BeforeEach void setup() {
|
JdbcDataSource source = new JdbcDataSource();
|
source.setURL("jdbc:h2:mem:" + UUID.randomUUID() + ";MODE=MySQL;DB_CLOSE_DELAY=-1");
|
db = new JdbcTemplate(source); named = new NamedParameterJdbcTemplate(source);
|
db.execute("CREATE TABLE mes_pro_batch(id BIGINT PRIMARY KEY,status INT,deleted INT DEFAULT 0,inbound_time TIMESTAMP,inbound_event_id BIGINT,inbound_device_code VARCHAR(64),inbound_line_code VARCHAR(64),inbound_bag_quantity INT,update_time TIMESTAMP)");
|
db.execute("CREATE TABLE mes_pro_pallet(id BIGINT PRIMARY KEY,batch_id BIGINT,status INT,deleted INT DEFAULT 0,inbound_time TIMESTAMP,inbound_event_id BIGINT,update_time TIMESTAMP)");
|
db.execute("CREATE TABLE mes_pro_bag_code(id BIGINT PRIMARY KEY,batch_id BIGINT,pallet_id BIGINT,status INT,deleted INT DEFAULT 0,inbound_time TIMESTAMP,inbound_event_id BIGINT,inbound_device_code VARCHAR(64),inbound_line_code VARCHAR(64),update_time TIMESTAMP)");
|
db.update("INSERT INTO mes_pro_batch(id,status) VALUES(1,20)");
|
db.update("INSERT INTO mes_pro_pallet(id,batch_id,status) VALUES(1,1,10),(2,1,10)");
|
db.update("INSERT INTO mes_pro_bag_code(id,batch_id,pallet_id,status) VALUES(1,1,1,20),(2,1,1,20),(3,1,2,20),(4,1,2,30)");
|
}
|
private int run(Class<?> mapper, String method, Map<String,Object> values) {
|
var target = java.util.Arrays.stream(mapper.getMethods()).filter(m -> m.getName().equals(method)).findFirst().orElseThrow();
|
String sql = String.join(" ", target.getAnnotation(Update.class).value()).replaceAll("#\\{(\\w+)\\}", ":$1");
|
return named.update(sql, values);
|
}
|
private Map<String,Object> args(Long id) {
|
Map<String,Object> values = new HashMap<>();
|
values.put("id",id);values.put("batchId",1L);values.put("palletId",id);
|
values.put("eventId",10L);values.put("deviceCode","PDA");values.put("lineCode","L1");
|
return values;
|
}
|
private void refresh() {
|
run(MesProPalletMapper.class,"markCompletedInboundByBatchId",args(1L));
|
run(MesProBatchMapper.class,"refreshInboundProgress",args(1L));
|
}
|
private int batchValue(String column) { return db.queryForObject("SELECT " + column + " FROM mes_pro_batch WHERE id=1",Integer.class); }
|
@Test void bagThenPalletThenLastBagCompletesOnlyAtTheEnd() {
|
run(MesProBagCodeMapper.class,"markInbound",args(1L));refresh();
|
assertEquals(20,batchValue("status")); assertEquals(1,batchValue("inbound_bag_quantity"));
|
assertNull(db.queryForObject("SELECT inbound_event_id FROM mes_pro_pallet WHERE id=1",Long.class));
|
assertEquals(1,run(MesProBagCodeMapper.class,"markInboundByPalletId",args(1L)));
|
refresh();assertEquals(20,batchValue("status"));assertEquals(2,batchValue("inbound_bag_quantity"));
|
assertEquals(10L,db.queryForObject("SELECT inbound_event_id FROM mes_pro_pallet WHERE id=1",Long.class));
|
run(MesProBagCodeMapper.class,"markInbound",args(3L));refresh();
|
assertEquals(30,batchValue("status"));assertEquals(3,batchValue("inbound_bag_quantity"));
|
assertEquals(10L,db.queryForObject("SELECT inbound_event_id FROM mes_pro_pallet WHERE id=2",Long.class));
|
assertNull(db.queryForObject("SELECT inbound_event_id FROM mes_pro_bag_code WHERE id=4",Long.class));
|
assertEquals(0,run(MesProBagCodeMapper.class,"markInboundByBatchId",args(1L)));
|
refresh();assertEquals(3,batchValue("inbound_bag_quantity"));
|
}
|
@Test void wholeBatchExcludesCanceledAndDeletedBags() {
|
db.update("UPDATE mes_pro_bag_code SET deleted=1 WHERE id=2");
|
assertEquals(2,run(MesProBagCodeMapper.class,"markInboundByBatchId",args(1L)));
|
refresh();assertEquals(30,batchValue("status"));assertEquals(2,batchValue("inbound_bag_quantity"));
|
}
|
@Test void emptyBatchDoesNotCompleteAndCanceledBatchDoesNotChange() {
|
db.update("DELETE FROM mes_pro_bag_code");refresh();assertEquals(20,batchValue("status"));
|
db.update("UPDATE mes_pro_batch SET status=40");refresh();assertEquals(40,batchValue("status"));
|
}
|
}
|
