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

backend/internal/service/grok_stream_idle_test.go

@@ -0,0 +1,25 @@
+//go:build unit
+
+package service
+
+import (
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestResolveGrokStreamIdleTimeout(t *testing.T) {
+	require.Equal(t, 90*time.Second, resolveGrokStreamIdleTimeout(90))
+	require.Equal(t, defaultGrokStreamIdleTimeout, resolveGrokStreamIdleTimeout(0))
+	require.Equal(t, defaultGrokStreamIdleTimeout, resolveGrokStreamIdleTimeout(-1))
+}
+
+func TestGrokStreamIdleFailoverError(t *testing.T) {
+	account := &Account{ID: 1, Platform: PlatformGrok, Type: AccountTypeOAuth}
+	err := grokStreamIdleFailoverError(account, 180*time.Second)
+	require.NotNil(t, err)
+	require.Equal(t, 502, err.StatusCode)
+	require.True(t, err.SafeToFailoverAfterWrite)
+	require.Contains(t, string(err.ResponseBody), "empty_upstream")
+}