李建琦
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
27 lines
1.1 KiB
Go
27 lines
1.1 KiB
Go
package openai
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMatchClientEntries_WhitelistAND(t *testing.T) {
|
|
wl := []AllowedClientEntry{{Originator: "opencode", UAContains: []string{"opencode/"}}}
|
|
require.True(t, MatchClientEntries("opencode/1.2 (x)", "opencode", wl))
|
|
require.False(t, MatchClientEntries("opencode/1.2 (x)", "other", wl), "originator 不符不放")
|
|
require.False(t, MatchClientEntries("curl/8", "opencode", wl), "UA marker 缺失不放")
|
|
require.False(t, MatchClientEntries("opencode/1.2", "opencode", []AllowedClientEntry{{Originator: "opencode"}}), "空 UAContains 安全失败")
|
|
}
|
|
|
|
func TestDenyEntries_BlacklistOR(t *testing.T) {
|
|
bl := []AllowedClientEntry{
|
|
{Originator: "evilbot"},
|
|
{UAContains: []string{"badscan/"}},
|
|
}
|
|
require.True(t, MatchDenyEntries("anything/1", "evilbot", bl), "originator 命中即拒")
|
|
require.True(t, MatchDenyEntries("badscan/9 (x)", "whatever", bl), "UA marker 命中即拒")
|
|
require.False(t, MatchDenyEntries("codex_cli_rs/0.141.0", "codex_cli_rs", bl), "都不命中不拒")
|
|
require.False(t, MatchDenyEntries("x", "y", []AllowedClientEntry{{}}), "全空条目安全忽略")
|
|
}
|