ModelRouter/internal/provider/registry_pinned_name_test.go

package provider

import "testing"

// ★ `源名:模型名` 这种限定写法，源名本身**含 `-`** 时必须仍能解开。
//
// 真实缺陷（用户报「无法使用 channel 注册的模型」，2026-10-17）：
//
//	WebUI 的模型下拉框用 `源名:模型名` 作为 option value —— 这是**有意**的：
//	本实例有 44 个模型被多个源同时提供（glm-5.2 被 6 个源提供），
//	用户必须能指定「从哪个源调用」。
//
//	但 ResolvePinned 用 `IndexAny(model, "-:/")` 找**第一个**分隔符，
//	而渠道源名形如 `channels-trae-b8dd6b3df30e`（含 `-`）⇒ 被切成
//
//	  src="channels"、model="trae-b8dd6b3df30e:Doubao-Seed-2.1-Pro"
//
//	⇒ 源名永不匹配 ⇒ Resolve() 返回 nil
//	⇒ 客户端看到 `model "channels-trae-xxx:..." is not configured`。
func TestResolvePinnedWithHyphenatedSourceName(t *testing.T) {
	cases := []struct {
		name    string
		sources map[string][]string // 源名 → 模型
		input   string
		wantSrc string
		wantMdl string
		wantOK  bool
	}{
		{
			name: "渠道源名含两个 -（真实形态）",
			sources: map[string][]string{
				"channels-trae-b8dd6b3df30e": {"Doubao-Seed-2.1-Pro", "glm-5.2"},
				"channels-buddy-buddy-main":  {"hy4-preview", "glm-5.2"},
			},
			input:   "channels-trae-b8dd6b3df30e:Doubao-Seed-2.1-Pro",
			wantSrc: "channels-trae-b8dd6b3df30e",
			wantMdl: "Doubao-Seed-2.1-Pro",
			wantOK:  true,
		},
		{
			name: "同名模型跨源：必须能选中第二个源",
			sources: map[string][]string{
				"channels-trae-b8dd6b3df30e": {"glm-5.2"},
				"channels-buddy-buddy-main":  {"glm-5.2"},
			},
			input:   "channels-buddy-buddy-main:glm-5.2",
			wantSrc: "channels-buddy-buddy-main",
			wantMdl: "glm-5.2",
			wantOK:  true,
		},
		{
			name: "斜杠写法仍要支持",
			sources: map[string][]string{
				"channels-trae-b8dd6b3df30e": {"glm-5.2"},
			},
			input:   "channels-trae-b8dd6b3df30e/glm-5.2",
			wantSrc: "channels-trae-b8dd6b3df30e",
			wantMdl: "glm-5.2",
			wantOK:  true,
		},
		{
			// 自带 ":" 的第三方模型名，且**没有**任何源名与之匹配 ⇒ 走旧兜底。
			// ⚠️ 不能拿 "our-free-model_cohere/..." 当例子：本实例真有一个叫
			//    `our-free-model_cohere` 的源，而仓库里还有叫 `our` 的源 ——
			//    最长前缀匹配会正确命中 `our`，那个例子的预期是我写错了。
			name: "无源名匹配时走旧兜底",
			sources: map[string][]string{
				"channels-trae-b8dd6b3df30e": {"glm-5.2"},
			},
			input:   "vendor-lib/thing:free",
			wantSrc: "vendor",
			wantMdl: "lib/thing:free",
			wantOK:  true,
		},
	}
	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			var ps []*Provider
			for name, models := range c.sources {
				ps = append(ps, newTestProvider(t, src(name, "http://127.0.0.1:1", "openai", models...)))
			}
			src, m, ok := splitPinned(c.input, ps)
			if ok != c.wantOK {
				t.Fatalf("ok 应为 %v，实得 %v（输入 %q）", c.wantOK, ok, c.input)
			}
			if !ok {
				return
			}
			if src != c.wantSrc {
				t.Fatalf("★ 源名解析错：期望 %q，实得 %q\n"+
					"  输入 %q —— 源名含 `-` 时不能按第一个分隔符切", c.wantSrc, src, c.input)
			}
			if m != c.wantMdl {
				t.Fatalf("模型名解析错：期望 %q，实得 %q", c.wantMdl, m)
			}
		})
	}
}