new-api-custom/service/relay_attempt_test.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",
			Mismatch:       true,
		}}
		require.Equal(t, "mismatch_allowed", resolveAttemptGuardResult("", info))
	})
}