You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
975 B
32 lines
975 B
package comm
|
|
|
|
import "testing"
|
|
|
|
func TestAiVendorBlockedOnlyWhenLocked(t *testing.T) {
|
|
t.Setenv("AI_ONLY_IFLYTEK", "")
|
|
if AiVendorBlocked("stt_azure", "azure", "1") {
|
|
t.Fatal("锁关着时不应拦截微软听写")
|
|
}
|
|
t.Setenv("AI_ONLY_IFLYTEK", "1")
|
|
if !AiOnlyIflytek() {
|
|
t.Fatal("AI_ONLY_IFLYTEK=1 应开启")
|
|
}
|
|
if !AiVendorBlocked("mt_alibaba", "alibaba", "3") {
|
|
t.Fatal("锁开着应拦截阿里翻译")
|
|
}
|
|
if !AiVendorBlocked("llm_mobile_elf", "chinamobile", "5") {
|
|
t.Fatal("锁开着应拦截灵犀")
|
|
}
|
|
if !AiVendorBlocked("sts_iflytek_aiui", "iflytek", "7") {
|
|
t.Fatal("AIUI 不是这次的 SparkChain，应拦截")
|
|
}
|
|
if AiVendorBlocked("sparkchain", "iflytek", "12") {
|
|
t.Fatal("应用参数里的 SparkChain 不是 AI 类别，应继续下发")
|
|
}
|
|
if AiVendorBlocked("mt_iflytek", "iflytek", "3") {
|
|
t.Fatal("讯飞机器翻译应放行")
|
|
}
|
|
if AiVendorBlocked("oss_aliyun", "alibaba", "6") {
|
|
t.Fatal("对象存储不在这把锁里")
|
|
}
|
|
}
|
|
|