李建琦
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
60 lines
2.2 KiB
Go
60 lines
2.2 KiB
Go
//go:build unit
|
|
|
|
package service
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/pkg/ctxkey"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestResolveUsageBillingRequestID_ForcedWebSearchBeatsClientID(t *testing.T) {
|
|
t.Parallel()
|
|
ctx := context.WithValue(context.Background(), ctxkey.ClientRequestID, "client-shared-id")
|
|
got := resolveUsageBillingRequestID(ctx, "web_search:uuid-1")
|
|
require.Equal(t, "web_search:uuid-1", got)
|
|
}
|
|
|
|
func TestResolveUsageBillingRequestID_ClientWinsOverPlainUpstream(t *testing.T) {
|
|
t.Parallel()
|
|
ctx := context.WithValue(context.Background(), ctxkey.ClientRequestID, "client-shared-id")
|
|
got := resolveUsageBillingRequestID(ctx, "resp_abc")
|
|
require.Equal(t, "client:client-shared-id", got)
|
|
}
|
|
|
|
func TestIsForcedUsageBillingRequestID(t *testing.T) {
|
|
t.Parallel()
|
|
require.True(t, isForcedUsageBillingRequestID("web_search:x"))
|
|
require.True(t, isForcedUsageBillingRequestID("grok-video:task-1"))
|
|
require.True(t, isForcedUsageBillingRequestID("grok_audio:up-1"))
|
|
require.True(t, isForcedUsageBillingRequestID("grok_realtime:sess-1"))
|
|
require.False(t, isForcedUsageBillingRequestID("resp_abc"))
|
|
}
|
|
|
|
func TestStableGrokAudioBillingRequestID(t *testing.T) {
|
|
t.Parallel()
|
|
require.Equal(t, "grok_audio:up-1", StableGrokAudioBillingRequestID("up-1"))
|
|
require.Equal(t, "grok_audio:up-1", StableGrokAudioBillingRequestID("grok_audio:up-1"))
|
|
got := StableGrokAudioBillingRequestID("")
|
|
require.True(t, strings.HasPrefix(got, "grok_audio:"))
|
|
require.Greater(t, len(got), len("grok_audio:"))
|
|
}
|
|
|
|
func TestStableGrokRealtimeBillingRequestID(t *testing.T) {
|
|
t.Parallel()
|
|
require.Equal(t, "grok_realtime:s1", StableGrokRealtimeBillingRequestID("s1"))
|
|
require.Equal(t, "grok_realtime:s1", StableGrokRealtimeBillingRequestID("grok_realtime:s1"))
|
|
got := StableGrokRealtimeBillingRequestID("")
|
|
require.True(t, strings.HasPrefix(got, "grok_realtime:"))
|
|
}
|
|
|
|
func TestResolveUsageBillingRequestID_ForcedGrokAudioBeatsClientID(t *testing.T) {
|
|
t.Parallel()
|
|
ctx := context.WithValue(context.Background(), ctxkey.ClientRequestID, "client-shared-id")
|
|
got := resolveUsageBillingRequestID(ctx, StableGrokAudioBillingRequestID("up-9"))
|
|
require.Equal(t, "grok_audio:up-9", got)
|
|
}
|