feat: implement NATS-based NixOS deployment system
Implement the complete homelab-deploy system with three operational modes: - Listener mode: Runs on NixOS hosts as a systemd service, subscribes to NATS subjects with configurable templates, executes nixos-rebuild on deployment requests with concurrency control - MCP mode: MCP server exposing deploy, deploy_admin, and list_hosts tools for AI assistants with tiered access control - CLI mode: Manual deployment commands with subject alias support via environment variables Key components: - internal/messages: Request/response types with validation - internal/nats: Client wrapper with NKey authentication - internal/deploy: Executor with timeout and lock for concurrency - internal/listener: Subject template expansion and request handling - internal/cli: Deploy logic with alias resolution - internal/mcp: MCP server with mcp-go integration - nixos/module.nix: NixOS module with hardened systemd service Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
98
internal/deploy/lock_test.go
Normal file
98
internal/deploy/lock_test.go
Normal file
@@ -0,0 +1,98 @@
|
||||
package deploy
|
||||
|
||||
import (
|
||||
"sync"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestLock_TryAcquire(t *testing.T) {
|
||||
l := NewLock()
|
||||
|
||||
// First acquire should succeed
|
||||
if !l.TryAcquire("request-1") {
|
||||
t.Error("first TryAcquire should succeed")
|
||||
}
|
||||
|
||||
// Second acquire should fail
|
||||
if l.TryAcquire("request-2") {
|
||||
t.Error("second TryAcquire should fail while lock is held")
|
||||
}
|
||||
|
||||
// Verify holder
|
||||
if got := l.Holder(); got != "request-1" {
|
||||
t.Errorf("Holder() = %q, want %q", got, "request-1")
|
||||
}
|
||||
|
||||
// Release and try again
|
||||
l.Release()
|
||||
|
||||
if !l.TryAcquire("request-3") {
|
||||
t.Error("TryAcquire should succeed after Release")
|
||||
}
|
||||
|
||||
if got := l.Holder(); got != "request-3" {
|
||||
t.Errorf("Holder() = %q, want %q", got, "request-3")
|
||||
}
|
||||
}
|
||||
|
||||
func TestLock_IsHeld(t *testing.T) {
|
||||
l := NewLock()
|
||||
|
||||
if l.IsHeld() {
|
||||
t.Error("new lock should not be held")
|
||||
}
|
||||
|
||||
l.TryAcquire("test")
|
||||
|
||||
if !l.IsHeld() {
|
||||
t.Error("lock should be held after TryAcquire")
|
||||
}
|
||||
|
||||
l.Release()
|
||||
|
||||
if l.IsHeld() {
|
||||
t.Error("lock should not be held after Release")
|
||||
}
|
||||
}
|
||||
|
||||
func TestLock_Concurrent(t *testing.T) {
|
||||
l := NewLock()
|
||||
var wg sync.WaitGroup
|
||||
acquired := make(chan string, 100)
|
||||
|
||||
// Try to acquire from multiple goroutines
|
||||
for i := range 100 {
|
||||
wg.Add(1)
|
||||
go func(id int) {
|
||||
defer wg.Done()
|
||||
holder := string(rune('A' + (id % 26)))
|
||||
if l.TryAcquire(holder) {
|
||||
acquired <- holder
|
||||
}
|
||||
}(i)
|
||||
}
|
||||
|
||||
wg.Wait()
|
||||
close(acquired)
|
||||
|
||||
// Only one should have succeeded
|
||||
count := 0
|
||||
for range acquired {
|
||||
count++
|
||||
}
|
||||
|
||||
if count != 1 {
|
||||
t.Errorf("expected exactly 1 successful acquire, got %d", count)
|
||||
}
|
||||
}
|
||||
|
||||
func TestLock_ReleaseUnheld(t *testing.T) {
|
||||
l := NewLock()
|
||||
|
||||
// Releasing an unheld lock should not panic
|
||||
l.Release()
|
||||
|
||||
if l.IsHeld() {
|
||||
t.Error("lock should not be held after Release on unheld lock")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user