package cn.iocoder.yudao.module.mes.service.pro.scan; import cn.iocoder.yudao.framework.common.exception.ServiceException; import cn.iocoder.yudao.framework.security.core.LoginUser; import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils; import cn.iocoder.yudao.module.mes.controller.admin.pro.scan.vo.MesProScanEventCreateReqVO; import cn.iocoder.yudao.module.mes.dal.dataobject.pro.scan.MesProScanEventDO; import cn.iocoder.yudao.module.mes.dal.mysql.pro.scan.MesProScanEventMapper; import org.junit.jupiter.api.*; import org.springframework.security.core.context.SecurityContextHolder; import org.springframework.test.util.ReflectionTestUtils; import java.math.BigDecimal; import static org.junit.jupiter.api.Assertions.*; import static org.mockito.Mockito.*; class MobileScanRetryTest { private final MesProScanEventMapper mapper=mock(MesProScanEventMapper.class); private final MesProScanEventServiceImpl service=new MesProScanEventServiceImpl(); private MesProScanEventCreateReqVO req; private MesProScanEventDO existing; @BeforeEach void setup(){ SecurityFrameworkUtils.setLoginUser(new LoginUser().setId(7L),new org.springframework.mock.web.MockHttpServletRequest()); ReflectionTestUtils.setField(service,"eventMapper",mapper); req=new MesProScanEventCreateReqVO();req.setEventId("original");req.setBagCode("123");req.setQuantity(new BigDecimal("1.00"));req.setLocationId(2L);req.setAreaId(3L); existing=new MesProScanEventDO().setEventId("original").setBagCode("123").setScanType("BAG").setQuantity(BigDecimal.ONE).setLocationId(2L).setAreaId(3L).setStatus("SUCCESS"); existing.setCreator("7"); when(mapper.selectByEventIdForUpdate("original")).thenReturn(existing); } @AfterEach void cleanup(){SecurityContextHolder.clearContext();} @Test void sameOperationReturnsOriginalWithoutAnotherInventoryWrite(){ assertSame(existing,service.createInbound(req));verifyNoMoreInteractionsAfterLookup(); } @Test void conflictNeverOverwritesOriginalSuccess(){ req.setAreaId(4L); assertThrows(ServiceException.class,()->service.createInbound(req)); assertEquals("SUCCESS",existing.getStatus());assertNull(existing.getErrorCode()); verifyNoMoreInteractionsAfterLookup(); } @Test void anotherUserCannotReplayOriginal(){ existing.setCreator("8"); assertThrows(ServiceException.class,()->service.createInbound(req));verifyNoMoreInteractionsAfterLookup(); } @Test void changedBagQuantityCannotReuseOriginalOperation(){ req.setQuantity(BigDecimal.TEN); assertThrows(ServiceException.class,()->service.createInbound(req));verifyNoMoreInteractionsAfterLookup(); } @Test void palletReplayKeepsServerCalculatedQuantity(){ req.setScanType("PALLET");existing.setScanType("PALLET").setQuantity(BigDecimal.TEN); assertSame(existing,service.createInbound(req));assertEquals(BigDecimal.TEN,existing.getQuantity()); verifyNoMoreInteractionsAfterLookup(); } private void verifyNoMoreInteractionsAfterLookup(){verify(mapper).selectByEventIdForUpdate("original");verifyNoMoreInteractions(mapper);} }