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("对象存储不在这把锁里")
	}
}