【优化】mj 测试增加 WssNotify

cherishsince
1 parent 59a289f1
Showing 1 changed file with 7 additions and 1 deletions
yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/midjourney/MidjourneyWebSocketTests.java
 package cn.iocoder.yudao.framework.ai.midjourney;
 import cn.hutool.core.io.FileUtil;
+import cn.iocoder.yudao.framework.ai.midjourney.webSocket.WssNotify;
 import cn.iocoder.yudao.framework.ai.midjourney.webSocket.listener.MidjourneyMessageListener;
 import cn.iocoder.yudao.framework.ai.midjourney.webSocket.MidjourneyWebSocketStarter;
 import org.junit.Before;
@@ -38,7 +39,12 @@ public class MidjourneyWebSocketTests {
         var webSocketStarter = new MidjourneyWebSocketStarter(wssUrl, null, midjourneyConfig, messageListener);
         try {
-            webSocketStarter.start();
+            webSocketStarter.start(new WssNotify() {
+                @Override
+                public void notify(int code, String message) {
+
+                }
+            });
         } catch (Exception e) {
             throw new RuntimeException(e);
         }