AskaEth
7ae4298871
- 陌生人私聊：不再走「值不值得回」的概率门，唯一闸门 = 每发送者 6 条/分滑窗 + 同会话 1 个在跑回合 - 被限流只登记不回，且留带 source 的可见告警（不静默） - 安静时段在回复侧只作信号（规则层不再静默），群聊取消夜间静默 - 锁定流控常量：FlowDirectDefault 加字面量钉桩测试（放大它 = 静默关流控，且会让额度循环跑百万轮）
332 lines
13 KiB
Go
332 lines
13 KiB
Go
package channelbridge
|
||
|
||
import (
|
||
"context"
|
||
"strings"
|
||
"testing"
|
||
"time"
|
||
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/channel/adapter"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/channel/session"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/history"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/loop"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/model"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/user"
|
||
)
|
||
|
||
// stubLightJudge 是可配置的 LightJudge，记录自己被咨询了多少次，
|
||
// 并保留最后一次收到的输入（用于断言 inQuiet 真值确实传到了下游判定器）。
|
||
type stubLightJudge struct {
|
||
resp LightDecision
|
||
err error
|
||
calls int
|
||
last LightInput
|
||
}
|
||
|
||
func (s *stubLightJudge) Judge(_ context.Context, in LightInput) (LightDecision, error) {
|
||
s.calls++
|
||
s.last = in
|
||
return s.resp, s.err
|
||
}
|
||
|
||
// nonOwnerGroupMsg 返回一条来自成员发送者的第三方群消息，带
|
||
// 指定的提及。内容不做加工，便于测试自行控制信号。
|
||
func nonOwnerGroupMsg(content string, mentions ...adapter.Mention) adapter.UnifiedMessage {
|
||
return adapter.UnifiedMessage{
|
||
Platform: "telegram", ChannelID: "-20", ChannelType: adapter.ChannelGroup,
|
||
GroupID: "-20", GroupName: "Gang", SenderID: "9", SenderName: "张三",
|
||
Sender: adapter.Actor{Role: adapter.RoleMember}, Mentions: mentions,
|
||
Content: content, ContentType: adapter.ContentText,
|
||
}
|
||
}
|
||
|
||
// nonOwnerSession 是非主人测试所用的受限（群聊 / 陌生人）会话
|
||
// 绑定：记忆只读、无主人。
|
||
func nonOwnerSession() session.Session {
|
||
return session.Session{SessionID: "sess-g", IsGroup: true, Owner: false, MemWritable: false}
|
||
}
|
||
|
||
func TestNonOwnerSilencedRegisteredOnly(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
chat := &stubChatter{res: &loop.Result{Text: "不该回"}}
|
||
sess := &stubSessions{sess: nonOwnerSession()}
|
||
hist := &captureHistory{}
|
||
b, err := New(Options{
|
||
Loop: chat, Sessions: sess, Owner: &user.OwnerProfile{}, History: hist,
|
||
Now: func() time.Time { return time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC) },
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("New: %v", err)
|
||
}
|
||
|
||
// 无 @/关键词/名字信号的非主人消息被静默（设计/12 §6 策略二）：
|
||
// 它只登记到隔离会话日志，绝不驱动循环。
|
||
b.process(context.Background(), nonOwnerGroupMsg("hello all"))
|
||
|
||
if chat.sessionID != "" || len(chat.messages) != 0 {
|
||
t.Fatalf("loop ran for a no-signal non-owner message: session=%q messages=%+v", chat.sessionID, chat.messages)
|
||
}
|
||
recs := hist.records()
|
||
if len(recs) != 1 {
|
||
t.Fatalf("records = %d, want 1 (inbound registered, no assistant reply)", len(recs))
|
||
}
|
||
if recs[0].SessionID != "sess-g" || recs[0].Role != history.RoleUser || recs[0].Content != "hello all" {
|
||
t.Fatalf("registered record = %+v, want isolated sess-g user 'hello all'", recs[0])
|
||
}
|
||
}
|
||
|
||
func TestNonOwnerMentionResponds(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
chat := &stubChatter{res: &loop.Result{Text: "收到"}}
|
||
sess := &stubSessions{sess: nonOwnerSession()}
|
||
hist := &captureHistory{}
|
||
b, err := New(Options{
|
||
Loop: chat, Sessions: sess, Owner: &user.OwnerProfile{}, History: hist,
|
||
Now: func() time.Time { return time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC) },
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("New: %v", err)
|
||
}
|
||
|
||
// 被点名（提及）的非主人消息值得回复（策略二），因此它
|
||
// 针对隔离群会话驱动完整循环。
|
||
b.process(context.Background(), nonOwnerGroupMsg("hello", adapter.Mention{UserID: "bot"}))
|
||
|
||
if chat.sessionID != "sess-g" {
|
||
t.Fatalf("loop session = %q, want isolated sess-g", chat.sessionID)
|
||
}
|
||
// 受限群上下文 → 第三方边界系统消息加上
|
||
// 用户消息（design/16 §4.6 §1），与 TestGroupMessageThirdPartySenderNameAndBoundary 一致。
|
||
if len(chat.messages) != 2 || chat.messages[1].Role != model.RoleUser {
|
||
t.Fatalf("loop messages = %+v, want boundary + user", chat.messages)
|
||
}
|
||
if chat.messages[1].Content[0].Text != "hello" {
|
||
t.Fatalf("loop user text = %q, want undecorated 'hello'", chat.messages[1].Content[0].Text)
|
||
}
|
||
if !strings.Contains(chat.messages[0].Content[0].Text, "群「Gang」") {
|
||
t.Fatalf("origin/boundary system = %q, want group Gang", chat.messages[0].Content[0].Text)
|
||
}
|
||
}
|
||
|
||
func TestNonOwnerHighValueKeywordResponds(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
chat := &stubChatter{res: &loop.Result{Text: "我看看"}}
|
||
sess := &stubSessions{sess: nonOwnerSession()}
|
||
b, err := New(Options{
|
||
Loop: chat, Sessions: sess, Owner: &user.OwnerProfile{},
|
||
Now: func() time.Time { return time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC) },
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("New: %v", err)
|
||
}
|
||
|
||
// 高价值关键词（求助）即使没有 @ 也值得回复（策略二）。
|
||
b.process(context.Background(), nonOwnerGroupMsg("有人在吗，紧急求助"))
|
||
if chat.sessionID != "sess-g" {
|
||
t.Fatalf("loop session = %q, want isolated sess-g for a high-value keyword", chat.sessionID)
|
||
}
|
||
}
|
||
|
||
// TestNonOwnerQuietHoursRulesDoNotSilenceGroup 钉住裁决②（主人 2026-10-05）：
|
||
// **安静时段不再是规则层的硬静默**——群里没被 @、无关键词的普通消息在 02:00
|
||
// 仍会走到轻思考（LightJudge），由它决定回不回，而不是连判定器都不咨询就被
|
||
// 静默。变异验证：把 `if inQuiet { return verdictSilent }` 加回 ruleLightJudge，
|
||
// 本用例立刻变红（judge.calls 会是 0、循环不会跑）。
|
||
func TestNonOwnerQuietHoursRulesDoNotSilenceGroup(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
judge := &stubLightJudge{resp: LightDecision{Respond: true, Reason: "夜里这场值得聊"}}
|
||
chat := &stubChatter{res: &loop.Result{Text: "在的"}}
|
||
sess := &stubSessions{sess: nonOwnerSession()}
|
||
b, err := New(Options{
|
||
Loop: chat, Sessions: sess, Owner: &user.OwnerProfile{}, Light: judge,
|
||
QuietStartHour: 23, QuietEndHour: 7,
|
||
Now: func() time.Time { return time.Date(2026, 1, 1, 2, 0, 0, 0, time.UTC) },
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("New: %v", err)
|
||
}
|
||
|
||
// 无 @、无关键词、无主人显示名的群消息，落在安静时段内。
|
||
b.process(context.Background(), nonOwnerGroupMsg("hello all"))
|
||
|
||
if judge.calls != 1 {
|
||
t.Fatalf("light judge calls = %d, want 1（安静时段不得在规则层硬静默，应交给轻思考）", judge.calls)
|
||
}
|
||
// 安静时段作为**信号**必须是真的（而非硬编码 false）传到下游判定器。
|
||
if !judge.last.InQuietHours {
|
||
t.Fatalf("light judge InQuietHours = false, want true at 02:00（安静时段信号失真）")
|
||
}
|
||
if chat.sessionID != "sess-g" {
|
||
t.Fatalf("loop session = %q, want sess-g（轻思考说回就该回）", chat.sessionID)
|
||
}
|
||
}
|
||
|
||
// TestNonOwnerQuietHoursMentionResponds 保留「安静时段不是硬性全静默」的正面口径：
|
||
// 被直接点名的群消息在 02:00 仍直接回，且**根本不咨询轻思考**（规则已能给结论）。
|
||
func TestNonOwnerQuietHoursMentionResponds(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
judge := &stubLightJudge{resp: LightDecision{Respond: false, Reason: "不该被问到"}}
|
||
chat := &stubChatter{res: &loop.Result{Text: "在的"}}
|
||
sess := &stubSessions{sess: nonOwnerSession()}
|
||
b, err := New(Options{
|
||
Loop: chat, Sessions: sess, Owner: &user.OwnerProfile{}, Light: judge,
|
||
QuietStartHour: 23, QuietEndHour: 7,
|
||
Now: func() time.Time { return time.Date(2026, 1, 1, 2, 0, 0, 0, time.UTC) },
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("New: %v", err)
|
||
}
|
||
|
||
// 策略二：被直接点名的消息在安静时段仍会得到回复。
|
||
b.process(context.Background(), nonOwnerGroupMsg("在吗", adapter.Mention{UserID: "bot"}))
|
||
if chat.sessionID != "sess-g" {
|
||
t.Fatalf("loop session = %q, want sess-g (mention should break quiet-hours silence)", chat.sessionID)
|
||
}
|
||
if judge.calls != 0 {
|
||
t.Fatalf("light judge calls = %d, want 0（@ 是规则可直接判定的强信号）", judge.calls)
|
||
}
|
||
}
|
||
|
||
// TestNonOwnerQuietHoursKeywordResponds 补齐另一条强信号：命中高价值关键词的群消息
|
||
// 在 02:00 也直接回，同样不咨询轻思考。
|
||
func TestNonOwnerQuietHoursKeywordResponds(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
judge := &stubLightJudge{resp: LightDecision{Respond: false, Reason: "不该被问到"}}
|
||
chat := &stubChatter{res: &loop.Result{Text: "我看看"}}
|
||
sess := &stubSessions{sess: nonOwnerSession()}
|
||
b, err := New(Options{
|
||
Loop: chat, Sessions: sess, Owner: &user.OwnerProfile{}, Light: judge,
|
||
QuietStartHour: 23, QuietEndHour: 7,
|
||
Now: func() time.Time { return time.Date(2026, 1, 1, 2, 0, 0, 0, time.UTC) },
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("New: %v", err)
|
||
}
|
||
|
||
b.process(context.Background(), nonOwnerGroupMsg("有人在吗，紧急求助"))
|
||
if chat.sessionID != "sess-g" {
|
||
t.Fatalf("loop session = %q, want sess-g (keyword should break quiet-hours silence)", chat.sessionID)
|
||
}
|
||
if judge.calls != 0 {
|
||
t.Fatalf("light judge calls = %d, want 0（高价值关键词是规则可直接判定的强信号）", judge.calls)
|
||
}
|
||
}
|
||
|
||
func TestOwnerUnaffectedInQuietHours(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
chat := &stubChatter{res: &loop.Result{Text: "主人好"}}
|
||
sess := &stubSessions{sess: session.Session{SessionID: "sess-1", Owner: true, MemWritable: true}}
|
||
ad := &fakeAdapter{name: "telegram"}
|
||
b, err := New(Options{
|
||
Loop: chat, Sessions: sess, Owner: testOwnerProfile("telegram", "100"), History: &captureHistory{},
|
||
QuietStartHour: 23, QuietEndHour: 7,
|
||
Now: func() time.Time { return time.Date(2026, 1, 1, 2, 0, 0, 0, time.UTC) },
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("New: %v", err)
|
||
}
|
||
b.Add(ad)
|
||
|
||
// 主人消息绝不被拦，即使处于安静时段（策略五只管
|
||
// 非主人流量）：主人私聊仍路由到 Web 主会话
|
||
// 并回复。
|
||
b.process(context.Background(), directMessage())
|
||
|
||
if chat.sessionID != history.MainSessionID {
|
||
t.Fatalf("owner loop session = %q, want %q", chat.sessionID, history.MainSessionID)
|
||
}
|
||
if len(ad.sends) != 1 || ad.sends[0].Content != "主人好" {
|
||
t.Fatalf("owner reply = %+v, want '主人好'", ad.sends)
|
||
}
|
||
}
|
||
|
||
func TestLightJudgeConsultedWhenInconclusive(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
// 白天无信号的消息对规则而言无结论，因此注入的
|
||
// LightJudge 获得一次判断机会（设计/12 §6 策略二：规则不确定才轻 LLM）。
|
||
judge := &stubLightJudge{resp: LightDecision{Respond: true, Reason: "值得回"}}
|
||
chat := &stubChatter{res: &loop.Result{Text: "嗯"}}
|
||
sess := &stubSessions{sess: nonOwnerSession()}
|
||
b, err := New(Options{
|
||
Loop: chat, Sessions: sess, Owner: &user.OwnerProfile{},
|
||
Light: judge,
|
||
Now: func() time.Time { return time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC) },
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("New: %v", err)
|
||
}
|
||
|
||
b.process(context.Background(), nonOwnerGroupMsg("今天天气不错"))
|
||
if judge.calls != 1 {
|
||
t.Fatalf("light judge calls = %d, want 1 (inconclusive rules escalated)", judge.calls)
|
||
}
|
||
if chat.sessionID != "sess-g" {
|
||
t.Fatalf("loop session = %q, want sess-g (judge said respond)", chat.sessionID)
|
||
}
|
||
}
|
||
|
||
func TestLightJudgeSilenceIsRespected(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
judge := &stubLightJudge{resp: LightDecision{Respond: false, Reason: "不值得"}}
|
||
chat := &stubChatter{res: &loop.Result{Text: "不该回"}}
|
||
sess := &stubSessions{sess: nonOwnerSession()}
|
||
hist := &captureHistory{}
|
||
b, err := New(Options{
|
||
Loop: chat, Sessions: sess, Owner: &user.OwnerProfile{}, History: hist,
|
||
Light: judge,
|
||
Now: func() time.Time { return time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC) },
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("New: %v", err)
|
||
}
|
||
|
||
b.process(context.Background(), nonOwnerGroupMsg("今天天气不错"))
|
||
if judge.calls != 1 {
|
||
t.Fatalf("light judge calls = %d, want 1", judge.calls)
|
||
}
|
||
if chat.sessionID != "" {
|
||
t.Fatalf("loop ran after the judge said silence: session=%q", chat.sessionID)
|
||
}
|
||
if len(hist.records()) != 1 {
|
||
t.Fatalf("records = %d, want 1 (registered, no reply)", len(hist.records()))
|
||
}
|
||
}
|
||
|
||
func TestLightJudgeErrorFailsClosedToSilence(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
judge := &stubLightJudge{err: context.DeadlineExceeded}
|
||
chat := &stubChatter{res: &loop.Result{Text: "不该回"}}
|
||
sess := &stubSessions{sess: nonOwnerSession()}
|
||
hist := &captureHistory{}
|
||
b, err := New(Options{
|
||
Loop: chat, Sessions: sess, Owner: &user.OwnerProfile{}, History: hist,
|
||
Light: judge,
|
||
Now: func() time.Time { return time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC) },
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("New: %v", err)
|
||
}
|
||
|
||
// 轻量判定器出错时失败关闭为静默（不打扰为第一原则）。
|
||
b.process(context.Background(), nonOwnerGroupMsg("今天天气不错"))
|
||
if judge.calls != 1 {
|
||
t.Fatalf("light judge calls = %d, want 1", judge.calls)
|
||
}
|
||
if chat.sessionID != "" || len(chat.messages) != 0 {
|
||
t.Fatalf("loop ran after judge error: session=%q messages=%+v", chat.sessionID, chat.messages)
|
||
}
|
||
if len(hist.records()) != 1 {
|
||
t.Fatalf("records = %d, want 1 (registered, no reply)", len(hist.records()))
|
||
}
|
||
}
|