31 lines
951 B
Go
31 lines
951 B
Go
package service
|
|
|
|
import (
|
|
"testing"
|
|
|
|
relaycommon "github.com/QuantumNous/new-api/relay/common"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestResolveAttemptGuardResultForDetectionOnly(t *testing.T) {
|
|
t.Run("matching declaration is verified", func(t *testing.T) {
|
|
info := &relaycommon.RelayInfo{ResponseModel: &relaycommon.ResponseModel{
|
|
RequestedModel: "gpt-6-astra",
|
|
ReturnedModel: "gpt-6-astra",
|
|
}}
|
|
require.Equal(t, "accepted", resolveAttemptGuardResult("", info))
|
|
})
|
|
|
|
t.Run("missing declaration remains unchecked", func(t *testing.T) {
|
|
require.Equal(t, "unchecked", resolveAttemptGuardResult("", &relaycommon.RelayInfo{}))
|
|
})
|
|
|
|
t.Run("allowed mismatch remains visible", func(t *testing.T) {
|
|
info := &relaycommon.RelayInfo{ResponseModel: &relaycommon.ResponseModel{
|
|
RequestedModel: "gpt-6-astra",
|
|
ReturnedModel: "gpt-5.6-luna",
|
|
}}
|
|
require.Equal(t, "mismatch_allowed", resolveAttemptGuardResult("", info))
|
|
})
|
|
}
|