arcrun-collector/supervisor/supervisor_test.go

package supervisor

import (
	"os"
	"path/filepath"
	"runtime"
	"sync/atomic"
	"testing"
	"time"
)

// fakeCollector 寫一個假的 collector 執行檔（shell 腳本）：忽略參數、印 n 輪 JSON。
// stayAlive=true → 印完 sleep（模擬常駐 daemon，狀態維持 watching）；false → 印完即退出（觸發重起）。
func fakeCollector(t *testing.T, rounds int, stayAlive bool) string {
	t.Helper()
	if runtime.GOOS == "windows" {
		t.Skip("fake collector script 走 POSIX shell；Windows 測略過（supervisor 邏輯與平台無關）")
	}
	dir := t.TempDir()
	p := filepath.Join(dir, "fake-collector.sh")
	body := "#!/bin/sh\n"
	for i := 0; i < rounds; i++ {
		body += "printf '{\"at\":\"2026-07-21T00:00:0" + itoa(i) + "Z\",\"folder\":\"/tmp/kb\",\"results\":[]}\\n'\n"
	}
	if stayAlive {
		body += "sleep 2\n"
	}
	if err := os.WriteFile(p, []byte(body), 0o755); err != nil {
		t.Fatal(err)
	}
	return p
}

func itoa(i int) string { return string(rune('0' + i)) }

func waitFor(t *testing.T, d time.Duration, cond func() bool, msg string) {
	t.Helper()
	deadline := time.Now().Add(d)
	for time.Now().Before(deadline) {
		if cond() {
			return
		}
		time.Sleep(5 * time.Millisecond)
	}
	t.Fatalf("timeout waiting for: %s", msg)
}

func TestSupervisorParsesRoundsAndWatches(t *testing.T) {
	bin := fakeCollector(t, 2, true)
	s := New(bin, "cfg.json")
	s.Backoff = 20 * time.Millisecond
	s.Start()
	defer s.Stop()

	waitFor(t, 2*time.Second, func() bool { return s.Status().Rounds >= 2 }, "2 rounds parsed")
	st := s.Status()
	if st.State != StateWatching {
		t.Fatalf("expected watching, got %q", st.State)
	}
	if st.LastRoundAt.IsZero() {
		t.Fatal("LastRoundAt should be parsed from stdout 'at'")
	}
	if st.LastRoundAt.Year() != 2026 {
		t.Fatalf("LastRoundAt parse wrong: %v", st.LastRoundAt)
	}
}

func TestSupervisorRestartsOnExit(t *testing.T) {
	bin := fakeCollector(t, 1, false)
	s := New(bin, "cfg.json")
	s.Backoff = 20 * time.Millisecond
	s.Start()
	defer s.Stop()

	// 假 collector 印一輪就退出 → supervisor 視為非預期退出 → 重起
	waitFor(t, 2*time.Second, func() bool { return s.Status().Restarts >= 2 }, "restarts accumulate")
}

func TestSupervisorStop(t *testing.T) {
	bin := fakeCollector(t, 1, false)
	s := New(bin, "cfg.json")
	s.Backoff = 20 * time.Millisecond
	s.Start()
	waitFor(t, 2*time.Second, func() bool { return s.Status().Restarts >= 1 }, "running")
	s.Stop()
	waitFor(t, 2*time.Second, func() bool { return s.Status().State == StateStopped }, "stopped after Stop()")
	// Stop 後不應再累加重起
	r := s.Status().Restarts
	time.Sleep(120 * time.Millisecond)
	if s.Status().Restarts != r {
		t.Fatalf("restarts kept growing after Stop: %d → %d", r, s.Status().Restarts)
	}
}

// 換資料夾＝Stop 緊接 Start；舊 loop 的 setState(stopped) 不可覆寫新 loop 的狀態。
func TestSupervisorStopThenStartNotLeftStopped(t *testing.T) {
	bin := fakeCollector(t, 2, true) // stayAlive → 新 loop 進 watching
	s := New(bin, "cfg.json")
	s.Backoff = 20 * time.Millisecond
	s.Start()
	waitFor(t, 2*time.Second, func() bool { return s.Status().State == StateWatching }, "first watching")
	s.Stop()  // 同步等舊 loop 收掉
	s.Start() // 立刻重起（模擬換資料夾）
	defer s.Stop()
	waitFor(t, 2*time.Second, func() bool { return s.Status().State == StateWatching }, "watching again after Stop→Start")
	// 稍等，確認不會被舊 loop 尾巴改回 stopped
	time.Sleep(100 * time.Millisecond)
	if got := s.Status().State; got != StateWatching {
		t.Fatalf("Stop→Start left state=%q (舊 loop 覆寫了新狀態)", got)
	}
}

func TestSupervisorOnChangeFires(t *testing.T) {
	bin := fakeCollector(t, 2, true)
	s := New(bin, "cfg.json")
	s.Backoff = 20 * time.Millisecond
	var calls int32
	s.SetOnChange(func(Status) { atomic.AddInt32(&calls, 1) })
	s.Start()
	defer s.Stop()
	waitFor(t, 2*time.Second, func() bool { return atomic.LoadInt32(&calls) > 0 }, "onChange fired")
}