You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.2 KiB
91 lines
2.2 KiB
package filetrans_test
|
|
|
|
import (
|
|
"yunyan/sys/aliyun/filetrans"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
const testApiKey = "sk-ec05b3b04275433ea4f4ea8e5299a8cf"
|
|
|
|
func TestCreateTask(t *testing.T) {
|
|
sys, err := filetrans.NewSys(
|
|
filetrans.SetApiKey(testApiKey),
|
|
)
|
|
if err != nil {
|
|
t.Fatalf("初始化失败: %v", err)
|
|
}
|
|
|
|
taskID, err := sys.CreateTask(
|
|
"https://deepsound-1253517901.cos.ap-guangzhou.myqcloud.com/User/1942916924698525696/LocalAudio/2026/02/27/jLEIQzYIAnqy.wav",
|
|
"vi-VN",
|
|
true,
|
|
"",
|
|
)
|
|
if err != nil {
|
|
t.Fatalf("创建任务失败: %v", err)
|
|
}
|
|
fmt.Printf("创建任务成功: taskID=%s\n", taskID)
|
|
}
|
|
|
|
func TestQueryTask(t *testing.T) {
|
|
sys, err := filetrans.NewSys(
|
|
filetrans.SetApiKey(testApiKey),
|
|
)
|
|
if err != nil {
|
|
t.Fatalf("初始化失败: %v", err)
|
|
}
|
|
|
|
// 替换为 TestCreateTask 返回的实际 taskID
|
|
taskID := "1c09487e-9e92-4c4d-ba62-a442bacfc309"
|
|
|
|
status, contexts, err := sys.QueryTask(taskID)
|
|
if err != nil {
|
|
t.Fatalf("查询失败: %v", err)
|
|
}
|
|
fmt.Printf("查询结果: status=%s\n", status)
|
|
for i, ctx := range contexts {
|
|
fmt.Printf(" [%d] %dms-%dms Speaker(%s): %s\n", i, ctx.StartTime, ctx.EndTime, ctx.Speaker, ctx.Content)
|
|
}
|
|
}
|
|
|
|
func TestCreateAndPollTask(t *testing.T) {
|
|
sys, err := filetrans.NewSys(
|
|
filetrans.SetApiKey(testApiKey),
|
|
)
|
|
if err != nil {
|
|
t.Fatalf("初始化失败: %v", err)
|
|
}
|
|
|
|
taskID, err := sys.CreateTask(
|
|
"https://deepsound-1253517901.cos.ap-guangzhou.myqcloud.com/User/2014183956768882688/LocalAudio/2026/01/22/kUqCZgiVkspH.wav",
|
|
"zh",
|
|
true,
|
|
"",
|
|
)
|
|
if err != nil {
|
|
t.Fatalf("创建任务失败: %v", err)
|
|
}
|
|
fmt.Printf("任务已提交: taskID=%s\n", taskID)
|
|
|
|
for i := 0; i < 60; i++ {
|
|
time.Sleep(5 * time.Second)
|
|
status, contexts, err := sys.QueryTask(taskID)
|
|
fmt.Printf(" 轮询 #%d: status=%s\n", i+1, status)
|
|
if err != nil {
|
|
t.Fatalf("查询失败: %v", err)
|
|
}
|
|
if status == filetrans.StatusSuccess {
|
|
fmt.Printf("转写完成，共 %d 条结果:\n", len(contexts))
|
|
for j, ctx := range contexts {
|
|
fmt.Printf(" [%d] %dms-%dms Speaker(%s): %s\n", j, ctx.StartTime, ctx.EndTime, ctx.Speaker, ctx.Content)
|
|
}
|
|
return
|
|
}
|
|
if status == filetrans.StatusFailed {
|
|
t.Fatalf("任务失败")
|
|
}
|
|
}
|
|
t.Fatal("超时：任务未在 5 分钟内完成")
|
|
}
|
|
|