Sub2API v1.0 - AI API 网关(二开初始版本,基于上游 Wei-Shaw/sub2api)
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
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
This commit is contained in:
@@ -0,0 +1,41 @@
|
||||
package repository
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/Wei-Shaw/sub2api/internal/service"
|
||||
"github.com/alicebob/miniredis/v2"
|
||||
"github.com/redis/go-redis/v9"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestGatewayCacheReasoningContent(t *testing.T) {
|
||||
mr := miniredis.RunT(t)
|
||||
client := redis.NewClient(&redis.Options{Addr: mr.Addr()})
|
||||
cache := NewGatewayCache(client)
|
||||
ctx := context.Background()
|
||||
|
||||
// 未命中返回哨兵错误,区别于真实读取失败。
|
||||
_, err := cache.GetReasoningContent(ctx, "item_missing")
|
||||
require.ErrorIs(t, err, service.ErrReasoningContentNotFound)
|
||||
|
||||
// 写入后可读回。
|
||||
require.NoError(t, cache.SetReasoningContent(ctx, "item_abc", "think hard", time.Minute))
|
||||
got, err := cache.GetReasoningContent(ctx, "item_abc")
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, "think hard", got)
|
||||
|
||||
// ttl<=0 时兜底为默认 7 天。
|
||||
require.NoError(t, cache.SetReasoningContent(ctx, "item_ttl", "x", 0))
|
||||
ttl := mr.TTL(reasoningContentPrefix + "item_ttl")
|
||||
require.Greater(t, ttl, 6*24*time.Hour)
|
||||
require.LessOrEqual(t, ttl, reasoningContentDefaultTTL)
|
||||
|
||||
// 空 itemID / 空 content 是 no-op(无可缓存内容不算错误)。
|
||||
require.NoError(t, cache.SetReasoningContent(ctx, "", "x", time.Minute))
|
||||
require.NoError(t, cache.SetReasoningContent(ctx, "item_empty", "", time.Minute))
|
||||
_, err = cache.GetReasoningContent(ctx, "item_empty")
|
||||
require.ErrorIs(t, err, service.ErrReasoningContentNotFound)
|
||||
}
|
||||
Reference in New Issue
Block a user