package memory import ( "testing" "yunyan/comm" ) // 模型时常无视「不要 markdown 代码块」,也常在数组前后带一句解释。 // 这些都不是异常输入,是实测里的常态,解析必须扛得住。 func TestParseExtractedTodos_Tolerance(t *testing.T) { cases := []struct { name string raw string want int }{ {"裸数组", `[{"title":"跟进A客户","owner":"王总","due_date":"2026-09-10","due_raw":"下周四前"}]`, 1}, {"带 json 代码块", "```json\n[{\"title\":\"跟进A客户\"}]\n```", 1}, {"带无语言代码块", "```\n[{\"title\":\"跟进A客户\"}]\n```", 1}, {"前后有解释文字", "好的,以下是待办:\n[{\"title\":\"跟进A客户\"}]\n希望对你有帮助。", 1}, {"空数组", `[]`, 0}, {"空数组带代码块", "```json\n[]\n```", 0}, {"多条", `[{"title":"A"},{"title":"B"},{"title":"C"}]`, 3}, } for _, c := range cases { got, err := parseExtractedTodos(c.raw) if err != nil { t.Errorf("%s: 解析失败 %v", c.name, err) continue } if len(got) != c.want { t.Errorf("%s: 期望 %d 条,得到 %d 条", c.name, c.want, len(got)) } } } // 真正解析不出来时必须报错,不能悄悄返回空数组—— // 那样「模型输出坏了」和「本次会议确实没有待办」就分不开了。 func TestParseExtractedTodos_RealFailureReturnsError(t *testing.T) { for _, raw := range []string{"", "本次会议无明确待办事项。", "{不是JSON"} { if _, err := parseExtractedTodos(raw); err == nil { t.Errorf("输入 %q 应该报错而不是返回空数组", raw) } } } func TestParseExtractedTodos_FieldMapping(t *testing.T) { raw := `[{"title":"跟进A客户","owner":"王总","due_date":"2026-09-10","due_raw":"下周四前"}]` got, err := parseExtractedTodos(raw) if err != nil || len(got) != 1 { t.Fatalf("解析失败 err=%v n=%d", err, len(got)) } if got[0].Title != "跟进A客户" || got[0].Owner != "王总" || got[0].DueDate != "2026-09-10" || got[0].DueRaw != "下周四前" { t.Errorf("字段映射错: %+v", got[0]) } } // 内容门槛:拦住 6 秒测试通话、20 秒单人自述这类东西, // 同时不能误伤「上游没给数据」和「运营把门槛关掉」两种情况。 func TestMeetingContentEnough(t *testing.T) { const ( minSec = int32(60) minChars = 120 ) cases := []struct { name string seconds int32 runes int minSec int32 minChars int want bool }{ {"6秒测试通话", 6, 300, minSec, minChars, false}, {"20秒单人自述:时长先拦住", 20, 90, minSec, minChars, false}, {"够长但正文太薄(长段噪音)", 600, 40, minSec, minChars, false}, {"正常会议", 900, 3000, minSec, minChars, true}, {"刚好卡在门槛上算通过", 60, 120, minSec, minChars, true}, // 下面两条是刻意的失败方向:宁可多抽,也不要因为一个字段没传就静默关掉抽取 {"上游没给时长 → 不按时长拦", 0, 3000, minSec, minChars, true}, {"上游没给字数 → 不按字数拦", 900, 0, minSec, minChars, true}, {"门槛配 0 = 不限", 6, 10, 0, 0, true}, } for _, c := range cases { t.Run(c.name, func(t *testing.T) { got, why := meetingContentEnough(c.seconds, c.runes, c.minSec, c.minChars) if got != c.want { t.Fatalf("期望 %v,得到 %v(%s)", c.want, got, why) } if !got && why == "" { t.Error("拦下来了却没给出原因,日志里会只剩一句「跳过抽取」") } }) } } // 门槛的默认值必须真的生效:LoadConfig 不写这两项时是 60/120, // 显式写 0 则表示不限——两者不能混为一谈(零值与「没配」在 mapstructure 里长得一样)。 func TestOptionsExtractThresholdDefaults(t *testing.T) { var noCfg Options if err := noCfg.LoadConfig(map[string]interface{}{}); err != nil { t.Fatal(err) } if noCfg.MeetingExtractMinSeconds != comm.MeetingMinSeconds || noCfg.MeetingExtractMinChars != 120 { t.Errorf("没配时应取默认 %d/120,得到 %d/%d", comm.MeetingMinSeconds, noCfg.MeetingExtractMinSeconds, noCfg.MeetingExtractMinChars) } var zeroCfg Options if err := zeroCfg.LoadConfig(map[string]interface{}{ "meeting_extract_min_seconds": 0, "meeting_extract_min_chars": 0, }); err != nil { t.Fatal(err) } if zeroCfg.MeetingExtractMinSeconds != 0 || zeroCfg.MeetingExtractMinChars != 0 { t.Errorf("显式配 0 应表示不限,得到 %d/%d", zeroCfg.MeetingExtractMinSeconds, zeroCfg.MeetingExtractMinChars) } }