sync: collector/ 同步自 inkstone/arcrun-rag@80b3e5c（桌面小幫手 0.18.52）

cloudcheck_test.go

@@ -0,0 +1,107 @@
+// cloudcheck_test.go — 每輪固定問雲端的兩件事，一分鐘只真的問一次（inkstone/arcrun-rag#121）。
+package collector
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"sync/atomic"
+	"testing"
+	"time"
+)
+
+// init：測試環境預設不快取——既有測試（例如「第一輪查得到版本、第二輪 DNS 抽風」）
+// 是連續兩輪背靠背跑，模擬的是「每一輪都真的問過」。
+// 需要驗證節流本身的測試（下方與 route_backoff_measure_test.go）自行 save/restore 成真值。
+func init() {
+	cloudCheckInterval = 0
+}
+
+func withCloudCheckInterval(t *testing.T, d time.Duration) {
+	t.Helper()
+	old := cloudCheckInterval
+	cloudCheckInterval = d
+	resetCloudChecks()
+	t.Cleanup(func() {
+		cloudCheckInterval = old
+		resetCloudChecks()
+	})
+}
+
+func withClock(t *testing.T, start time.Time) *atomic.Int64 {
+	t.Helper()
+	var ns atomic.Int64
+	ns.Store(start.UnixNano())
+	old := directNow
+	directNow = func() time.Time { return time.Unix(0, ns.Load()).UTC() }
+	t.Cleanup(func() { directNow = old })
+	return &ns
+}
+
+func TestCloudVersion_AskedOncePerMinute(t *testing.T) {
+	withCloudCheckInterval(t, 60*time.Second)
+	t0 := time.Date(2026, 9, 13, 5, 0, 0, 0, time.UTC)
+	clock := withClock(t, t0)
+
+	var calls int
+	orig := fetchCloudVersion
+	fetchCloudVersion = func(string) (string, bool) { calls++; return "1.4.60", true }
+	defer func() { fetchCloudVersion = orig }()
+
+	u := "https://a.workers.dev"
+	for i := 0; i < 12; i++ { // 一分鐘內每 5 秒一輪
+		clock.Store(t0.Add(time.Duration(i*5) * time.Second).UnixNano())
+		if v, ok := cloudVersionThrottled(u, false); v != "1.4.60" || !ok {
+			t.Fatalf("快取回的答案要與真問到的一樣：%q %v", v, ok)
+		}
+	}
+	if calls != 1 {
+		t.Fatalf("一分鐘內 12 輪應該只真的問 1 次，got %d", calls)
+	}
+	clock.Store(t0.Add(60 * time.Second).UnixNano())
+	cloudVersionThrottled(u, false)
+	if calls != 2 {
+		t.Fatalf("滿一分鐘要重問，got %d", calls)
+	}
+	cloudVersionThrottled(u, true)
+	if calls != 3 {
+		t.Fatalf("按「立刻同步」的那一輪要照問，got %d", calls)
+	}
+	cloudVersionThrottled("https://b.workers.dev", false)
+	if calls != 4 {
+		t.Fatalf("不同知識庫各問各的，got %d", calls)
+	}
+}
+
+func TestProbeWorkersAI_AskedOncePerMinute(t *testing.T) {
+	withCloudCheckInterval(t, 60*time.Second)
+	cloudRoutes.reset()
+	defer cloudRoutes.reset()
+	t0 := time.Date(2026, 9, 13, 5, 0, 0, 0, time.UTC)
+	clock := withClock(t, t0)
+
+	var probes atomic.Int32
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if strings.HasSuffix(r.URL.Path, "/portal/daemon/extract") {
+			probes.Add(1)
+			w.WriteHeader(http.StatusBadRequest) // 空 text 的探測：路由存在＝可用
+		}
+	}))
+	defer srv.Close()
+
+	cfg := &DirectConfig{CypherURL: srv.URL, APIKey: "k"}
+	for i := 0; i < 12; i++ {
+		clock.Store(t0.Add(time.Duration(i*5) * time.Second).UnixNano())
+		if st := cfg.probeWorkersAI(); !st.Ready {
+			t.Fatalf("第 %d 輪：探測結果要沿用「可用」，got %+v", i, st)
+		}
+	}
+	if n := probes.Load(); n != 1 {
+		t.Fatalf("一分鐘內 12 輪應該只探測 1 次，got %d", n)
+	}
+	clock.Store(t0.Add(61 * time.Second).UnixNano())
+	cfg.probeWorkersAI()
+	if n := probes.Load(); n != 2 {
+		t.Fatalf("滿一分鐘要重新探測，got %d", n)
+	}
+}