Files
sub2api/backend/internal/service/grok_stream_idle_test.go
T
李建琦 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
Sub2API v1.0 - AI API 网关(二开初始版本,基于上游 Wei-Shaw/sub2api)
2026-08-21 18:30:13 +08:00

26 lines
761 B
Go

//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")
}