MailUI4Agents/plugins/homeagent-mail-bridge/attach_test.go

package main

// attach.go 的测试。
//
// 事故背景（本轮实测，生产环境）：homeagent 的 upload_attachment 按
// **顶层** `attachment_id` 解析服务端响应，而服务端返回的是
// `{"attachment":{"attachment_id":…}}` —— 三个字段全解成零值。
//
// 那是最坏的一种失败形状：上传其实成功了（HTTP 200、文件已落盘、库里已登记），
// 没有任何一层报错，但模型看到的是 `id= filename= size=0KB`。拿着空 id 它
// 发不出这个附件，24 小时后 GC 把那个没人引用的文件清掉，现场不留痕迹。
//
// 同时 send_mail 根本没声明 attachment_ids 参数，且服务端解请求体时没开
// DisallowUnknownFields —— 实测传 `attachments:[{...}]` 返回 200，
// 而那封邮件的附件数是 0。两个 bug 叠起来，homeagent 的附件发送链路
// **从来没成功过一次**。

import (
	"strings"
	"testing"
)

// ─── stringList ───

func TestStringList_JSONShape(t *testing.T) {
	// 运行时的真实形状：SDK 把工具参数当 JSON 解出来，
	// 数组是 []interface{}，元素是 interface{}(string)
	got := stringList([]interface{}{"a", "b"})
	if len(got) != 2 || got[0] != "a" || got[1] != "b" {
		t.Fatalf("want [a b], got %#v", got)
	}
}

func TestStringList_NotTypedSlice(t *testing.T) {
	// 这条是整个文件存在的理由：写 args[...].([]string) 会拿到 nil。
	// 若哪天有人把 stringList 换回类型断言，这个用例会失败。
	var v interface{} = []interface{}{"id-1"}
	if _, ok := v.([]string); ok {
		t.Fatal("[]interface{} 不该能断言成 []string —— 前提变了，附件字段会静默消失")
	}
	if got := stringList(v); len(got) != 1 || got[0] != "id-1" {
		t.Fatalf("stringList 必须能吃下 []interface{}，got %#v", got)
	}
}

func TestStringList_SingleString(t *testing.T) {
	// 模型常见的偷懒写法。意图无歧义，拒绝它只换来一次重试。
	got := stringList("only-one")
	if len(got) != 1 || got[0] != "only-one" {
		t.Fatalf("want [only-one], got %#v", got)
	}
}

func TestStringList_DropsBadElementsKeepsGood(t *testing.T) {
	// 逐项校验而非整批放弃：否则「三个附件里有一个写错」会变成
	// 「一个附件都没发出」。
	got := stringList([]interface{}{"good-1", nil, 42, "", "  ", "good-2", map[string]interface{}{}})
	if len(got) != 2 || got[0] != "good-1" || got[1] != "good-2" {
		t.Fatalf("want [good-1 good-2], got %#v", got)
	}
}

func TestStringList_TrimsSpace(t *testing.T) {
	got := stringList([]interface{}{"  padded-id  "})
	if len(got) != 1 || got[0] != "padded-id" {
		t.Fatalf("want [padded-id], got %#v", got)
	}
}

func TestStringList_EmptyAndNil(t *testing.T) {
	// nil / 空数组 / 全是坏元素都要返回空，让调用方省略 payload 字段而不是传 []
	for name, in := range map[string]interface{}{
		"nil":    nil,
		"空数组":    []interface{}{},
		"空字符串":   "",
		"只有空白":   "   ",
		"全是坏元素":  []interface{}{nil, 1, false},
		"不认识的类型": map[string]interface{}{"a": 1},
		"数字":     42,
		"全是空串元素": []interface{}{"", "  "},
	} {
		if got := stringList(in); len(got) != 0 {
			t.Fatalf("%s 应返回空，got %#v", name, got)
		}
	}
}

func TestStringList_TypedSliceAlsoWorks(t *testing.T) {
	// 单测里手写参数会走到这一支
	got := stringList([]string{"x", "", "y"})
	if len(got) != 2 || got[0] != "x" || got[1] != "y" {
		t.Fatalf("want [x y], got %#v", got)
	}
}

// ─── formatSize ───

func TestFormatSize(t *testing.T) {
	cases := []struct {
		in   int64
		want string
	}{
		// 关键用例：小文件不能显示成 0KB。原先写的是 size/1024 + "KB"，
		// 一个 800 字节的附件显示成 `0KB`，模型会据此认为上传失败。
		{0, "0 B"},
		{1, "1 B"},
		{800, "800 B"},
		{1023, "1023 B"},
		{1024, "1.0 KB"},
		{1536, "1.5 KB"},
		{28, "28 B"}, // 本轮探针的真实大小
		{1024 * 1024, "1.0 MB"},
		{25 << 20, "25.0 MB"}, // 附件上限
		{-1, "0 B"},           // 不该出现，但不能打印成 -1 B
	}
	for _, c := range cases {
		if got := formatSize(c.in); got != c.want {
			t.Errorf("formatSize(%d) = %q, want %q", c.in, got, c.want)
		}
	}
}

func TestFormatSizeNeverZeroKBForNonEmpty(t *testing.T) {
	// 负向对照的固化：任何非空文件都不该显示成「0」开头的大小
	for _, n := range []int64{1, 10, 100, 800, 1023} {
		got := formatSize(n)
		if strings.HasPrefix(got, "0 ") || got == "0KB" || strings.HasPrefix(got, "0.0") {
			t.Errorf("formatSize(%d) = %q —— 非空文件不能显示成零", n, got)
		}
	}
}