admin
f21fa5b5e8
Commilitia Drop：自托管的跨设备剪贴板同步与点对点文件传输。 - 后端 Go（chi / SQLite WAL / SSE Hub / WebRTC signaling + 状态机 / Relay ring buffer），编译进单个 distroless 镜像（前端 go:embed）。 - 前端 React + TanStack Router + Zustand，自实现 SSE + WebRTC P2P，NAT 受阻时回退服务端中继；聚珍（Juzhen）CJK 综合排版。 - 桌面端 Wails v2（macOS / Windows），瘦客户端复用 web。 - 鉴权 OIDC PKCE（自建 Casdoor 等），refresh_token 信封加密存系统密钥库；iOS Shortcut 用 HS256 scoped token。 架构文档与变更记录见 docs 分支（PROJECT_BRIEF / FRONTEND_DESIGN / CHANGELOG）。 本次为公开发布初始提交：完整开发历史（含部署细节）留存于私有归档，公开仓库自此干净起步。
117 lines
3.0 KiB
Go
117 lines
3.0 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"sort"
|
|
"testing"
|
|
)
|
|
|
|
func TestBootstrapCreatesAllTables(t *testing.T) {
|
|
tmp := filepath.Join(t.TempDir(), "test.db")
|
|
d, err := Open(tmp)
|
|
if err != nil {
|
|
t.Fatalf("open: %v", err)
|
|
}
|
|
defer d.Close()
|
|
|
|
if err := Bootstrap(context.Background(), d); err != nil {
|
|
t.Fatalf("bootstrap: %v", err)
|
|
}
|
|
|
|
want := []string{"clipboard_state", "devices", "shortcut_tokens", "transfer_sessions"}
|
|
rows, err := d.Query("SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%' ORDER BY name")
|
|
if err != nil {
|
|
t.Fatalf("query tables: %v", err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
var got []string
|
|
for rows.Next() {
|
|
var name string
|
|
if err := rows.Scan(&name); err != nil {
|
|
t.Fatalf("scan: %v", err)
|
|
}
|
|
got = append(got, name)
|
|
}
|
|
sort.Strings(got)
|
|
if len(got) != len(want) {
|
|
t.Fatalf("table count: got %v, want %v", got, want)
|
|
}
|
|
for i := range want {
|
|
if got[i] != want[i] {
|
|
t.Errorf("table[%d]: got %q, want %q", i, got[i], want[i])
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestBootstrapIsIdempotent(t *testing.T) {
|
|
tmp := filepath.Join(t.TempDir(), "test.db")
|
|
d, err := Open(tmp)
|
|
if err != nil {
|
|
t.Fatalf("open: %v", err)
|
|
}
|
|
defer d.Close()
|
|
|
|
for i := 0; i < 3; i++ {
|
|
if err := Bootstrap(context.Background(), d); err != nil {
|
|
t.Fatalf("bootstrap iter %d: %v", i, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestOpenEnablesWALMode(t *testing.T) {
|
|
tmp := filepath.Join(t.TempDir(), "test.db")
|
|
d, err := Open(tmp)
|
|
if err != nil {
|
|
t.Fatalf("open: %v", err)
|
|
}
|
|
defer d.Close()
|
|
|
|
var mode string
|
|
if err := d.QueryRow("PRAGMA journal_mode").Scan(&mode); err != nil {
|
|
t.Fatalf("pragma: %v", err)
|
|
}
|
|
if mode != "wal" {
|
|
t.Errorf("journal_mode: got %q, want %q", mode, "wal")
|
|
}
|
|
}
|
|
|
|
// Simulates the prod upgrade: a clipboard_state created before the version
|
|
// column existed must gain it (defaulting to 0) on bootstrap, idempotently.
|
|
func TestBootstrapAddsClipboardVersionToLegacyDB(t *testing.T) {
|
|
tmp := filepath.Join(t.TempDir(), "legacy.db")
|
|
d, err := Open(tmp)
|
|
if err != nil {
|
|
t.Fatalf("open: %v", err)
|
|
}
|
|
defer d.Close()
|
|
|
|
_, err = d.Exec(`CREATE TABLE clipboard_state (
|
|
user_id TEXT PRIMARY KEY, content_type TEXT NOT NULL, content TEXT,
|
|
source_device TEXT, updated_at INTEGER NOT NULL)`)
|
|
if err != nil {
|
|
t.Fatalf("legacy schema: %v", err)
|
|
}
|
|
if _, err := d.Exec(`INSERT INTO clipboard_state (user_id, content_type, content, updated_at)
|
|
VALUES ('u', 'text/plain', 'hi', 100)`); err != nil {
|
|
t.Fatalf("seed row: %v", err)
|
|
}
|
|
|
|
if err := Bootstrap(context.Background(), d); err != nil {
|
|
t.Fatalf("bootstrap: %v", err)
|
|
}
|
|
|
|
var version int64
|
|
if err := d.QueryRow(`SELECT version FROM clipboard_state WHERE user_id='u'`).Scan(&version); err != nil {
|
|
t.Fatalf("select version (column missing?): %v", err)
|
|
}
|
|
if version != 0 {
|
|
t.Errorf("legacy row version: got %d, want 0", version)
|
|
}
|
|
// Second bootstrap must be a no-op (column already present).
|
|
if err := Bootstrap(context.Background(), d); err != nil {
|
|
t.Fatalf("second bootstrap: %v", err)
|
|
}
|
|
}
|