package cn.iocoder.yudao.module.mes.service.dv.spare;
|
|
import cn.iocoder.yudao.framework.common.exception.ServiceException;
|
import cn.iocoder.yudao.module.mes.controller.admin.dv.spare.vo.*;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.dv.spare.*;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.dv.repair.MesDvRepairDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.dv.machinery.MesDvMachineryDO;
|
import cn.iocoder.yudao.module.mes.dal.mysql.dv.spare.*;
|
import cn.iocoder.yudao.module.mes.dal.mysql.dv.repair.MesDvRepairMapper;
|
import cn.iocoder.yudao.module.mes.dal.mysql.dv.machinery.MesDvMachineryMapper;
|
import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
|
import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.aop.framework.ProxyFactory;
|
import org.springframework.jdbc.core.*;
|
import org.springframework.jdbc.datasource.*;
|
import org.springframework.test.util.ReflectionTestUtils;
|
import org.springframework.transaction.annotation.AnnotationTransactionAttributeSource;
|
import org.springframework.transaction.interceptor.TransactionInterceptor;
|
import java.lang.reflect.*;
|
import java.math.BigDecimal;
|
import java.util.*;
|
import java.util.concurrent.*;
|
import java.util.concurrent.atomic.AtomicBoolean;
|
import static org.junit.jupiter.api.Assertions.*;
|
|
class MesDvSpareServiceTest {
|
@Test void openingStockIsRecordedAndMetadataCannotOverwriteStock() {
|
var f=new Fixture();
|
var req=f.save().setInitialQuantity(new BigDecimal("12"));
|
Long id=f.service.create(req,1L);
|
assertEquals(new BigDecimal("12.000"),f.stock(id));
|
assertEquals(1,f.count());
|
req.setId(id).setInitialQuantity(new BigDecimal("999")).setName("New name");
|
f.service.update(req);
|
assertEquals(new BigDecimal("12.000"),f.stock(id));
|
req.setUnit("箱");
|
assertThrows(ServiceException.class,()->f.service.update(req));
|
}
|
@Test void draftRepairAllowsIssueAndPartialReturnsPreserveTrace() {
|
var f=new Fixture();var issue=f.issue(4);
|
Long id=f.service.issue(issue,1L);
|
assertEquals(new BigDecimal("6.000"),f.stock(1L));
|
f.service.returnPart(f.action(1).setOriginalId(id),1L);
|
f.service.returnPart(f.action(2).setOriginalId(id),1L);
|
assertEquals(new BigDecimal("9.000"),f.stock(1L));
|
assertEquals(3,f.count());
|
assertEquals("R-001",f.jdbc.queryForObject("select repair_code from movement where type=4 limit 1",String.class));
|
assertThrows(ServiceException.class,()->f.service.returnPart(f.action(2).setOriginalId(id),1L));
|
assertEquals(3,f.count());
|
}
|
@Test void insufficientStockAndCompletedRepairCannotIssue() {
|
var f=new Fixture();
|
assertThrows(ServiceException.class,()->f.service.issue(f.issue(11),1L));
|
f.jdbc.update("update repair set status=4");
|
assertThrows(ServiceException.class,()->f.service.issue(f.issue(1),1L));
|
assertEquals(new BigDecimal("10.000"),f.stock(1L));assertEquals(0,f.count());
|
}
|
@Test void repeatedRequestDoesNotDoubleDeductAndChangedPayloadIsRejected() {
|
var f=new Fixture();var req=f.issue(2);
|
assertEquals(f.service.issue(req,1L),f.service.issue(req,1L));
|
assertEquals(new BigDecimal("8.000"),f.stock(1L));assertEquals(1,f.count());
|
req.setQuantity(BigDecimal.ONE);
|
assertThrows(ServiceException.class,()->f.service.issue(req,1L));
|
}
|
@Test void receiveAndReturnAreIdempotentAndDisabledPartsStillReturn() {
|
var f=new Fixture();var receive=f.action(3).setSparePartId(1L);
|
assertEquals(f.service.receive(receive,1L),f.service.receive(receive,1L));
|
var id=f.service.issue(f.issue(2),1L);
|
f.jdbc.update("update part set status=1 where id=1");
|
assertThrows(ServiceException.class,()->f.service.receive(f.action(1).setSparePartId(1L),1L));
|
assertThrows(ServiceException.class,()->f.service.issue(f.issue(1),1L));
|
var back=f.action(2).setOriginalId(id);
|
assertEquals(f.service.returnPart(back,1L),f.service.returnPart(back,1L));
|
assertEquals(new BigDecimal("13.000"),f.stock(1L));assertEquals(3,f.count());
|
}
|
@Test void movementFailureRollsBackStockChange() {
|
var f=new Fixture();f.failWrite.set(true);
|
assertThrows(IllegalStateException.class,()->f.service.issue(f.issue(3),1L));
|
assertEquals(new BigDecimal("10.000"),f.stock(1L));assertEquals(0,f.count());
|
}
|
@Test void concurrentRequestsCannotOverdrawStock() throws Exception {
|
var f=new Fixture();var start=new CountDownLatch(1);
|
var pool=Executors.newFixedThreadPool(2);
|
try {
|
Callable<Boolean> job=()->{start.await();try{f.service.issue(f.issue(7),1L);return true;}catch(ServiceException e){return false;}};
|
var a=pool.submit(job);var b=pool.submit(job);start.countDown();
|
assertNotEquals(a.get(10,TimeUnit.SECONDS),b.get(10,TimeUnit.SECONDS));
|
assertEquals(new BigDecimal("3.000"),f.stock(1L));assertEquals(1,f.count());
|
} finally {pool.shutdownNow();}
|
}
|
@Test void concurrentReturnsCannotExceedOriginalQuantity() throws Exception {
|
var f=new Fixture();Long id=f.service.issue(f.issue(4),1L);var start=new CountDownLatch(1);var pool=Executors.newFixedThreadPool(2);
|
try {
|
Callable<Boolean> job=()->{start.await();try{f.service.returnPart(f.action(3).setOriginalId(id),1L);return true;}catch(ServiceException e){return false;}};
|
var a=pool.submit(job);var b=pool.submit(job);start.countDown();
|
assertNotEquals(a.get(10,TimeUnit.SECONDS),b.get(10,TimeUnit.SECONDS));
|
assertEquals(new BigDecimal("9.000"),f.stock(1L));assertEquals(2,f.count());
|
} finally {pool.shutdownNow();}
|
}
|
@Test void invalidQuantityUserAndHistoryDeletionAreBlocked() {
|
var f=new Fixture();
|
assertThrows(ServiceException.class,()->f.service.issue(f.issue(0),1L));
|
assertThrows(ServiceException.class,()->f.service.issue(f.issue(1).setReceiverUserId(9L),1L));
|
assertThrows(ServiceException.class,()->f.service.delete(1L));
|
f.service.issue(f.issue(1),1L);
|
f.jdbc.update("update part set stock_quantity=0 where id=1");
|
assertThrows(ServiceException.class,()->f.service.delete(1L));
|
}
|
@Test void invalidReturnAndWhitespaceReasonAreBlocked() {
|
var f=new Fixture();
|
assertThrows(ServiceException.class,()->f.service.returnPart(f.action(1).setOriginalId(99L),1L));
|
assertThrows(ServiceException.class,()->f.service.issue(f.issue(1).setRemark(" "),1L));
|
assertEquals(0,f.count());
|
}
|
@SuppressWarnings("unchecked") static <T>T stub(Class<T> type,InvocationHandler h) { return (T)Proxy.newProxyInstance(type.getClassLoader(),new Class<?>[]{type},h); }
|
static class Fixture {
|
final JdbcTemplate jdbc;
|
final MesDvSpareService service;
|
final AtomicBoolean failWrite=new AtomicBoolean();
|
Fixture() {
|
var ds=new DriverManagerDataSource("jdbc:h2:mem:"+UUID.randomUUID()+";MODE=MySQL;DB_CLOSE_DELAY=-1;LOCK_TIMEOUT=5000","sa","");
|
jdbc=new JdbcTemplate(ds);
|
jdbc.execute("create table part(id bigint auto_increment primary key,code varchar(64) unique,name varchar(128),unit varchar(32),stock_quantity decimal(18,3),safety_quantity decimal(18,3),status int)");
|
jdbc.execute("create table repair(id bigint primary key,code varchar(64),machinery_id bigint,status int)");
|
jdbc.execute("create table movement(id bigint auto_increment primary key,request_key varchar(64) unique,spare_part_id bigint,spare_part_code varchar(64),spare_part_name varchar(128),unit varchar(32),type int,quantity decimal(18,3),before_quantity decimal(18,3),after_quantity decimal(18,3),repair_id bigint,repair_code varchar(64),machinery_id bigint,machinery_name varchar(128),original_id bigint,receiver_user_id bigint,receiver_name varchar(128),operator_user_id bigint,remark varchar(500))");
|
jdbc.update("insert into part values(1,'S-001','轴承','件',10,2,0)");
|
jdbc.update("insert into repair values(1,'R-001',1,0)");
|
var target=new MesDvSpareService();
|
ReflectionTestUtils.setField(target,"partMapper",stub(MesDvSparePartMapper.class,(p,m,a)->switch(m.getName()) {
|
case "lock" -> onePart((Long)a[0],true);
|
case "selectById" -> onePart((Long)a[0],false);
|
case "selectOne" -> jdbc.query("select * from part where code=?",new BeanPropertyRowMapper<>(MesDvSparePartDO.class),a[1]).stream().findFirst().orElse(null);
|
case "insert" -> {var row=(MesDvSparePartDO)a[0];jdbc.update("insert into part(code,name,unit,stock_quantity,safety_quantity,status)values(?,?,?,?,?,?)",row.getCode(),row.getName(),row.getUnit(),row.getStockQuantity(),row.getSafetyQuantity(),row.getStatus());row.setId(jdbc.queryForObject("select id from part where code=?",Long.class,row.getCode()));yield 1;}
|
case "updateById" -> {var row=(MesDvSparePartDO)a[0];if(row.getCode()!=null)yield jdbc.update("update part set code=?,name=?,unit=?,status=?,safety_quantity=?,stock_quantity=? where id=?",row.getCode(),row.getName(),row.getUnit(),row.getStatus(),row.getSafetyQuantity(),row.getStockQuantity(),row.getId());yield jdbc.update("update part set stock_quantity=? where id=?",row.getStockQuantity(),row.getId());}
|
default -> throw new AssertionError(m.getName());
|
}));
|
ReflectionTestUtils.setField(target,"movementMapper",stub(MesDvSpareMovementMapper.class,(p,m,a)->switch(m.getName()){
|
case "byKey" -> jdbc.query("select * from movement where request_key=?",new BeanPropertyRowMapper<>(MesDvSpareMovementDO.class),a[0]).stream().findFirst().orElse(null);
|
case "lock" -> jdbc.query("select * from movement where id=? for update",new BeanPropertyRowMapper<>(MesDvSpareMovementDO.class),a[0]).stream().findFirst().orElse(null);
|
case "returns" -> jdbc.query("select * from movement where type=4",new BeanPropertyRowMapper<>(MesDvSpareMovementDO.class)).stream().filter(row->((Collection<?>)a[0]).contains(row.getOriginalId())).toList();
|
case "selectCount" -> jdbc.queryForObject("select count(*) from movement where spare_part_id=?",Long.class,a[1]);
|
case "insert" -> {if(failWrite.get())throw new IllegalStateException("simulated write failure");var row=(MesDvSpareMovementDO)a[0];jdbc.update("insert into movement(request_key,spare_part_id,spare_part_code,spare_part_name,unit,type,quantity,before_quantity,after_quantity,repair_id,repair_code,machinery_id,machinery_name,original_id,receiver_user_id,receiver_name,operator_user_id,remark)values(?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)",row.getRequestKey(),row.getSparePartId(),row.getSparePartCode(),row.getSparePartName(),row.getUnit(),row.getType(),row.getQuantity(),row.getBeforeQuantity(),row.getAfterQuantity(),row.getRepairId(),row.getRepairCode(),row.getMachineryId(),row.getMachineryName(),row.getOriginalId(),row.getReceiverUserId(),row.getReceiverName(),row.getOperatorUserId(),row.getRemark());row.setId(jdbc.queryForObject("select id from movement where request_key=?",Long.class,row.getRequestKey()));yield 1;}
|
default -> throw new AssertionError(m.getName());
|
}));
|
ReflectionTestUtils.setField(target,"repairMapper",stub(MesDvRepairMapper.class,(p,m,a)->jdbc.query("select * from repair where id=? for update",new BeanPropertyRowMapper<>(MesDvRepairDO.class),a[0]).stream().findFirst().orElse(null)));
|
ReflectionTestUtils.setField(target,"machineryMapper",stub(MesDvMachineryMapper.class,(p,m,a)->new MesDvMachineryDO().setId(1L).setName("设备A")));
|
ReflectionTestUtils.setField(target,"adminUserApi",stub(AdminUserApi.class,(p,m,a)->Objects.equals(a[0],1L)?new AdminUserRespDTO().setId(1L).setNickname("维修员").setStatus(0):null));
|
var proxy=new ProxyFactory(target);
|
proxy.addAdvice(new TransactionInterceptor(new DataSourceTransactionManager(ds),new AnnotationTransactionAttributeSource()));
|
service=(MesDvSpareService)proxy.getProxy();
|
}
|
MesDvSparePartDO onePart(Long id,boolean lock){return jdbc.query("select * from part where id=?"+(lock?" for update":""),new BeanPropertyRowMapper<>(MesDvSparePartDO.class),id).stream().findFirst().orElse(null);}
|
MesDvSparePartSaveReqVO save(){return new MesDvSparePartSaveReqVO().setCode("S-002").setName("皮带").setUnit("件").setInitialQuantity(BigDecimal.ZERO).setSafetyQuantity(BigDecimal.ONE).setStatus(0);}
|
MesDvSpareActionReqVO action(int qty){return new MesDvSpareActionReqVO().setRequestKey(UUID.randomUUID().toString()).setQuantity(BigDecimal.valueOf(qty)).setRemark("维修用途");}
|
MesDvSpareActionReqVO issue(int qty){return action(qty).setSparePartId(1L).setRepairId(1L).setReceiverUserId(1L);}
|
BigDecimal stock(Long id){return jdbc.queryForObject("select stock_quantity from part where id=?",BigDecimal.class,id);}
|
int count(){return jdbc.queryForObject("select count(*) from movement",Integer.class);}
|
}
|
}
|
