95 lines
3.2 KiB
Go
95 lines
3.2 KiB
Go
package models
|
||
|
||
import "testing"
|
||
|
||
func TestParseAddress(t *testing.T) {
|
||
cases := []struct {
|
||
in string
|
||
name string
|
||
path string
|
||
session string
|
||
mode SessionMode
|
||
}{
|
||
// 用户给出的两个例子
|
||
{"deepseekharness@/program.upadtefeature", "deepseekharness", "/program", "upadtefeature", SessionNamed},
|
||
{"pi@root.new", "pi", "root", "new", SessionNew},
|
||
|
||
// 常规形态
|
||
{"builder@ModelRouter.fix-memory-leak", "builder", "ModelRouter", "fix-memory-leak", SessionNamed},
|
||
{"@builder@ModelRouter.new", "builder", "ModelRouter", "new", SessionNew},
|
||
{"human@.new", "human", "", "new", SessionNew},
|
||
|
||
// 省略 session 位 = 默认会话（不等于 new）
|
||
{"human", "human", "", "", SessionDefault},
|
||
{"ops@prod", "ops", "prod", "", SessionDefault},
|
||
|
||
// path 内含 . 与 /（按最后一个 . 切）
|
||
{"agent@/home/a.b/c.deploy", "agent", "/home/a.b/c", "deploy", SessionNamed},
|
||
}
|
||
|
||
for _, c := range cases {
|
||
got, err := ParseAddress(c.in)
|
||
if err != nil {
|
||
t.Fatalf("ParseAddress(%q) unexpected error: %v", c.in, err)
|
||
}
|
||
if got.Name != c.name || got.Path != c.path || got.Session != c.session {
|
||
t.Errorf("ParseAddress(%q) = {name:%q path:%q session:%q}, want {name:%q path:%q session:%q}",
|
||
c.in, got.Name, got.Path, got.Session, c.name, c.path, c.session)
|
||
}
|
||
if got.Mode() != c.mode {
|
||
t.Errorf("ParseAddress(%q).Mode() = %v, want %v", c.in, got.Mode(), c.mode)
|
||
}
|
||
}
|
||
}
|
||
|
||
// session 位省略与 new 必须是两种不同语义：
|
||
// 省略 → 默认会话；new → 强制新建；其他 → 必须已存在。
|
||
func TestSessionModeSemantics(t *testing.T) {
|
||
def, _ := ParseAddress("pi@root")
|
||
if def.Mode() != SessionDefault || def.IsNewSession() || !def.IsDefaultSession() {
|
||
t.Errorf("pi@root 应为默认会话，得到 mode=%v isNew=%v", def.Mode(), def.IsNewSession())
|
||
}
|
||
|
||
new_, _ := ParseAddress("pi@root.new")
|
||
if new_.Mode() != SessionNew || !new_.IsNewSession() || new_.IsDefaultSession() {
|
||
t.Errorf("pi@root.new 应为新建，得到 mode=%v", new_.Mode())
|
||
}
|
||
|
||
named, _ := ParseAddress("pi@root.fix-leak")
|
||
if named.Mode() != SessionNamed || named.IsNewSession() || named.IsDefaultSession() {
|
||
t.Errorf("pi@root.fix-leak 应为具名会话，得到 mode=%v", named.Mode())
|
||
}
|
||
}
|
||
|
||
func TestParseAddressErrors(t *testing.T) {
|
||
for _, in := range []string{"", " ", "@", "@ModelRouter.new"} {
|
||
if _, err := ParseAddress(in); err == nil {
|
||
t.Errorf("ParseAddress(%q) expected error, got nil", in)
|
||
}
|
||
}
|
||
}
|
||
|
||
func TestParseAddressList(t *testing.T) {
|
||
list, err := ParseAddressList("pi@root.new, deepseekharness@/program.upadtefeature;ops@prod")
|
||
if err != nil {
|
||
t.Fatalf("unexpected error: %v", err)
|
||
}
|
||
if len(list) != 3 {
|
||
t.Fatalf("got %d addresses, want 3", len(list))
|
||
}
|
||
if list[0].Name != "pi" || !list[0].IsNewSession() {
|
||
t.Errorf("addr[0] = %+v", list[0])
|
||
}
|
||
if list[1].Path != "/program" || list[1].Session != "upadtefeature" {
|
||
t.Errorf("addr[1] = %+v", list[1])
|
||
}
|
||
if list[2].Name != "ops" || list[2].Path != "prod" || list[2].Mode() != SessionDefault {
|
||
t.Errorf("addr[2] = %+v", list[2])
|
||
}
|
||
|
||
empty, err := ParseAddressList(" ")
|
||
if err != nil || empty != nil {
|
||
t.Errorf("empty list = %v, %v; want nil, nil", empty, err)
|
||
}
|
||
}
|