sync: collector/ 同步自 inkstone/arcrun-rag@d129f8b（桌面小幫手 0.18.48）

folder_progress_wiring_test.go

@@ -0,0 +1,138 @@
+package collector
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"path/filepath"
+	"testing"
+	"time"
+)
+
+// TestFolderProgressLandsInStatus 守住 `inkstone/arcrun-rag#159` 的接線：
+// **逐資料夾的同步現況真的會寫進 status.json**，而且是每個根各一份。
+//
+// 為什麼要有這一條：小幫手畫面上那個打勾完全靠這個欄位。它沒寫出來的話，
+// 前端拿不到資料 ⇒ 每一列都會落到 unknown ⇒ 使用者又回到「看不出同步了沒」。
+func TestFolderProgressLandsInStatus(t *testing.T) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(http.StatusOK)
+		_, _ = w.Write([]byte(`{"ok":true}`))
+	}))
+	defer srv.Close()
+
+	base := t.TempDir()
+	rootA := filepath.Join(base, "A")
+	rootB := filepath.Join(base, "B")
+	for _, d := range []string{rootA, rootB} {
+		if err := os.MkdirAll(d, 0o755); err != nil {
+			t.Fatal(err)
+		}
+	}
+	_ = os.WriteFile(filepath.Join(rootA, "a1.md"), []byte("# A1"), 0o644)
+	_ = os.WriteFile(filepath.Join(rootA, "a2.md"), []byte("# A2"), 0o644)
+	_ = os.WriteFile(filepath.Join(rootB, "b1.md"), []byte("# B1"), 0o644)
+
+	cfg := &DirectConfig{
+		Manifest:   filepath.Join(base, "manifest.json"),
+		Accounts:   []AccountConfig{{CypherURL: srv.URL, Namespace: "ns", WatchFolders: []string{rootA, rootB}}},
+		MaxRemoved: DefaultMaxRemovedRatio,
+	}
+	RunDirectOnce(cfg, false)
+
+	st, err := LoadSyncStatus(StatusFilePath(cfg.Manifest))
+	if err != nil {
+		t.Fatalf("讀不到 status.json：%v", err)
+	}
+	if len(st.FolderProgress) != 2 {
+		t.Fatalf("folder_progress 應該每個看守資料夾各一份（2），got %d：%+v",
+			len(st.FolderProgress), st.FolderProgress)
+	}
+	if got := st.FolderProgress[rootA].Total; got != 2 {
+		t.Errorf("A 的分母應為 2，got %d", got)
+	}
+	if got := st.FolderProgress[rootB].Total; got != 1 {
+		t.Errorf("B 的分母應為 1，got %d", got)
+	}
+	// 🔴 逐資料夾加起來要等於首頁那行總量——它們本來就是同一組數字被切開，
+	//    對不起來就代表有人另算了一套（那正是這張票要拔掉的病）。
+	var sum SyncProgress
+	for _, p := range st.FolderProgress {
+		sum = sum.Add(p)
+	}
+	if sum.Total != st.Progress.Total-st.Progress.Unreadable {
+		t.Errorf("逐資料夾合計 %d ≠ 首頁總量 %d（扣掉沒進 manifest 的 %d）",
+			sum.Total, st.Progress.Total, st.Progress.Unreadable)
+	}
+}
+
+// TestCloudVersionSurvivesOneFailedProbe 守住票的追加條件 6：
+// **「連不上」的判準不能是「這一輪解析失敗」。**
+//
+// leo 2026-08-28 畫面上寫「目前連不上這個知識庫，查不到版本」，
+// 而同一台機器 curl 打同一支 /health 是 HTTP 200——那是機器當下 DNS 抽風。
+// 一次探測失敗不准抹掉我們已經知道的版本。
+func TestCloudVersionSurvivesOneFailedProbe(t *testing.T) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(http.StatusOK)
+		_, _ = w.Write([]byte(`{"ok":true}`))
+	}))
+	defer srv.Close()
+
+	base := t.TempDir()
+	root := filepath.Join(base, "R")
+	if err := os.MkdirAll(root, 0o755); err != nil {
+		t.Fatal(err)
+	}
+	_ = os.WriteFile(filepath.Join(root, "r.md"), []byte("# R"), 0o644)
+
+	cfg := &DirectConfig{
+		Manifest:   filepath.Join(base, "manifest.json"),
+		Accounts:   []AccountConfig{{CypherURL: srv.URL, Namespace: "ns", WatchFolders: []string{root}}},
+		MaxRemoved: DefaultMaxRemovedRatio,
+	}
+
+	orig := fetchCloudVersion
+	defer func() { fetchCloudVersion = orig }()
+	// 「最新版是多少」也要釘住，否則這條測試會依賴 install.arcrun.dev 通不通
+	// ——那樣測的就不是我們要測的東西了（而且 CI 離線就紅）。
+	origLatest := fetchLatestCloudReleaseRaw
+	fetchLatestCloudReleaseRaw = func() (string, bool) { return "1.4.58", true }
+	latestMu.Lock()
+	latestFetched = time.Time{} // 清掉節流快取，讓 stub 這一輪真的生效
+	latestMu.Unlock()
+	defer func() {
+		fetchLatestCloudReleaseRaw = origLatest
+		latestMu.Lock()
+		latestFetched = time.Time{}
+		latestMu.Unlock()
+	}()
+
+	// 第一輪：查得到版本。
+	fetchCloudVersion = func(string) (string, bool) { return "1.4.60", true }
+	RunDirectOnce(cfg, false)
+
+	// 第二輪：探測失敗（模擬 DNS 抽風）。
+	fetchCloudVersion = func(string) (string, bool) { return "", false }
+	RunDirectOnce(cfg, false)
+
+	st, err := LoadSyncStatus(StatusFilePath(cfg.Manifest))
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(st.AccountDetails) != 1 {
+		t.Fatalf("應該只有一個帳號，got %d", len(st.AccountDetails))
+	}
+	for host, acc := range st.AccountDetails {
+		if acc.CloudVersion != "1.4.60" {
+			t.Errorf("%s：一次探測失敗就把版本抹掉了（got %q，應沿用 1.4.60）", host, acc.CloudVersion)
+		}
+		if !acc.CloudUpdateKnown {
+			t.Errorf("%s：版本明明還知道，卻標成「查不到」", host)
+		}
+		// 可達性是另一件事，要照實記——不准為了讓畫面好看就說連得上。
+		if acc.CloudCheckOK {
+			t.Errorf("%s：這一輪明明沒連上，cloud_check_ok 卻是 true", host)
+		}
+	}
+}