send-stream

yudao-module-tik/src/test/java/cn/iocoder/yudao/module/tik/voice/service/LatentsyncServiceImplTest.java

@@ -0,0 +1,62 @@
+package cn.iocoder.yudao.module.tik.voice.service;
+
+import cn.iocoder.yudao.module.tik.voice.client.LatentsyncClient;
+import cn.iocoder.yudao.module.tik.voice.client.dto.LatentsyncSubmitRequest;
+import cn.iocoder.yudao.module.tik.voice.client.dto.LatentsyncSubmitResponse;
+import cn.iocoder.yudao.module.tik.voice.vo.AppTikLatentsyncSubmitReqVO;
+import cn.iocoder.yudao.module.tik.voice.vo.AppTikLatentsyncSubmitRespVO;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.ArgumentCaptor;
+import org.mockito.Mock;
+import org.mockito.junit.jupiter.MockitoExtension;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+@ExtendWith(MockitoExtension.class)
+class LatentsyncServiceImplTest {
+
+    @Mock
+    private LatentsyncClient latentsyncClient;
+
+    private LatentsyncServiceImpl latentsyncService;
+
+    @BeforeEach
+    void setUp() {
+        latentsyncService = new LatentsyncServiceImpl(latentsyncClient);
+    }
+
+    @Test
+    void submitTask_success() {
+        AppTikLatentsyncSubmitReqVO reqVO = new AppTikLatentsyncSubmitReqVO();
+        reqVO.setAudioUrl("https://cdn.example.com/audio.wav");
+        reqVO.setVideoUrl("https://cdn.example.com/video.mp4");
+        reqVO.setGuidanceScale(2);
+        reqVO.setSeed(999);
+
+        LatentsyncSubmitResponse clientResp = new LatentsyncSubmitResponse();
+        clientResp.setRequestId("task-123");
+        clientResp.setStatus("IN_QUEUE");
+        clientResp.setQueuePosition(0);
+        when(latentsyncClient.submitTask(org.mockito.Mockito.any())).thenReturn(clientResp);
+
+        AppTikLatentsyncSubmitRespVO respVO = latentsyncService.submitTask(reqVO);
+
+        assertThat(respVO.getRequestId()).isEqualTo("task-123");
+        assertThat(respVO.getStatus()).isEqualTo("IN_QUEUE");
+        assertThat(respVO.getQueuePosition()).isZero();
+
+        ArgumentCaptor<LatentsyncSubmitRequest> captor = ArgumentCaptor.forClass(LatentsyncSubmitRequest.class);
+        verify(latentsyncClient).submitTask(captor.capture());
+        LatentsyncSubmitRequest submitRequest = captor.getValue();
+        assertThat(submitRequest.getAudioUrl()).isEqualTo(reqVO.getAudioUrl());
+        assertThat(submitRequest.getVideoUrl()).isEqualTo(reqVO.getVideoUrl());
+        assertThat(submitRequest.getGuidanceScale()).isEqualTo(reqVO.getGuidanceScale());
+        assertThat(submitRequest.getSeed()).isEqualTo(reqVO.getSeed());
+    }
+}
+
+