mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-11-06 08:49:24 +00:00
IanShaw027
165b072908
Align gateway Grok media/voice paths and model lists, harden upstream failure and quota handling, clear non-Grok video generation config migration, and polish temp-unsched/status indicators with model whitelist updates.
86 lines
4.1 KiB
Go
86 lines
4.1 KiB
Go
package service
|
||
|
||
import (
|
||
"testing"
|
||
|
||
"github.com/stretchr/testify/require"
|
||
)
|
||
|
||
func TestCountGrokNativeSearchCallsFromJSONBytes(t *testing.T) {
|
||
t.Parallel()
|
||
require.Equal(t, 0, countGrokNativeSearchCallsFromJSONBytes(nil))
|
||
require.Equal(t, 0, countGrokNativeSearchCallsFromJSONBytes([]byte(`{"output":[]}`)))
|
||
body := []byte(`{"output":[
|
||
{"type":"web_search_call","id":"ws1","status":"completed"},
|
||
{"type":"x_search_call","id":"xs1"},
|
||
{"type":"function_call","name":"tool_search","call_id":"ts1"},
|
||
{"type":"function_call","name":"lookup","call_id":"other"}
|
||
]}`)
|
||
require.Equal(t, 3, countGrokNativeSearchCallsFromJSONBytes(body))
|
||
}
|
||
|
||
func TestCountGrokNativeSearchCallsFromJSONBytes_PrefersNestedResponse(t *testing.T) {
|
||
t.Parallel()
|
||
body := []byte(`{"output":[{"type":"web_search_call","id":"duplicate"}],"response":{"output":[{"type":"web_search_call","id":"duplicate"},{"type":"x_search_call","id":"xs1"}]}}`)
|
||
require.Equal(t, 2, countGrokNativeSearchCallsFromJSONBytes(body))
|
||
}
|
||
|
||
func TestCountGrokNativeSearchCallsFromJSONBytes_FallsBackWhenNestedOutputNull(t *testing.T) {
|
||
t.Parallel()
|
||
body := []byte(`{"output":[{"type":"web_search_call","id":"ws1"}],"response":{"output":null}}`)
|
||
require.Equal(t, 1, countGrokNativeSearchCallsFromJSONBytes(body))
|
||
}
|
||
|
||
func TestCountGrokNativeSearchCallsFromSSEBodyDedups(t *testing.T) {
|
||
t.Parallel()
|
||
sse := stringsJoin(
|
||
`data: {"type":"response.output_item.done","item":{"type":"web_search_call","id":"ws1","call_id":"c1"}}`,
|
||
`data: {"type":"response.output_item.done","item":{"type":"web_search_call","id":"ws1","call_id":"c1"}}`,
|
||
`data: {"type":"response.completed","response":{"output":[{"type":"web_search_call","id":"ws1","call_id":"c1"},{"type":"x_search_call","id":"xs1","call_id":"c2"}]}}`,
|
||
)
|
||
require.Equal(t, 2, countGrokNativeSearchCallsFromSSEBody(sse))
|
||
}
|
||
|
||
func TestCountGrokNativeSearchCallsInSSEDataDedup_LiveStreamPath(t *testing.T) {
|
||
t.Parallel()
|
||
// Mirrors the live streaming accumulator: item.done then response.completed
|
||
// for the same call_id must bill once (regression for ~2× surcharge).
|
||
seen := make(map[string]struct{})
|
||
done := []byte(`{"type":"response.output_item.done","item":{"type":"web_search_call","id":"ws1","call_id":"c1"}}`)
|
||
completed := []byte(`{"type":"response.completed","response":{"output":[{"type":"web_search_call","id":"ws1","call_id":"c1"},{"type":"x_search_call","id":"xs1","call_id":"c2"}]}}`)
|
||
require.Equal(t, 1, countGrokNativeSearchCallsInSSEDataDedup(done, seen))
|
||
require.Equal(t, 1, countGrokNativeSearchCallsInSSEDataDedup(completed, seen))
|
||
// Raw (no-dedup) path still double-counts the same envelope pair.
|
||
require.Equal(t, 1, countGrokNativeSearchCallsInSSEData(done))
|
||
require.Equal(t, 2, countGrokNativeSearchCallsInSSEData(completed))
|
||
}
|
||
|
||
func TestCountGrokNativeSearchCallsInSSEDataDedup_NoIDStillDedups(t *testing.T) {
|
||
t.Parallel()
|
||
// Upstream sometimes omits call_id/id; synthetic keys must still prevent 2×.
|
||
seen := make(map[string]struct{})
|
||
done := []byte(`{"type":"response.output_item.done","item":{"type":"web_search_call"}}`)
|
||
completed := []byte(`{"type":"response.completed","response":{"output":[{"type":"web_search_call"}]}}`)
|
||
require.Equal(t, 1, countGrokNativeSearchCallsInSSEDataDedup(done, seen))
|
||
require.Equal(t, 0, countGrokNativeSearchCallsInSSEDataDedup(completed, seen))
|
||
}
|
||
|
||
func TestCountGrokNativeSearchCallsInSSEDataDedup_MultipleNoIDCalls(t *testing.T) {
|
||
t.Parallel()
|
||
seen := make(map[string]struct{})
|
||
firstDone := []byte(`{"type":"response.output_item.done","item":{"type":"web_search_call"}}`)
|
||
secondDone := []byte(`{"type":"response.output_item.done","item":{"type":"web_search_call"}}`)
|
||
completed := []byte(`{"type":"response.completed","response":{"output":[{"type":"web_search_call"},{"type":"web_search_call"}]}}`)
|
||
require.Equal(t, 1, countGrokNativeSearchCallsInSSEDataDedup(firstDone, seen))
|
||
require.Equal(t, 1, countGrokNativeSearchCallsInSSEDataDedup(secondDone, seen))
|
||
require.Equal(t, 0, countGrokNativeSearchCallsInSSEDataDedup(completed, seen))
|
||
}
|
||
|
||
func stringsJoin(lines ...string) string {
|
||
out := ""
|
||
for _, l := range lines {
|
||
out += l + "\n\n"
|
||
}
|
||
return out
|
||
}
|