package core

import (
	"testing"
	"time"

	agentIO "gitcode.com/JianFeeeee/HomeAgent/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)
	}
}