Add test coverage for daemon lifecycle and config modules
- Add config_test.go: tests for daemon configuration (local/global, sync behavior) - Add daemon_test.go: tests for daemon lifecycle (creation, shutdown, database path resolution) - Add process_test.go: tests for daemon lock acquisition and release - Closes bd-2b34.6, bd-2b34.7 Amp-Thread-ID: https://ampcode.com/threads/T-4419d1ab-4105-4e75-bea8-1837ee80e2c2 Co-authored-by: Amp <amp@ampcode.com>
This commit is contained in:
98
internal/daemonrunner/config_test.go
Normal file
98
internal/daemonrunner/config_test.go
Normal file
@@ -0,0 +1,98 @@
|
||||
package daemonrunner
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestConfigDefaults(t *testing.T) {
|
||||
cfg := Config{
|
||||
Interval: 5 * time.Second,
|
||||
AutoCommit: true,
|
||||
AutoPush: false,
|
||||
Global: false,
|
||||
}
|
||||
|
||||
if cfg.Interval != 5*time.Second {
|
||||
t.Errorf("Expected Interval 5s, got %v", cfg.Interval)
|
||||
}
|
||||
if !cfg.AutoCommit {
|
||||
t.Error("Expected AutoCommit to be true")
|
||||
}
|
||||
if cfg.AutoPush {
|
||||
t.Error("Expected AutoPush to be false")
|
||||
}
|
||||
if cfg.Global {
|
||||
t.Error("Expected Global to be false")
|
||||
}
|
||||
}
|
||||
|
||||
func TestConfigLocalDaemon(t *testing.T) {
|
||||
cfg := Config{
|
||||
Global: false,
|
||||
WorkspacePath: "/tmp/test-workspace",
|
||||
BeadsDir: "/tmp/test-workspace/.beads",
|
||||
DBPath: "/tmp/test-workspace/.beads/beads.db",
|
||||
SocketPath: "/tmp/test-workspace/.beads/bd.sock",
|
||||
LogFile: "/tmp/test-workspace/.beads/daemon.log",
|
||||
PIDFile: "/tmp/test-workspace/.beads/daemon.pid",
|
||||
}
|
||||
|
||||
if cfg.Global {
|
||||
t.Error("Expected local daemon (Global=false)")
|
||||
}
|
||||
if cfg.WorkspacePath == "" {
|
||||
t.Error("Expected WorkspacePath to be set for local daemon")
|
||||
}
|
||||
if cfg.DBPath == "" {
|
||||
t.Error("Expected DBPath to be set for local daemon")
|
||||
}
|
||||
}
|
||||
|
||||
func TestConfigGlobalDaemon(t *testing.T) {
|
||||
cfg := Config{
|
||||
Global: true,
|
||||
BeadsDir: "/home/user/.beads",
|
||||
SocketPath: "/home/user/.beads/global.sock",
|
||||
LogFile: "/home/user/.beads/global-daemon.log",
|
||||
PIDFile: "/home/user/.beads/global-daemon.pid",
|
||||
}
|
||||
|
||||
if !cfg.Global {
|
||||
t.Error("Expected global daemon (Global=true)")
|
||||
}
|
||||
if cfg.WorkspacePath != "" {
|
||||
t.Error("Expected WorkspacePath to be empty for global daemon")
|
||||
}
|
||||
if cfg.DBPath != "" {
|
||||
t.Error("Expected DBPath to be empty for global daemon")
|
||||
}
|
||||
}
|
||||
|
||||
func TestConfigSyncBehavior(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
autoCommit bool
|
||||
autoPush bool
|
||||
}{
|
||||
{"no sync", false, false},
|
||||
{"commit only", true, false},
|
||||
{"commit and push", true, true},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
cfg := Config{
|
||||
AutoCommit: tt.autoCommit,
|
||||
AutoPush: tt.autoPush,
|
||||
}
|
||||
|
||||
if cfg.AutoCommit != tt.autoCommit {
|
||||
t.Errorf("Expected AutoCommit=%v, got %v", tt.autoCommit, cfg.AutoCommit)
|
||||
}
|
||||
if cfg.AutoPush != tt.autoPush {
|
||||
t.Errorf("Expected AutoPush=%v, got %v", tt.autoPush, cfg.AutoPush)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
129
internal/daemonrunner/daemon_test.go
Normal file
129
internal/daemonrunner/daemon_test.go
Normal file
@@ -0,0 +1,129 @@
|
||||
package daemonrunner
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestNew(t *testing.T) {
|
||||
cfg := Config{
|
||||
Interval: 5 * time.Second,
|
||||
AutoCommit: true,
|
||||
Global: false,
|
||||
}
|
||||
|
||||
daemon := New(cfg, "0.19.0")
|
||||
|
||||
if daemon == nil {
|
||||
t.Fatal("Expected non-nil daemon")
|
||||
}
|
||||
if daemon.cfg.Interval != cfg.Interval {
|
||||
t.Errorf("Expected interval %v, got %v", cfg.Interval, daemon.cfg.Interval)
|
||||
}
|
||||
if daemon.Version != "0.19.0" {
|
||||
t.Errorf("Expected version 0.19.0, got %s", daemon.Version)
|
||||
}
|
||||
}
|
||||
|
||||
func TestStop(t *testing.T) {
|
||||
cfg := Config{
|
||||
Interval: 5 * time.Second,
|
||||
}
|
||||
daemon := New(cfg, "0.19.0")
|
||||
|
||||
// Stop should not error even with no server running
|
||||
if err := daemon.Stop(); err != nil {
|
||||
t.Errorf("Stop() returned unexpected error: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestDetermineDatabasePath(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
beadsDir := filepath.Join(tmpDir, ".beads")
|
||||
dbPath := filepath.Join(beadsDir, "beads.db")
|
||||
|
||||
if err := os.MkdirAll(beadsDir, 0755); err != nil {
|
||||
t.Fatalf("Failed to create beads dir: %v", err)
|
||||
}
|
||||
if err := os.WriteFile(dbPath, []byte("test"), 0644); err != nil {
|
||||
t.Fatalf("Failed to create db file: %v", err)
|
||||
}
|
||||
|
||||
cfg := Config{
|
||||
WorkspacePath: tmpDir,
|
||||
}
|
||||
daemon := New(cfg, "0.19.0")
|
||||
|
||||
// Override working directory for test
|
||||
oldWd, _ := os.Getwd()
|
||||
defer func() { _ = os.Chdir(oldWd) }()
|
||||
if err := os.Chdir(tmpDir); err != nil {
|
||||
t.Fatalf("Failed to change directory: %v", err)
|
||||
}
|
||||
|
||||
if err := daemon.determineDatabasePath(); err != nil {
|
||||
t.Errorf("determineDatabasePath() failed: %v", err)
|
||||
}
|
||||
|
||||
// Use EvalSymlinks to handle /var vs /private/var on macOS
|
||||
expectedDB, _ := filepath.EvalSymlinks(dbPath)
|
||||
actualDB, _ := filepath.EvalSymlinks(daemon.cfg.DBPath)
|
||||
if actualDB != expectedDB {
|
||||
t.Errorf("Expected DBPath %s, got %s", expectedDB, actualDB)
|
||||
}
|
||||
|
||||
expectedBeadsDir, _ := filepath.EvalSymlinks(beadsDir)
|
||||
actualBeadsDir, _ := filepath.EvalSymlinks(daemon.cfg.BeadsDir)
|
||||
if actualBeadsDir != expectedBeadsDir {
|
||||
t.Errorf("Expected BeadsDir %s, got %s", expectedBeadsDir, actualBeadsDir)
|
||||
}
|
||||
|
||||
expectedWS, _ := filepath.EvalSymlinks(tmpDir)
|
||||
actualWS, _ := filepath.EvalSymlinks(daemon.cfg.WorkspacePath)
|
||||
if actualWS != expectedWS {
|
||||
t.Errorf("Expected WorkspacePath %s, got %s", expectedWS, actualWS)
|
||||
}
|
||||
}
|
||||
|
||||
func TestDetermineDatabasePathAlreadySet(t *testing.T) {
|
||||
existingPath := "/already/set/beads.db"
|
||||
cfg := Config{
|
||||
DBPath: existingPath,
|
||||
}
|
||||
daemon := New(cfg, "0.19.0")
|
||||
|
||||
if err := daemon.determineDatabasePath(); err != nil {
|
||||
t.Errorf("determineDatabasePath() failed: %v", err)
|
||||
}
|
||||
|
||||
if daemon.cfg.DBPath != existingPath {
|
||||
t.Errorf("Expected DBPath unchanged: %s, got %s", existingPath, daemon.cfg.DBPath)
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetGlobalBeadsDir(t *testing.T) {
|
||||
beadsDir, err := getGlobalBeadsDir()
|
||||
if err != nil {
|
||||
t.Fatalf("getGlobalBeadsDir() failed: %v", err)
|
||||
}
|
||||
|
||||
if beadsDir == "" {
|
||||
t.Error("Expected non-empty beads directory")
|
||||
}
|
||||
|
||||
// Check directory was created
|
||||
if stat, err := os.Stat(beadsDir); err != nil {
|
||||
t.Errorf("Global beads directory not created: %v", err)
|
||||
} else if !stat.IsDir() {
|
||||
t.Error("Global beads path is not a directory")
|
||||
}
|
||||
|
||||
// Verify it's in home directory
|
||||
home, _ := os.UserHomeDir()
|
||||
expectedPath := filepath.Join(home, ".beads")
|
||||
if beadsDir != expectedPath {
|
||||
t.Errorf("Expected %s, got %s", expectedPath, beadsDir)
|
||||
}
|
||||
}
|
||||
118
internal/daemonrunner/process_test.go
Normal file
118
internal/daemonrunner/process_test.go
Normal file
@@ -0,0 +1,118 @@
|
||||
package daemonrunner
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestDaemonLockBasics(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
dbPath := filepath.Join(tmpDir, "beads.db")
|
||||
|
||||
// Acquire lock
|
||||
lock, err := acquireDaemonLock(tmpDir, dbPath, "0.19.0")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to acquire lock: %v", err)
|
||||
}
|
||||
defer lock.Close()
|
||||
|
||||
// Verify lock file was created
|
||||
lockPath := filepath.Join(tmpDir, "daemon.lock")
|
||||
if _, err := os.Stat(lockPath); os.IsNotExist(err) {
|
||||
t.Error("Lock file was not created")
|
||||
}
|
||||
|
||||
// Verify PID file was created
|
||||
pidPath := filepath.Join(tmpDir, "daemon.pid")
|
||||
if _, err := os.Stat(pidPath); os.IsNotExist(err) {
|
||||
t.Error("PID file was not created")
|
||||
}
|
||||
|
||||
// Read and verify lock metadata
|
||||
data, err := os.ReadFile(lockPath)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to read lock file: %v", err)
|
||||
}
|
||||
|
||||
var info DaemonLockInfo
|
||||
if err := json.Unmarshal(data, &info); err != nil {
|
||||
t.Fatalf("Failed to parse lock file: %v", err)
|
||||
}
|
||||
|
||||
if info.PID != os.Getpid() {
|
||||
t.Errorf("Expected PID %d, got %d", os.Getpid(), info.PID)
|
||||
}
|
||||
if info.Database != dbPath {
|
||||
t.Errorf("Expected database %s, got %s", dbPath, info.Database)
|
||||
}
|
||||
if info.Version != "0.19.0" {
|
||||
t.Errorf("Expected version 0.19.0, got %s", info.Version)
|
||||
}
|
||||
if info.StartedAt.IsZero() {
|
||||
t.Error("Expected non-zero StartedAt timestamp")
|
||||
}
|
||||
}
|
||||
|
||||
func TestDaemonLockExclusive(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
dbPath := filepath.Join(tmpDir, "beads.db")
|
||||
|
||||
// Acquire first lock
|
||||
lock1, err := acquireDaemonLock(tmpDir, dbPath, "0.19.0")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to acquire first lock: %v", err)
|
||||
}
|
||||
defer lock1.Close()
|
||||
|
||||
// Try to acquire second lock (should fail)
|
||||
lock2, err := acquireDaemonLock(tmpDir, dbPath, "0.19.0")
|
||||
if err != ErrDaemonLocked {
|
||||
if lock2 != nil {
|
||||
lock2.Close()
|
||||
}
|
||||
t.Errorf("Expected ErrDaemonLocked, got %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestDaemonLockRelease(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
dbPath := filepath.Join(tmpDir, "beads.db")
|
||||
|
||||
// Acquire lock
|
||||
lock, err := acquireDaemonLock(tmpDir, dbPath, "0.19.0")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to acquire lock: %v", err)
|
||||
}
|
||||
|
||||
// Release lock
|
||||
if err := lock.Close(); err != nil {
|
||||
t.Fatalf("Failed to release lock: %v", err)
|
||||
}
|
||||
|
||||
// Should be able to acquire again after release
|
||||
lock2, err := acquireDaemonLock(tmpDir, dbPath, "0.19.0")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to acquire lock after release: %v", err)
|
||||
}
|
||||
defer lock2.Close()
|
||||
}
|
||||
|
||||
func TestDaemonLockCloseIdempotent(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
dbPath := filepath.Join(tmpDir, "beads.db")
|
||||
|
||||
lock, err := acquireDaemonLock(tmpDir, dbPath, "0.19.0")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to acquire lock: %v", err)
|
||||
}
|
||||
|
||||
// Close multiple times should not error
|
||||
if err := lock.Close(); err != nil {
|
||||
t.Errorf("First close failed: %v", err)
|
||||
}
|
||||
if err := lock.Close(); err != nil {
|
||||
t.Errorf("Second close failed: %v", err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user