Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/internal/service/grok_audio_test.go

@@ -0,0 +1,89 @@
+package service
+
+import (
+	"context"
+	"io"
+	"sync/atomic"
+	"testing"
+
+	"github.com/Wei-Shaw/sub2api/internal/pkg/xai"
+	"github.com/stretchr/testify/require"
+)
+
+func TestBuildGrokVoiceURL_UsesAPIDefaultForCLIProxyBase(t *testing.T) {
+	account := &Account{
+		Platform: PlatformGrok,
+		Type:     AccountTypeOAuth,
+		Credentials: map[string]any{
+			"base_url": xai.DefaultCLIBaseURL,
+		},
+	}
+	url, err := buildGrokVoiceURL(account, nil, "tts")
+	require.NoError(t, err)
+	require.Equal(t, xai.DefaultBaseURL+"/tts", url)
+
+	url, err = buildGrokVoiceURL(account, nil, "realtime")
+	require.NoError(t, err)
+	require.Equal(t, xai.DefaultBaseURL+"/realtime", url)
+}
+
+func TestBuildGrokVoiceURL_EmptyBaseFallsBackToAPI(t *testing.T) {
+	account := &Account{
+		Platform:    PlatformGrok,
+		Type:        AccountTypeOAuth,
+		Credentials: map[string]any{},
+	}
+	url, err := buildGrokVoiceURL(account, nil, "stt")
+	require.NoError(t, err)
+	require.Equal(t, xai.DefaultBaseURL+"/stt", url)
+}
+
+func TestBuildGrokVoiceURL_RequiresEndpoint(t *testing.T) {
+	account := &Account{Platform: PlatformGrok, Type: AccountTypeOAuth}
+	_, err := buildGrokVoiceURL(account, nil, "  ")
+	require.Error(t, err)
+}
+
+func TestBuildGrokVoiceURL_EncodesCustomVoicePathSegments(t *testing.T) {
+	account := &Account{Platform: PlatformGrok, Type: AccountTypeOAuth}
+	got, err := buildGrokVoiceURL(account, nil, "custom-voices/nlbqfwie/audio")
+	require.NoError(t, err)
+	require.Equal(t, xai.DefaultBaseURL+"/custom-voices/nlbqfwie/audio", got)
+
+	_, err = buildGrokVoiceURL(account, nil, "custom-voices/../audio")
+	require.Error(t, err)
+}
+
+func TestForwardGrokVoice_RejectsNonGrok(t *testing.T) {
+	svc := &OpenAIGatewayService{}
+	_, err := svc.ForwardGrokVoice(context.Background(), nil, &Account{Platform: PlatformOpenAI}, "tts", []byte(`{}`), "application/json")
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "not supported")
+}
+
+func TestAwaitGrokRealtimeAudioObservedReadsFlagAfterRelayExits(t *testing.T) {
+	errCh := make(chan error, 1)
+	var observed atomic.Bool
+	go func() {
+		observed.Store(true)
+		errCh <- io.EOF
+	}()
+	got, err := awaitGrokRealtimeAudioObserved(errCh, &observed)
+	require.ErrorIs(t, err, io.EOF)
+	require.True(t, got, "audioObserved must be read after the relay returns, not before <-errCh")
+}
+
+func TestGrokRealtimeEventHasAudio(t *testing.T) {
+	require.False(t, grokRealtimeEventHasAudio([]byte(`{"type":"session.created"}`)))
+	require.False(t, grokRealtimeEventHasAudio([]byte(`{"type":"response.audio_transcript.delta","delta":"hi"}`)))
+	require.False(t, grokRealtimeEventHasAudio([]byte(`{"type":"response.audio.delta","delta":""}`)))
+	require.True(t, grokRealtimeEventHasAudio([]byte(`{"type":"response.audio.delta","delta":"abc"}`)))
+	require.True(t, grokRealtimeEventHasAudio([]byte(`{"type":"response.output_audio.delta","audio":"abc"}`)))
+}
+
+func TestForwardGrokVoice_RejectsUnknownEndpoint(t *testing.T) {
+	svc := &OpenAIGatewayService{}
+	_, err := svc.ForwardGrokVoice(context.Background(), nil, &Account{Platform: PlatformGrok}, "unknown", []byte(`{}`), "application/json")
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "unsupported")
+}