139 lines
5.0 KiB
Go
139 lines
5.0 KiB
Go
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)
|
||
}
|
||
}
|
||
}
|