- Increase busy_timeout from 5s to 15s - Set synchronous=NORMAL (safe with WAL, reduces fsync) - Limit MaxOpenConns to 4 to reduce write lock contention - Explicit wal_autocheckpoint=1000 Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
83 lines
1.8 KiB
Go
83 lines
1.8 KiB
Go
package storage
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
)
|
|
|
|
func TestNew(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
dataDir string
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "in-memory database",
|
|
dataDir: ":memory:",
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "empty string creates in-memory",
|
|
dataDir: "",
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "temp directory",
|
|
dataDir: t.TempDir(),
|
|
wantErr: false,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
ctx := context.Background()
|
|
db, err := New(ctx, tt.dataDir)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Fatalf("New() error = %v, wantErr %v", err, tt.wantErr)
|
|
}
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer db.Close()
|
|
|
|
// Verify WAL mode (in-memory uses "memory" journal mode)
|
|
if tt.dataDir != "" && tt.dataDir != ":memory:" {
|
|
var journalMode string
|
|
err = db.QueryRow("PRAGMA journal_mode").Scan(&journalMode)
|
|
if err != nil {
|
|
t.Fatalf("failed to query journal_mode: %v", err)
|
|
}
|
|
if journalMode != "wal" {
|
|
t.Errorf("journal_mode = %s, want wal", journalMode)
|
|
}
|
|
}
|
|
|
|
// Verify foreign keys are enabled
|
|
var fk int
|
|
err = db.QueryRow("PRAGMA foreign_keys").Scan(&fk)
|
|
if err != nil {
|
|
t.Fatalf("failed to query foreign_keys: %v", err)
|
|
}
|
|
if fk != 1 {
|
|
t.Errorf("foreign_keys = %d, want 1", fk)
|
|
}
|
|
|
|
// Verify busy_timeout
|
|
var timeout int
|
|
err = db.QueryRow("PRAGMA busy_timeout").Scan(&timeout)
|
|
if err != nil {
|
|
t.Fatalf("failed to query busy_timeout: %v", err)
|
|
}
|
|
if timeout != 15000 {
|
|
t.Errorf("busy_timeout = %d, want 15000", timeout)
|
|
}
|
|
|
|
// Verify database is usable
|
|
_, err = db.Exec("CREATE TABLE test (id INTEGER PRIMARY KEY)")
|
|
if err != nil {
|
|
t.Fatalf("failed to create test table: %v", err)
|
|
}
|
|
})
|
|
}
|
|
}
|