visionA/local-agent/server/internal/api/ws/hub_sentinel_test.go
jim800121chen 3f0175f1a9 feat(local-agent): Phase 0.5 visionA Agent — Wails 桌面 + tunnel client + 配對 UI
從 local-tool 複製出獨立的「visionA Agent」桌面應用(A3 純橋樑:
tunnel client + 配對 UI + 設定,不開 HTTP port、不做本機裝置/推論 UI)。
Bundle ID 與 local-tool 不同(com.innovedus.visiona-agent vs visiona-local),
雙 app 可共存。fork 後不主動 sync,需要時手動 cherry-pick。

Backend / Wails Go(AB1-AB13):
- internal/tunnel:6 狀態機(Idle/Connecting/Connected/Reconnecting/Failed/Stopped)
  + Pair/Unpair/Reconnect/Disconnect binding + ClientHooks event
- internal/auth:encrypted file token store(AES-GCM + scrypt + machineID
  fallback salt + 13 tests)
- internal/config:YAML validation + atomic write + 11 tests
- internal/log:ring buffer + ExportLog 升級 zip
- visionA-backend /api/pairing/exchange:SessionTokenStore + 17 new tests
- 三平台 build 驗證(macOS DMG 160 MB / Windows EXE / Linux AppImage)
- end-to-end 5 milestone 全綠(pairing → tunnel → forward → reuse 防護
  → tunnel drop failover)

Frontend / Next.js(AF1-AF7,沿用 visionA-frontend 基礎):
- AppShell + Header + TabNav(StatusView / PairView / SettingsView 三 tab)
- ConnectionStatusBadge 5 種狀態
- TokenInput regex 驗證 + 7 種錯誤 + 0.5s auto-switch 到狀態頁
- 設定頁 4 區塊(含重新配對 AlertDialog)
- agent-api.ts 封裝 Wails bindings(mock/real 雙實作)+ 90 tests

Phase 0.7 review-driven fix(Round 2):
- A1 Session fixation 防護(RotateSessionID)
- A3 mock pairing 預設改 false(必須明確 opt-in)+ startup log
- A4 Pair 失敗後 state 清理矩陣(exchange/Save/Start fail 各自終態)
- A5 Pair/Unpair/Reconnect lifecycleMu + 50 goroutine race test
- F1 重新配對次按鈕 / F2 PairView Esc cancel / F3 Wails BrowserOpenURL
  / F4 Settings draft 持久 + 未儲存 badge

驗證:agent backend go test -race -count=3 ./... 4 packages 全綠 /
agent frontend pnpm test 119 tests 全綠

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
2026-05-01 11:22:01 +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」即可
}