visionA/local-tool/server/internal/api/ws/hub_sentinel_test.go
jim800121chen 8cd5751ce3 feat(local-tool): M8 重構 — Wails 控制台 + 瀏覽器 Web UI(R5 決策)
依 R5 五輪決策把 visionA-local 從「Wails 內嵌 Next.js」重構為「Wails
本機伺服器控制台 + 瀏覽器 Web UI」模式(類比 Docker Desktop / Ollama)。

程式碼變動
  - M8-1 砍 yt-dlp 全套(後端 resolver / URL handler / 前端 URL tab /
    Makefile vendor / installer / bootstrap / CI workflow,-555 行)
  - M8-2 砍 Mock 模式全套(driver/mock、mock_camera、Settings runtimeMode、
    VISIONA_MOCK 環境變數,-528 行)
  - M8-3 ffmpeg 從 GPL 切換到 LGPL 混合方案:Windows/Linux 用 BtbN 現成
    LGPL binary,macOS 自 build minimal decoder-only 進 git
    (vendor/ffmpeg/macos/ffmpeg 5.7MB + ffprobe 5.6MB,比 GPL 版省 85% 空間)
  - M8-4 Wails Server Controller:state machine、log ring buffer 2000 行、
    preferences.json atomic write、boot-id、Gin SkipPaths、shutdown 7+1 秒、
    notify_*.go 三平台 OS 通知、watchServer 改 Error state 不 os.Exit
  - M8-4b 啟動階段管線 R5-E:6 階段進度 event、20s soft / 60s hard timeout、
    stage 5/6 skip 規則、sentinel file、RestartStartupSequence 5 步驟
  - M8-5 Wails 控制台 vanilla HTML/JS/CSS(9 檔 ~2012 行)取代 M7-B splash:
    state 視覺、log panel、startup progress panel、Stage 6 manual CTA
    pulse、shutdown modal、Settings、Dark Mode、i18n 中英雙語
  - M8-6 上傳影片副檔名擴充(mp4/avi/mov/mpeg/mpg)
  - M8-7 Web UI Server Offline Overlay(role=alertdialog + focus trap +
    wsEverConnected 容錯 + Page Visibility)
  - M8-8 CORS middleware(127.0.0.1/localhost only + suffix attack 防護)+
    ws/origin.go 獨立 WebSocket CheckOrigin 避 package cycle
  - MAJ-4 server:shutdown-imminent WebSocket broadcast 機制
    (/ws/system endpoint + notifyShutdownImminent helper)
  - M8-9 Boot-ID + 瀏覽器 tab 自動重連(sessionStorage loop guard)

品質
  - ~105+ 新 unit test + race detector (-count=2) 全綠
  - 10 個 milestone 全部通過 Reviewer 審查
  - 三方 v2 + v2.1 文件(PRD / Design Spec / TDD)+ 交叉互審紀錄
    收錄在 .autoflow/

交付前待處理(M8-10)
  - 重跑 make payload-macos 把舊 GPL 77MB binary 換成新 LGPL
  - 三平台 end-to-end build 驗證

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-15 17:57:54 +08:00

86 lines
2.3 KiB
Go
Raw Permalink Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

package ws
// hub_sentinel_test.go — M8-4b驗證 Hub 在第一個 client 連上時寫 sentinel file
//
// 測試對應 .autoflow/04-architecture/v2/startup-pipeline.md §3 階段 6。
import (
"os"
"path/filepath"
"testing"
"time"
)
// 模擬一個 dummy client不需要真的 WebSocket 連線Hub.Run 只關心 *Client pointer
func dummyClient() *Client {
return &Client{Send: make(chan []byte, 1)}
}
func TestHub_StartupSentinel_WrittenOnFirstRegister(t *testing.T) {
dir, err := os.MkdirTemp("", "ws-hub-sentinel-*")
if err != nil {
t.Fatalf("mkdtemp: %v", err)
}
defer os.RemoveAll(dir)
hub := NewHub()
hub.SetStartupSentinel(dir)
go hub.Run()
// 第一個 client 加入任意 room
hub.RegisterSync(&Subscription{Client: dummyClient(), Room: "test-room"})
// sentinel 應該在 dir 下出現
path := filepath.Join(dir, ".first-ws-connected")
if _, err := os.Stat(path); err != nil {
t.Fatalf("sentinel file 應該被寫入:%v", err)
}
}
func TestHub_StartupSentinel_WrittenOnlyOnce(t *testing.T) {
dir, err := os.MkdirTemp("", "ws-hub-sentinel-once-*")
if err != nil {
t.Fatalf("mkdtemp: %v", err)
}
defer os.RemoveAll(dir)
hub := NewHub()
hub.SetStartupSentinel(dir)
go hub.Run()
// 第一個 client
hub.RegisterSync(&Subscription{Client: dummyClient(), Room: "test-room"})
path := filepath.Join(dir, ".first-ws-connected")
info1, err := os.Stat(path)
if err != nil {
t.Fatalf("first sentinel: %v", err)
}
// 等一點時間確保 mtime 會不同(如果有寫第二次)
time.Sleep(50 * time.Millisecond)
// 第二、三個 client 加入
hub.RegisterSync(&Subscription{Client: dummyClient(), Room: "test-room"})
hub.RegisterSync(&Subscription{Client: dummyClient(), Room: "another-room"})
info2, err := os.Stat(path)
if err != nil {
t.Fatalf("second stat: %v", err)
}
if !info1.ModTime().Equal(info2.ModTime()) {
t.Fatalf("sentinel file 不應該被重寫mtime1=%v mtime2=%v", info1.ModTime(), info2.ModTime())
}
}
func TestHub_StartupSentinel_DisabledWhenDataDirEmpty(t *testing.T) {
hub := NewHub()
// 不呼叫 SetStartupSentinel → sentinelDataDir 為空
go hub.Run()
// 加 client 不該 panic、也不該寫任何檔案
hub.RegisterSync(&Subscription{Client: dummyClient(), Room: "test-room"})
// 沒有路徑可以驗證,這個測試主要驗證「不 panic」即可
}