李建琦
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
30 lines
870 B
Go
30 lines
870 B
Go
package openai
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMatchClientEntry_ReturnsHitEntry(t *testing.T) {
|
|
entries := []AllowedClientEntry{
|
|
{Originator: "opencode", UAContains: []string{"opencode/"}, SkipEngineFingerprint: true},
|
|
{Originator: "Claude Code", UAContains: []string{"Claude Code/"}},
|
|
}
|
|
|
|
e, ok := MatchClientEntry("opencode/1.0", "opencode", entries)
|
|
require.True(t, ok)
|
|
require.True(t, e.SkipEngineFingerprint)
|
|
|
|
e2, ok2 := MatchClientEntry("Claude Code/1.0 (x) (Claude Code; 1)", "Claude Code", entries)
|
|
require.True(t, ok2)
|
|
require.False(t, e2.SkipEngineFingerprint)
|
|
|
|
_, ok3 := MatchClientEntry("curl/8", "evil", entries)
|
|
require.False(t, ok3)
|
|
|
|
// 薄封装保持兼容
|
|
require.True(t, MatchClientEntries("opencode/1.0", "opencode", entries))
|
|
require.False(t, MatchClientEntries("curl/8", "evil", entries))
|
|
}
|