Files
Cyrene_v2/internal/runtime/voice/ttslaunch/probe_test.go
T
AskaEth 6906d274fc feat(voice): M47 后端启动后探活 emotionTTS,未起则异步拉起一键启动脚本
主人 2026-10-24:「tts启动脚本在这 ...\一键启动.bat,以后启动后端可以顺便检测
一下如果TTS没起就顺带启一下」。本片只做「启动时顺带探一下」,**默认关**(主人
早前说过「开机自启先不急 后面做」),未开开关时连探活都不发。

- 新包 internal/runtime/voice/ttslaunch:探 http://127.0.0.1:9880(毫秒级超时),
  已在跑什么都不做;没起则异步起脚本(cmd.exe /c 传参,不拼命令行字符串)
- 防重复:in-flight 单飞 + 一次尝试后的冷却(失败也算一次,重启风暴不会变成重试风暴)
- 失败只 WARN:探活超时/脚本不存在/起进程失败都只记带 source 的日志,
  组合根那条路径**没有 error 通道**
- 配置走面板文件 $DATA_DIR/config/voice-tts-launch.json(零新 .env 键)
- 接线在 cmd/cyrene 的网关 Listen 之后,整段在后台 goroutine 里,不占启动时长
  (用例以「挂载耗时 < 150ms 而探活超时 800ms」证明)
- 不新开端点、不加帧族;不碰 gateway 包
2026-10-26 18:47:36 +08:00

72 lines
2.2 KiB
Go

package ttslaunch
import (
"context"
"net"
"net/http"
"net/http/httptest"
"strings"
"testing"
"time"
)
// TestHTTPProbe_AnyResponseMeansRunning 钉住探活判据:**任何** HTTP 回应都算在跑
// (这个服务的根路径回什么状态码不稳定,只要有人在应答就不该再拉第二个实例)。
func TestHTTPProbe_AnyResponseMeansRunning(t *testing.T) {
for _, code := range []int{http.StatusOK, http.StatusNotFound, http.StatusMethodNotAllowed} {
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
w.WriteHeader(code)
}))
if err := httpProbe(srv.URL, time.Second)(context.Background()); err != nil {
t.Fatalf("状态码 %d 被判成没起: %v", code, err)
}
srv.Close()
}
}
// TestHTTPProbe_Unreachable 钉住"连不上 ⇒ 没起":先占一个端口再放掉,探它必然被拒。
func TestHTTPProbe_Unreachable(t *testing.T) {
addr := closedAddr(t)
if err := httpProbe("http://"+addr, 500*time.Millisecond)(context.Background()); err == nil {
t.Fatal("没人监听的端口被判成在跑")
}
}
// TestHTTPProbe_TimeoutIsHonored 钉住"短超时"确实生效:服务故意不回话,
// 100ms 的探活必须按超时失败,而不是等到服务自己醒。
func TestHTTPProbe_TimeoutIsHonored(t *testing.T) {
block := make(chan struct{})
srv := httptest.NewServer(http.HandlerFunc(func(http.ResponseWriter, *http.Request) { <-block }))
defer func() {
close(block)
srv.Close()
}()
start := time.Now()
err := httpProbe(srv.URL, 100*time.Millisecond)(context.Background())
elapsed := time.Since(start)
if err == nil {
t.Fatal("不回话的服务被判成在跑")
}
if elapsed > time.Second {
t.Fatalf("探活等了 %v, 明显没按 100ms 超时", elapsed)
}
if !strings.Contains(err.Error(), "probe") {
t.Fatalf("错误没标来源: %v", err)
}
}
// closedAddr 返回一个"刚被放掉"的回环地址(探它 = 连接被拒)。
func closedAddr(t *testing.T) string {
t.Helper()
ln, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
t.Fatalf("listen: %v", err)
}
addr := ln.Addr().String()
if err := ln.Close(); err != nil {
t.Fatalf("close: %v", err)
}
return addr
}