diff --git a/viewsh-module-ops/viewsh-module-environment-biz/src/test/java/com/viewsh/module/ops/environment/service/cleanorder/CleanOrderEndToEndTest.java b/viewsh-module-ops/viewsh-module-environment-biz/src/test/java/com/viewsh/module/ops/environment/service/cleanorder/CleanOrderEndToEndTest.java index de9bda5f..061c60e4 100644 --- a/viewsh-module-ops/viewsh-module-environment-biz/src/test/java/com/viewsh/module/ops/environment/service/cleanorder/CleanOrderEndToEndTest.java +++ b/viewsh-module-ops/viewsh-module-environment-biz/src/test/java/com/viewsh/module/ops/environment/service/cleanorder/CleanOrderEndToEndTest.java @@ -20,6 +20,7 @@ import com.viewsh.module.ops.environment.dal.mysql.workorder.OpsOrderCleanExtMap import com.viewsh.module.ops.environment.integration.consumer.*; import com.viewsh.framework.common.pojo.CommonResult; import com.viewsh.module.ops.environment.integration.listener.CleanOrderEventListener; +import com.viewsh.module.ops.environment.service.notification.CleanOrderNotificationService; import com.viewsh.module.ops.environment.service.cleanorder.CleanOrderServiceImpl; import com.viewsh.module.ops.environment.service.voice.VoiceBroadcastService; import com.viewsh.module.ops.environment.service.badge.BadgeDeviceStatusService; @@ -107,6 +108,9 @@ public class CleanOrderEndToEndTest { @Mock private EventLogRecorder eventLogRecorder; + + @Mock + private CleanOrderNotificationService cleanOrderNotificationService; @Spy private ObjectMapper objectMapper = new ObjectMapper(); @@ -156,6 +160,7 @@ public class CleanOrderEndToEndTest { // 注入 CleanOrderEventListener injectField(cleanOrderService, "cleanOrderEventListener", cleanOrderEventListener); + injectField(cleanOrderService, "cleanOrderNotificationService", cleanOrderNotificationService); injectField(cleanOrderService, "opsBusAreaMapper", opsBusAreaMapper); // 注入 CleanOrderAuditEventHandler 依赖 @@ -406,7 +411,7 @@ public class CleanOrderEndToEndTest { verify(orderQueueService).adjustPriority(eq(500L), eq(PriorityEnum.P0), anyString()); verify(orderQueueService).rebuildWaitingTasksByUserId(2001L, order.getAreaId()); - verify(cleanOrderEventListener).sendPriorityUpgradeNotification(eq(2001L), eq("WO-P2"), eq(orderId)); + verify(cleanOrderNotificationService).sendPriorityUpgradeNotification(eq(2001L), eq("WO-P2"), eq(orderId)); } @Test