后端优化，增加新需求

2026-07-21 19:53:47 +08:00
parent ba48bbc6af
commit bca1335dc6
16 changed files with 529 additions and 61 deletions
@@ -0,0 +1,82 @@
+package com.nanri.aiimage.modules.task.service;
+
+import com.nanri.aiimage.config.InstanceMetadata;
+import com.nanri.aiimage.modules.appearancepatent.service.AppearancePatentTaskService;
+import com.nanri.aiimage.modules.brand.service.BrandTaskService;
+import com.nanri.aiimage.modules.collectdata.service.CollectDataService;
+import com.nanri.aiimage.modules.deletebrand.service.DeleteBrandRunService;
+import com.nanri.aiimage.modules.patroldelete.service.PatrolDeleteTaskService;
+import com.nanri.aiimage.modules.pricetrack.service.PriceTrackTaskService;
+import com.nanri.aiimage.modules.productrisk.service.ProductRiskTaskService;
+import com.nanri.aiimage.modules.queryasin.service.QueryAsinTaskService;
+import com.nanri.aiimage.modules.shopmatch.service.ShopMatchTaskService;
+import com.nanri.aiimage.modules.similarasin.service.SimilarAsinTaskService;
+import com.nanri.aiimage.modules.task.mapper.FileResultMapper;
+import com.nanri.aiimage.modules.task.model.entity.FileResultEntity;
+import com.nanri.aiimage.modules.task.model.entity.TaskFileJobEntity;
+import com.nanri.aiimage.modules.withdraw.service.WithdrawTaskService;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.InOrder;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.junit.jupiter.MockitoExtension;
+
+import static org.mockito.Mockito.inOrder;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+@ExtendWith(MockitoExtension.class)
+class TaskResultFileJobWorkerTest {
+
+    @Mock private TaskFileJobService taskFileJobService;
+    @Mock private TaskDistributedLockService taskDistributedLockService;
+    @Mock private TaskResultPayloadService taskResultPayloadService;
+    @Mock private FileResultMapper fileResultMapper;
+    @Mock private TaskFileJobLocalDispatcher taskFileJobLocalDispatcher;
+    @Mock private InstanceMetadata instanceMetadata;
+    @Mock private ShopMatchTaskService shopMatchTaskService;
+    @Mock private PriceTrackTaskService priceTrackTaskService;
+    @Mock private ProductRiskTaskService productRiskTaskService;
+    @Mock private QueryAsinTaskService queryAsinTaskService;
+    @Mock private WithdrawTaskService withdrawTaskService;
+    @Mock private PatrolDeleteTaskService patrolDeleteTaskService;
+    @Mock private AppearancePatentTaskService appearancePatentTaskService;
+    @Mock private SimilarAsinTaskService similarAsinTaskService;
+    @Mock private DeleteBrandRunService deleteBrandRunService;
+    @Mock private BrandTaskService brandTaskService;
+    @Mock private CollectDataService collectDataService;
+
+    @InjectMocks private TaskResultFileJobWorker worker;
+
+    @Test
+    void withdrawFileSuccessFinalizesTaskAfterReleasingLock() {
+        long jobId = 13640L;
+        long taskId = 20140L;
+        long resultId = 22928L;
+        TaskFileJobEntity job = new TaskFileJobEntity();
+        job.setId(jobId);
+        job.setTaskId(taskId);
+        job.setResultId(resultId);
+        job.setModuleType("WITHDRAW");
+        job.setScopeKey("withdraw:20140");
+
+        FileResultEntity result = new FileResultEntity();
+        result.setResultFileUrl("result/withdraw/20140.xlsx");
+        TaskDistributedLockService.LockHandle lock = mock(TaskDistributedLockService.LockHandle.class);
+
+        when(taskFileJobService.markRunning(jobId)).thenReturn(true);
+        when(taskDistributedLockService.acquire("WITHDRAW", taskId, TaskDistributedLockService.DEFAULT_WAIT_MILLIS))
+                .thenReturn(lock);
+        when(fileResultMapper.selectById(resultId)).thenReturn(result);
+
+        worker.process(job);
+
+        verify(taskFileJobService).markSuccess(job, "result/withdraw/20140.xlsx");
+        verify(taskResultPayloadService).deleteLatest(taskId, "WITHDRAW", "withdraw:20140");
+        InOrder order = inOrder(lock, withdrawTaskService);
+        order.verify(lock).close();
+        order.verify(withdrawTaskService).tryFinalizeTask(taskId, false);
+    }
+}