MailUI4Agents/gateway/internal/models/address_test.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)
	}
}