李建琦
6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
90 lines
3.1 KiB
Go
90 lines
3.1 KiB
Go
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")
|
|
}
|