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

backend/internal/service/grok_search_count_test.go

@@ -0,0 +1,85 @@
+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
+}