package cn.iocoder.yudao.module.mes.service.qc.seedquality;
|
|
import cn.iocoder.yudao.module.mes.controller.admin.qc.seedquality.vo.*;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.bagcode.MesProBagCodeDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.batch.MesProBatchDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.pallet.MesProPalletDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.qc.seedquality.MesQcSeedQualityDO;
|
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 cn.iocoder.yudao.module.mes.dal.mysql.qc.seedquality.MesQcSeedQualityMapper;
|
import org.junit.jupiter.api.*;
|
import org.springframework.test.util.ReflectionTestUtils;
|
import java.util.List;
|
import static org.junit.jupiter.api.Assertions.*;
|
import static org.mockito.Mockito.*;
|
|
class SeedQualityGroupTest {
|
private final MesQcSeedQualityServiceImpl service=spy(new MesQcSeedQualityServiceImpl());
|
private final MesProBagCodeMapper bags=mock(MesProBagCodeMapper.class);
|
private final MesProBatchMapper batches=mock(MesProBatchMapper.class);
|
private final MesProPalletMapper pallets=mock(MesProPalletMapper.class);
|
private final MesQcSeedQualityMapper qualities=mock(MesQcSeedQualityMapper.class);
|
private final MesProBagCodeDO first=bag(1L),second=bag(2L),canceled=bag(3L).setStatus(30);
|
private MesQcSeedQualityGroupSubmitReqVO req;
|
private MesProBagCodeDO bag(Long id) {return new MesProBagCodeDO().setId(id).setCode("B"+id).setBatchId(10L).setPalletId(20L).setStatus(20).setInboundEventId(9L);}
|
@BeforeEach void setup() {
|
ReflectionTestUtils.setField(service,"bagCodeMapper",bags);ReflectionTestUtils.setField(service,"batchMapper",batches);
|
ReflectionTestUtils.setField(service,"palletMapper",pallets);ReflectionTestUtils.setField(service,"qualityMapper",qualities);
|
var batch=new MesProBatchDO().setId(10L).setCode("BATCH").setBusinessCode("BATCH-BUS").setStatus(30);
|
when(batches.selectByScanCode(anyString())).thenReturn(batch);when(batches.selectByScanCodeForUpdate(anyString())).thenReturn(batch);
|
var pallet=new MesProPalletDO().setId(20L).setCode("PALLET").setBusinessCode("PALLET-BUS").setBatchCode("BATCH").setBatchId(10L).setStatus(10);
|
when(pallets.selectByScanCode(anyString())).thenReturn(pallet);when(pallets.selectByScanCodeForUpdate(anyString())).thenReturn(pallet);
|
when(bags.selectListByBatchId(10L)).thenReturn(List.of(first,second,canceled));
|
when(bags.selectListByBatchIdForUpdate(10L)).thenReturn(List.of(first,second,canceled));
|
when(bags.selectListByPalletId(20L)).thenReturn(List.of(first,second,canceled));
|
when(bags.selectListByPalletIdForUpdate(20L)).thenReturn(List.of(first,second,canceled));
|
when(bags.selectListByIdsForUpdate(any())).thenReturn(List.of(first,second));
|
req=new MesQcSeedQualityGroupSubmitReqVO();req.setCode("BATCH-BUS");req.setScanType("BATCH");req.setBagIds(List.of(1L,2L));
|
req.setQualifiedWarehouseId(8L);req.setIndicators(List.of());
|
doReturn(101L).when(service).submit(any(MesQcSeedQualitySubmitReqVO.class));
|
}
|
@Test void batchPreviewCountsEffectiveInspectedAndPendingBags() {
|
when(qualities.selectListByCodes(any())).thenReturn(List.of(MesQcSeedQualityDO.builder().code("B1").build()));
|
var result=service.resolveScan("BATCH-BUS","BATCH");assertEquals(2,result.get("totalCount"));
|
assertEquals(List.of(1L),result.get("completedBagIds"));assertEquals(List.of(2L),result.get("pendingBagIds"));
|
verify(service,never()).submit(any());
|
}
|
@Test void palletProcessesOnlyConfirmedUninspectedBags() {
|
req.setScanType("PALLET");req.setCode("PALLET-BUS");req.setBagIds(List.of(2L));
|
when(qualities.selectListByCodesForUpdate(any())).thenReturn(List.of(MesQcSeedQualityDO.builder().code("B1").build()));
|
assertEquals(1,service.submitGroup(req).get("processedCount"));
|
var capture=org.mockito.ArgumentCaptor.forClass(MesQcSeedQualitySubmitReqVO.class);verify(service).submit(capture.capture());
|
assertEquals("B2",capture.getValue().getCode());assertEquals(8L,capture.getValue().getQualifiedWarehouseId());
|
}
|
@Test void changedConfirmedListDoesNotSubmitAnyBag() {
|
req.setBagIds(List.of(1L));assertThrows(IllegalArgumentException.class,()->service.submitGroup(req));verify(service,never()).submit(any());
|
}
|
@Test void unenteredBagBlocksWholeGroup() {
|
second.setInboundEventId(null);assertThrows(IllegalArgumentException.class,()->service.submitGroup(req));verify(service,never()).submit(any());
|
}
|
@Test void wrongTypeAndCanceledBatchCannotSubmit() {
|
req.setScanType("OTHER");assertThrows(IllegalArgumentException.class,()->service.submitGroup(req));
|
req.setScanType("BATCH");when(batches.selectByScanCodeForUpdate(anyString())).thenReturn(new MesProBatchDO().setId(10L).setCode("BATCH").setStatus(40));
|
assertThrows(IllegalArgumentException.class,()->service.submitGroup(req));verify(service,never()).submit(any());
|
}
|
@Test void singleBagUsesSameConfirmedGroupProtocol() {
|
req.setScanType("BAG");req.setCode("B1");req.setBagIds(List.of(1L));
|
when(bags.selectByScanCode("B1")).thenReturn(first);when(bags.selectListByIdsForUpdate(any())).thenReturn(List.of(first));
|
assertEquals(1,service.submitGroup(req).get("processedCount"));verify(service).submit(any());
|
}
|
@Test void failingSecondBagRollsBackWholeTransaction() {
|
var source=new org.h2.jdbcx.JdbcDataSource();source.setURL("jdbc:h2:mem:qcgroup"+java.util.UUID.randomUUID()+";DB_CLOSE_DELAY=-1");
|
var jdbc=new org.springframework.jdbc.core.JdbcTemplate(source);jdbc.execute("CREATE TABLE qc_result(id BIGINT)");
|
doAnswer(invocation->{var single=(MesQcSeedQualitySubmitReqVO)invocation.getArgument(0);jdbc.update("INSERT INTO qc_result(id) VALUES(?)",single.getCode().equals("B1")?1:2);if(single.getCode().equals("B2"))throw new IllegalStateException("库存流转失败");return 101L;}).when(service).submit(any());
|
var factory=new org.springframework.aop.framework.ProxyFactory(service);
|
factory.addAdvice(new org.springframework.transaction.interceptor.TransactionInterceptor(new org.springframework.jdbc.datasource.DataSourceTransactionManager(source),new org.springframework.transaction.annotation.AnnotationTransactionAttributeSource()));
|
var proxy=(MesQcSeedQualityService)factory.getProxy();
|
assertThrows(IllegalStateException.class,()->proxy.submitGroup(req));assertEquals(0,jdbc.queryForObject("SELECT COUNT(*) FROM qc_result",Integer.class));
|
}
|
}
|
