feat: 算力引擎 Go 后端核心(new-api)
- 统一 OpenAI 兼容 /v1 中继 + 渠道/额度/令牌/流水 - 各领域包:relay 模型网关、model 数据层、controller 管理 API
This commit is contained in:
@@ -0,0 +1,48 @@
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"github.com/QuantumNous/new-api/common"
|
||||
"github.com/QuantumNous/new-api/constant"
|
||||
"github.com/QuantumNous/new-api/model"
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func newTokenAutoGroupsContext() *gin.Context {
|
||||
gin.SetMode(gin.TestMode)
|
||||
ctx, _ := gin.CreateTestContext(httptest.NewRecorder())
|
||||
return ctx
|
||||
}
|
||||
|
||||
func TestSetupContextForTokenPreservesCustomAutoGroupsOrder(t *testing.T) {
|
||||
ctx := newTokenAutoGroupsContext()
|
||||
token := &model.Token{Id: 1, UserId: 2, AutoGroups: `["vip","default"]`}
|
||||
|
||||
require.NoError(t, SetupContextForToken(ctx, token))
|
||||
value, ok := common.GetContextKey(ctx, constant.ContextKeyTokenAutoGroups)
|
||||
require.True(t, ok)
|
||||
assert.Equal(t, []string{"vip", "default"}, value)
|
||||
}
|
||||
|
||||
func TestSetupContextForTokenTreatsStoredEmptyArrayAsInheritance(t *testing.T) {
|
||||
ctx := newTokenAutoGroupsContext()
|
||||
token := &model.Token{Id: 1, UserId: 2, AutoGroups: `[]`}
|
||||
|
||||
require.NoError(t, SetupContextForToken(ctx, token))
|
||||
_, ok := common.GetContextKey(ctx, constant.ContextKeyTokenAutoGroups)
|
||||
assert.False(t, ok)
|
||||
}
|
||||
|
||||
func TestSetupContextForTokenMalformedAutoGroupsFailsClosed(t *testing.T) {
|
||||
ctx := newTokenAutoGroupsContext()
|
||||
token := &model.Token{Id: 1, UserId: 2, AutoGroups: `not-json`}
|
||||
|
||||
require.NoError(t, SetupContextForToken(ctx, token))
|
||||
value, ok := common.GetContextKey(ctx, constant.ContextKeyTokenAutoGroups)
|
||||
require.True(t, ok)
|
||||
assert.Equal(t, []string{}, value)
|
||||
}
|
||||
Reference in New Issue
Block a user