mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-11-01 14:57:38 +00:00
4e70b8d5d8
批 1，不含 SDK 仓与目录改名。 - go.mod 第 1 行 module 路径 - 366 个 .go 文件的完整前缀 github.com/JianFeeeee/HomeAgent 机械替换 （857 处，仅完整前缀；展示名/注释里的 HomeAgent 字样留给批 5） - 未改：third_party 目录名、SDK 自身 go.mod module、go.mod replace 目标、 homed/waiter 可执行名、.hmap 后缀
90 lines
3.1 KiB
Go
90 lines
3.1 KiB
Go
package core
|
||
|
||
import (
|
||
"testing"
|
||
"time"
|
||
|
||
agentIO "github.com/JianFeeeee/HologramAgent/internal/agent/io"
|
||
)
|
||
|
||
// 抢占日志必须能说出**受害者是谁**（来源 + 类别 + 级别）。
|
||
// 此前 suspend/resume 不落日志，生产上无法回答"我的任务被谁打断了"。
|
||
func TestPreemptLogging_NamesVictimAndSource(t *testing.T) {
|
||
sp := newPreemptProvider("intr-done", "low-done")
|
||
a := newPreemptAgent(t, sp)
|
||
|
||
lowEvt, _ := textEvent("qq", "低优先级任务")
|
||
lowTask := newInputTask(lowEvt)
|
||
if !a.sched.enqueue(lowTask) {
|
||
t.Fatal("入队失败")
|
||
}
|
||
lt, _, _ := a.sched.nextRef()
|
||
|
||
done := make(chan struct{})
|
||
go func() { a.executeNewTask(lt); close(done) }()
|
||
select {
|
||
case <-sp.entered:
|
||
case <-time.After(3 * time.Second):
|
||
t.Fatal("provider 未被调用")
|
||
}
|
||
|
||
if a.sched.suspendDepth() != 0 {
|
||
t.Fatalf("初始栈深应为 0，实际 %d", a.sched.suspendDepth())
|
||
}
|
||
|
||
intrEvt := &agentIO.InputEvent{
|
||
Source: "cli", OutputChannel: "cli", Type: "text",
|
||
Payload: map[string]interface{}{"content": "紧急打断", "interrupt": true},
|
||
}
|
||
a.sched.requestPreempt(intrEvt, LevelInteractive)
|
||
a.cancelCurrentLLM()
|
||
|
||
select {
|
||
case <-done:
|
||
case <-time.After(3 * time.Second):
|
||
t.Fatal("未挂起")
|
||
}
|
||
if d := a.sched.suspendDepth(); d != 1 {
|
||
t.Fatalf("挂起后栈深应为 1，实际 %d", d)
|
||
}
|
||
// sourceOf 要能从任务/帧里取出可辨识来源（qq），而不是空串或 task#N。
|
||
if got := sourceOf(lt, nil); got != "qq" {
|
||
t.Fatalf("sourceOf(lt) = %q，期望 qq", got)
|
||
}
|
||
}
|
||
|
||
func TestSourceOf_SelfAndNilAreSafe(t *testing.T) {
|
||
if got := sourceOf(nil, nil); got != "?" {
|
||
t.Fatalf("nil -> %q", got)
|
||
}
|
||
st := newSelfTask(selfInputMsg{text: "x", channel: "cli"})
|
||
if got := sourceOf(st, nil); got != "self:cli" {
|
||
t.Fatalf("self task -> %q", got)
|
||
}
|
||
}
|
||
|
||
// 抢占日志必须把「入侵者」与「受害者」分开写，且受害者是**真正在跑的那个**。
|
||
//
|
||
// 镇的是一个自伤：第一版把日志打在 executeNewTask 里，而那时 nextRef 已经把
|
||
// s.running 换成了抢占者自己，于是日志写成 "victim = 入侵者"（实测输出过
|
||
// `preempt start: task#2 ... -> victim task#2 (cli)`）。判据必须落在
|
||
// registerInterrupt —— 那一刻 running 还没被换。
|
||
func TestSourceOf_DistinctVictimAndIntruder(t *testing.T) {
|
||
if sourceOf(newInputTask(&agentIO.InputEvent{Source: "qq"}), nil) != "qq" {
|
||
t.Fatal("queued qq 任务的来源应为 qq")
|
||
}
|
||
it := newInterruptTask(&agentIO.InputEvent{Source: "homeagent-mail-bridge"}, LevelMessage)
|
||
if got := sourceOf(it, nil); got != "homeagent-mail-bridge" {
|
||
t.Fatalf("中断来源应为 homeagent-mail-bridge，实际 %q", got)
|
||
}
|
||
if it.Class != TaskInterrupt || it.Level != LevelMessage {
|
||
t.Fatalf("中断任务应 class=interrupt level=2，实际 %v/%v", it.Class, it.Level)
|
||
}
|
||
// 邮件若走排队注入则会变成 TaskQueued —— 那种情况下 canPreempt 必为假。
|
||
q := &Task{Class: TaskQueued, Kind: TaskKindInput,
|
||
Event: &agentIO.InputEvent{Source: "homeagent-mail-bridge"}}
|
||
if canPreempt(q, newInputTask(&agentIO.InputEvent{Source: "qq"})) {
|
||
t.Fatal("排队输入永远不得抢占另一个排队任务")
|
||
}
|
||
}
|