teai/internal/workflow/stalled_entry_test.go

// stalled_entry_test.go — RunStalled／ResolveIssueXrefs 高階入口的單元測試
// （#7 CLI 接線；資料源注入假實作，離線可測）。
package workflow

import (
	"errors"
	"fmt"
	"strings"
	"testing"
	"time"
)

// fakeStalledSource 包 fakeSource 補上 StalledSource 面向。
type fakeStalledSource struct {
	*fakeSource
	username string
	// getIssueErr 模擬 GetIssue 失敗（xrefs 的 404 路徑）。
	getIssueErr error
	// timeline[owner/repo/number]：issue timeline 事件（無則回空清單，
	// 模擬 API 成功但無事件；timelineErr 模擬拉取失敗）。
	timeline    map[string][]TimelineEvent
	timelineErr map[string]error
}

func (f *fakeStalledSource) CurrentUser() (string, error) { return f.username, nil }

func (f *fakeStalledSource) IssueTimeline(owner, repo string, number int64) ([]TimelineEvent, error) {
	key := fmt.Sprintf("%s/%s/%d", owner, repo, number)
	if err := f.timelineErr[key]; err != nil {
		return nil, err
	}
	return f.timeline[key], nil
}

func (f *fakeStalledSource) GetIssue(owner, repo string, number int64) (Issue, error) {
	if f.getIssueErr != nil {
		return Issue{}, f.getIssueErr
	}
	i, ok := f.issues[owner+"/"+repo][number]
	if !ok {
		return Issue{}, &notFoundError{}
	}
	return i, nil
}

type notFoundError struct{}

func (e *notFoundError) Error() string { return "404" }

// newStalledFixture 建構掃描情境：alterminal 組織兩個倉庫，
// agents#1 是父追蹤項（引用 teai#2），teai#2 有較新留言。
func newStalledFixture() *fakeStalledSource {
	return &fakeStalledSource{
		fakeSource: &fakeSource{
			orgs: []string{"alterminal"},
			repos: map[string][]Repo{"alterminal": {
				{Name: "agents", HasPullRequests: boolPtr(true)},
				{Name: "teai", HasPullRequests: boolPtr(true)},
			}},
			issues: map[string]map[int64]Issue{
				"alterminal/agents": {
					1: {Number: 1, Title: "父追蹤項", State: "open",
						Body: "子項：teai#2", CreatedAt: iso(-600), UpdatedAt: iso(-600),
						Repository: &Repository{FullName: "alterminal/agents"}},
				},
				"alterminal/teai": {
					2: {Number: 2, Title: "子項", State: "open",
						CreatedAt: iso(-500), UpdatedAt: iso(-90),
						Repository: &Repository{FullName: "alterminal/teai"}},
				},
			},
			pulls: map[string][]PullRequest{},
			comments: map[string][]Comment{
				"alterminal/teai/2": {
					{CreatedAt: iso(-90), User: &User{UserName: "alex"}, Body: "推進中"},
				},
			},
		},
		username: "ceo",
	}
}

// TestRunStalledParentTracking 驗證 RunStalled：父追蹤項把子項活動
// 計入 last_activity（reason=parent-tracking），且門檻預設與覆寫皆可。
func TestRunStalledParentTracking(t *testing.T) {
	src := newStalledFixture()
	// 預設 4 小時門檻：子項 90 分鐘前有活動 → 父項 last_activity 同步，
	// 未過門檻，清單為空。
	items, err := RunStalled(src, 0, at(0))
	if err != nil {
		t.Fatalf("RunStalled: %v", err)
	}
	if len(items) != 0 {
		t.Fatalf("預設門檻下不應有停滯項，得到 %d 筆", len(items))
	}
	// 門檻降到 1 小時：父追蹤項 last_activity 取子項活動（-90m）→
	// 停滯 1.5h 列入（reason=parent-tracking、附 child_activity）；
	// 子項 teai#2 本身也停滯 1.5h（no-assignee）。
	items, err = RunStalled(src, 1, at(0))
	if err != nil {
		t.Fatalf("RunStalled: %v", err)
	}
	if len(items) != 2 {
		t.Fatalf("應有兩筆（父追蹤項＋子項），得到 %d：%+v", len(items), items)
	}
	var parent *Item
	for i := range items {
		if items[i].Number == 1 {
			parent = &items[i]
		}
	}
	if parent == nil {
		t.Fatalf("找不到 agents#1：%+v", items)
	}
	if parent.Reason != "parent-tracking" || parent.Repo != "alterminal/agents" {
		t.Fatalf("agents#1 應為 parent-tracking，得到 %s（%s）", parent.Reason, parent.Repo)
	}
	if len(parent.ChildActivity) == 0 {
		t.Fatalf("parent-tracking 應附 child_activity")
	}
	if parent.StalledHours != 1.5 {
		t.Fatalf("last_activity 應取子項活動（-90m → 1.5h），得到 %.1f", parent.StalledHours)
	}
}

// TestRunStalledNudgeCooldown 驗證催促冷卻：最後留言出自目前帳號
// 且在 24 小時內 → 不列入；冷卻過後再出現。
func TestRunStalledNudgeCooldown(t *testing.T) {
	src := newStalledFixture()
	setIssueBody(src, "alterminal", "agents", 1, "") // 移除子項引用
	src.comments["alterminal/agents/1"] = []Comment{
		{CreatedAt: iso(-2), User: &User{UserName: "ceo"}},
	}
	findOne := func(items []Item) bool {
		for _, it := range items {
			if it.Repo == "alterminal/agents" && it.Number == 1 {
				return true
			}
		}
		return false
	}
	items, err := RunStalled(src, 1, at(0))
	if err != nil {
		t.Fatalf("RunStalled: %v", err)
	}
	if findOne(items) {
		t.Fatalf("催促冷卻期內（2 小時前留言）agents#1 不應列入，得到 %+v", items)
	}
	src.comments["alterminal/agents/1"] = []Comment{
		{CreatedAt: iso(-30 * 60), User: &User{UserName: "ceo"}},
	}
	items, err = RunStalled(src, 1, at(0))
	if err != nil {
		t.Fatalf("RunStalled: %v", err)
	}
	if !findOne(items) {
		t.Fatalf("冷卻（24h）過後 agents#1 應再出現，得到 %+v", items)
	}
}

// TestResolveIssueXrefs 驗證 xrefs 入口：內文與留言的引用解析、
// 去重、出處標記與單段 repo 的存在性邊界。
func TestResolveIssueXrefs(t *testing.T) {
	src := newStalledFixture()
	setIssueBody(src, "alterminal", "agents", 1, "看 teai#2 與 #7，另見 alterminal/teai#2")
	src.comments["alterminal/agents/1"] = []Comment{
		{Body: "teai#2 已處理", User: &User{UserName: "alex"}, CreatedAt: iso(-10)},
		{Body: "nope#9 不存在", User: &User{UserName: "max"}, CreatedAt: iso(-5)},
	}
	// fixture 的組織倉庫只有 agents → teai#2 單段不解析（repo 不在
	// alterminal 組織清單）；補上 teai 倉庫讓單段解析成立。
	src.repos["alterminal"] = []Repo{
		{Name: "agents", HasPullRequests: boolPtr(true)},
		{Name: "teai", HasPullRequests: boolPtr(true)},
	}
	hits, err := ResolveIssueXrefs(src, "alterminal", "agents", 1)
	if err != nil {
		t.Fatalf("ResolveIssueXrefs: %v", err)
	}
	want := []XrefHit{
		{Owner: "alterminal", Repo: "teai", Number: 2, Source: "body"},
		{Owner: "alterminal", Repo: "agents", Number: 7, Source: "body"},
	}
	if len(hits) != len(want) {
		t.Fatalf("應解析出 %d 筆（留言重複的 teai#2 去重、nope#9 略過），得到 %d：%+v",
			len(want), len(hits), hits)
	}
	for i := range want {
		if hits[i] != want[i] {
			t.Fatalf("hits[%d] = %+v, want %+v", i, hits[i], want[i])
		}
	}
}

// TestResolveIssueXrefsCommentSource 驗證出處：僅出現在留言的引用
// 標記為 comment。
func TestResolveIssueXrefsCommentSource(t *testing.T) {
	src := newStalledFixture()
	setIssueBody(src, "alterminal", "agents", 1, "內文無引用")
	src.comments["alterminal/agents/1"] = []Comment{
		{Body: "追蹤 #9", User: &User{UserName: "alex"}, CreatedAt: iso(-10)},
	}
	hits, err := ResolveIssueXrefs(src, "alterminal", "agents", 1)
	if err != nil {
		t.Fatalf("ResolveIssueXrefs: %v", err)
	}
	if len(hits) != 1 || hits[0].Source != "comment" || hits[0].Number != 9 {
		t.Fatalf("應輸出一筆 source=comment 的 #9，得到 %+v", hits)
	}
}

// TestResolveIssueXrefsGetError 驗證 GetIssue 失敗（如 404）層層回傳，
// 由 CLI 映射結束碼。
func TestResolveIssueXrefsGetError(t *testing.T) {
	src := newStalledFixture()
	src.getIssueErr = &notFoundError{}
	if _, err := ResolveIssueXrefs(src, "alterminal", "agents", 1); err == nil {
		t.Fatalf("GetIssue 失敗應回傳錯誤")
	}
}

func boolPtr(b bool) *bool { return &b }

// setIssueBody 修改 map 內 issue 的內文（map 元素不可定址，需整體重塞）。
func setIssueBody(src *fakeStalledSource, owner, repo string, number int64, body string) {
	key := owner + "/" + repo
	issue := src.issues[key][number]
	issue.Body = body
	src.issues[key][number] = issue
}

// TestResolveIssueXrefsEmptyBody 驗證無引用時輸出空（CLI 印 []）。
func TestResolveIssueXrefsEmptyBody(t *testing.T) {
	src := newStalledFixture()
	setIssueBody(src, "alterminal", "agents", 1, strings.Repeat("無引用內文", 3))
	delete(src.comments, "alterminal/agents/1")
	hits, err := ResolveIssueXrefs(src, "alterminal", "agents", 1)
	if err != nil {
		t.Fatalf("ResolveIssueXrefs: %v", err)
	}
	if len(hits) != 0 {
		t.Fatalf("無引用應輸出空清單，得到 %+v", hits)
	}
	_ = time.Now // time 僅供 iso/at 使用；保留 import。
}

// TestRunStalledChildXrefRefresh 回歸測試（agents #38、teai #34）：
// 子項 updated_at 被跨 issue 引用（comment_ref）刷新時，子項有效活動
// 應經 directLastActivity 排除引用事件後採計，父追蹤項的停滯時鐘
// 不被虛假刷新。
func TestRunStalledChildXrefRefresh(t *testing.T) {
	src := newStalledFixture()
	// 子項 teai#2：直接活動為 created(-500) 與留言(-90)，
	// updated_at 卻是 -10（他處留言提及刷新）。
	issue := src.issues["alterminal/teai"][2]
	issue.UpdatedAt = iso(-10)
	src.issues["alterminal/teai"][2] = issue
	// timeline：-10 的刷新是 comment_ref（引用事件，排除），
	// -90 有直接事件（label）可對應 → 有效活動 -90。
	src.timeline = map[string][]TimelineEvent{
		"alterminal/teai/2": {
			{Type: "label", CreatedAt: iso(-90)},
			{Type: "comment_ref", CreatedAt: iso(-10)},
		},
	}
	// 門檻 1 小時：父項 last_activity 取子項有效活動 -90 → 停滯 1.5h 列入；
	// 若誤採被刷新的 -10，父項與子項都不會列入。
	items, err := RunStalled(src, 1, at(0))
	if err != nil {
		t.Fatalf("RunStalled: %v", err)
	}
	if len(items) != 2 {
		t.Fatalf("應有兩筆（父追蹤項＋子項），得到 %d：%+v", len(items), items)
	}
	var parent *Item
	for i := range items {
		if items[i].Repo == "alterminal/agents" && items[i].Number == 1 {
			parent = &items[i]
		}
	}
	if parent == nil {
		t.Fatalf("找不到 agents#1：%+v", items)
	}
	if parent.Reason != "parent-tracking" {
		t.Fatalf("agents#1 應為 parent-tracking，得到 %s", parent.Reason)
	}
	if parent.StalledHours != 1.5 {
		t.Fatalf("父項停滯應以子項有效活動（-90m → 1.5h）計，得到 %.1f", parent.StalledHours)
	}
}

// TestRunStalledTimelineAllXrefFallback 回歸測試（agents #40、teai #34）：
// issue 自身 updated_at 晚於全部已知直接活動、且 timeline 全為引用事件時，
// 回退已知直接活動的最晚時刻（不高估活動、不低估停滯）；
// timeline 拉取失敗則保守沿用 updated_at。
func TestRunStalledTimelineAllXrefFallback(t *testing.T) {
	src := newStalledFixture()
	setIssueBody(src, "alterminal", "agents", 1, "") // 移除父項引用，teai#2 獨立判定
	delete(src.comments, "alterminal/teai/2")
	// agents#1 給一則近期留言，避免本身停滯（10h）干擾本測試的斷言。
	src.comments["alterminal/agents/1"] = []Comment{
		{CreatedAt: iso(-10), User: &User{UserName: "alex"}},
	}
	// teai#2：僅有 created(-500) 此一已知直接活動，
	// updated_at -10 完全由他處提及刷新。
	issue := src.issues["alterminal/teai"][2]
	issue.UpdatedAt = iso(-10)
	src.issues["alterminal/teai"][2] = issue
	// 情境一：timeline 全為引用事件 → 回退 knownMax（created -500）
	// → 停滯 8.3h，超過預設門檻 4h，列入。
	src.timeline = map[string][]TimelineEvent{
		"alterminal/teai/2": {{Type: "issue_ref", CreatedAt: iso(-10)}},
	}
	items, err := RunStalled(src, 0, at(0))
	if err != nil {
		t.Fatalf("RunStalled: %v", err)
	}
	if len(items) != 1 || items[0].Number != 2 {
		t.Fatalf("全引用事件應回退直接活動 → teai#2 停滯 8.3h 列入，得到 %+v", items)
	}
	if items[0].StalledHours != 8.3 {
		t.Fatalf("回退 created(-500) → 停滯應 8.3h，得到 %.1f", items[0].StalledHours)
	}
	// 情境二：timeline 拉取失敗 → 保守沿用 updated_at(-10) → 0.2h 未達門檻，不列入。
	src.timeline = nil
	src.timelineErr = map[string]error{"alterminal/teai/2": errors.New("boom")}
	items, err = RunStalled(src, 0, at(0))
	if err != nil {
		t.Fatalf("RunStalled: %v", err)
	}
	if len(items) != 0 {
		t.Fatalf("timeline 失敗應保守沿用 updated_at → 不列入，得到 %+v", items)
	}
}