From 4ebac835de11d5dd8fc2d4e676a0414f367a674d Mon Sep 17 00:00:00 2001 From: hhs <386998068@qq.com> Date: Sun, 14 Jun 2026 14:04:30 +0800 Subject: [PATCH] =?UTF-8?q?chore:=20=E5=88=A0=E9=99=A4=20deepgram=5Ftest.g?= =?UTF-8?q?o=20=E6=B5=8B=E8=AF=95=E6=96=87=E4=BB=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- backend/internal/ai/stt/deepgram_test.go | 191 ----------------------- 1 file changed, 191 deletions(-) delete mode 100644 backend/internal/ai/stt/deepgram_test.go diff --git a/backend/internal/ai/stt/deepgram_test.go b/backend/internal/ai/stt/deepgram_test.go deleted file mode 100644 index 70cc52e..0000000 --- a/backend/internal/ai/stt/deepgram_test.go +++ /dev/null @@ -1,191 +0,0 @@ -package stt - -import ( - "context" - "encoding/json" - "net/http" - "net/http/httptest" - "strings" - "testing" - "time" - - "github.com/gorilla/websocket" - "go.uber.org/zap" -) - -var upgrader = websocket.Upgrader{ - CheckOrigin: func(r *http.Request) bool { return true }, -} - -// newMockDeepgram 创建模拟 Deepgram WebSocket 服务。 -// 返回 httptest.Server 和对应的 ws:// URL。 -func newMockDeepgram(t *testing.T, handler func(conn *websocket.Conn)) *httptest.Server { - t.Helper() - srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { - conn, err := upgrader.Upgrade(w, r, nil) - if err != nil { - t.Logf("upgrade error: %v", err) - return - } - defer conn.Close() - handler(conn) - })) - return srv -} - -// wsToWss 将 http:// 转换为 ws://。 -func wsToWss(httpURL string) string { - return "ws" + strings.TrimPrefix(httpURL, "http") -} - -func TestDeepgramService_Recognize_Success(t *testing.T) { - srv := newMockDeepgram(t, func(conn *websocket.Conn) { - // 读取音频数据 - _, _, err := conn.ReadMessage() - if err != nil { - t.Errorf("read audio: %v", err) - return - } - - // 发送中间结果（非 final） - intermediate := deepgramResponse{ - IsFinal: false, - } - intermediate.Channel.Alternatives = []struct { - Transcript string `json:"transcript"` - Confidence float64 `json:"confidence"` - }{{Transcript: "你好", Confidence: 0.9}} - data, _ := json.Marshal(intermediate) - _ = conn.WriteMessage(websocket.TextMessage, data) - - // 发送最终结果 - final := deepgramResponse{ - IsFinal: true, - } - final.Channel.Alternatives = []struct { - Transcript string `json:"transcript"` - Confidence float64 `json:"confidence"` - }{{Transcript: "你好世界", Confidence: 0.95}} - data, _ = json.Marshal(final) - _ = conn.WriteMessage(websocket.TextMessage, data) - - // 等待客户端关闭 - _, _, _ = conn.ReadMessage() - }) - defer srv.Close() - - svc := NewDeepgramService("test-key", "", wsToWss(srv.URL)+"/v1/listen", 0, zap.NewNop().Sugar()) - - text, err := svc.Recognize(context.Background(), []byte("fake-pcm-audio"), Options{ - Encoding: "pcm_s16le", - SampleRate: 16000, - Language: "zh-CN", - }) - if err != nil { - t.Fatalf("Recognize() error: %v", err) - } - if text != "你好世界" { - t.Errorf("Recognize() = %q, want %q", text, "你好世界") - } -} - -func TestDeepgramService_Recognize_EmptyAudio(t *testing.T) { - svc := NewDeepgramService("test-key", "", "ws://localhost", 0, zap.NewNop().Sugar()) - _, err := svc.Recognize(context.Background(), nil, Options{}) - if err == nil { - t.Fatal("Recognize() with empty audio should return error") - } -} - -func TestDeepgramService_Recognize_ConnectError(t *testing.T) { - svc := NewDeepgramService("test-key", "", "ws://localhost:1", 0, zap.NewNop().Sugar()) - ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second) - defer cancel() - - _, err := svc.Recognize(ctx, []byte("audio"), Options{}) - if err == nil { - t.Fatal("Recognize() with bad endpoint should return error") - } -} - -func TestDeepgramService_Recognize_Timeout(t *testing.T) { - // 模拟一个永不响应的服务端 - srv := newMockDeepgram(t, func(conn *websocket.Conn) { - // 读取音频但不发送任何结果，让客户端超时 - _, _, _ = conn.ReadMessage() - time.Sleep(10 * time.Second) - }) - defer srv.Close() - - svc := NewDeepgramService("test-key", "", wsToWss(srv.URL)+"/v1/listen", 0, zap.NewNop().Sugar()) - - ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second) - defer cancel() - - _, err := svc.Recognize(ctx, []byte("audio"), Options{}) - if err == nil { - t.Fatal("Recognize() should timeout") - } -} - -func TestDeepgramService_Recognize_MultipleFinals(t *testing.T) { - srv := newMockDeepgram(t, func(conn *websocket.Conn) { - _, _, _ = conn.ReadMessage() - - // 发送多个 final 结果（多句话场景） - for _, text := range []string{"你好", "世界"} { - resp := deepgramResponse{IsFinal: true} - resp.Channel.Alternatives = []struct { - Transcript string `json:"transcript"` - Confidence float64 `json:"confidence"` - }{{Transcript: text, Confidence: 0.9}} - data, _ := json.Marshal(resp) - _ = conn.WriteMessage(websocket.TextMessage, data) - } - - _, _, _ = conn.ReadMessage() - }) - defer srv.Close() - - svc := NewDeepgramService("test-key", "", wsToWss(srv.URL)+"/v1/listen", 0, zap.NewNop().Sugar()) - - text, err := svc.Recognize(context.Background(), []byte("audio"), Options{}) - if err != nil { - t.Fatalf("Recognize() error: %v", err) - } - if text != "你好世界" { - t.Errorf("Recognize() = %q, want %q", text, "你好世界") - } -} - -func TestDeepgramService_buildURL(t *testing.T) { - svc := NewDeepgramService("key", "", "wss://api.deepgram.com/v1/listen", 0, zap.NewNop().Sugar()) - - tests := []struct { - name string - opts Options - want []string // URL 中应包含的参数 - }{ - { - name: "defaults", - opts: Options{}, - want: []string{"encoding=pcm_s16le", "sample_rate=16000", "language=zh-CN"}, - }, - { - name: "custom", - opts: Options{Encoding: "wav", SampleRate: 44100, Language: "en"}, - want: []string{"encoding=wav", "sample_rate=44100", "language=en"}, - }, - } - - for _, tt := range tests { - t.Run(tt.name, func(t *testing.T) { - u := svc.buildURL(tt.opts) - for _, param := range tt.want { - if !strings.Contains(u, param) { - t.Errorf("buildURL() = %q, should contain %q", u, param) - } - } - }) - } -}