后台更新 服务优化

backend-java/src/test/java/com/nanri/aiimage/modules/dedupe/service/DedupeTotalDataServiceTest.java

@@ -43,6 +43,7 @@ import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.Mockito.never;
+import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
@@ -172,7 +173,8 @@ class DedupeTotalDataServiceTest {
     void duplicateExcelValueKeepsOriginalUploader() throws Exception {
         when(adminUserMapper.selectById(23L)).thenReturn(user(23L, "normal", "member-a"));
         stubWritableGroup(23L, 7L);
-        when(dedupeTotalDataMapper.selectOne(any())).thenReturn(data(91L, 99L));
+        when(dedupeTotalDataMapper.selectExistingDataValues(List.of("B012345678")))
+                .thenReturn(List.of("B012345678"));
         MockMultipartFile file = asinWorkbook("B012345678");
 
         var result = service.importFromExcel(file, 7L, 23L);
@@ -187,7 +189,10 @@ class DedupeTotalDataServiceTest {
         when(adminUserMapper.selectById(23L)).thenReturn(user(23L, "normal", "member-a"));
         stubWritableGroup(23L, 7L);
         when(dedupeTotalDataMapper.selectOne(any())).thenReturn(null);
+        when(dedupeTotalDataMapper.selectExistingDataValues(List.of("B012345678"))).thenReturn(List.of());
         when(transactionManager.getTransaction(any())).thenReturn(transactionStatus);
+        when(dedupeTotalDataMapper.insertBatchIgnore(any()))
+                .thenThrow(new DuplicateKeyException("duplicate batch"));
         when(dedupeTotalDataMapper.insert(any(DedupeTotalDataEntity.class)))
                 .thenThrow(new DuplicateKeyException("duplicate"));
 
@@ -195,7 +200,7 @@ class DedupeTotalDataServiceTest {
 
         assertEquals(0, result.getInsertedCount());
         assertEquals(1, result.getSkippedCount());
-        verify(transactionManager).rollback(transactionStatus);
+        verify(transactionManager, times(2)).rollback(transactionStatus);
     }
 
     @Test