ananke/internal/cluster/orchestrator_test.go

package cluster

import (
	"log"
	"os"
	"reflect"
	"testing"
	"time"

	"scm.bstein.dev/bstein/hecate/internal/config"
	"scm.bstein.dev/bstein/hecate/internal/state"
)

func TestParseVaultSealed(t *testing.T) {
	sealed, err := parseVaultSealed(`{"initialized":true,"sealed":true}`)
	if err != nil {
		t.Fatalf("parse sealed=true: %v", err)
	}
	if !sealed {
		t.Fatalf("expected sealed=true")
	}

	sealed, err = parseVaultSealed(`{"initialized":true,"sealed":false}`)
	if err != nil {
		t.Fatalf("parse sealed=false: %v", err)
	}
	if sealed {
		t.Fatalf("expected sealed=false")
	}
}

func TestParseVaultSealedRejectsEmpty(t *testing.T) {
	if _, err := parseVaultSealed("   "); err == nil {
		t.Fatalf("expected parse error for empty status payload")
	}
}

func TestParseVaultSealedWithKubectlPreamble(t *testing.T) {
	raw := "Defaulted container \"vault\" out of: vault, setup-config (init)\n{\"sealed\":true,\"initialized\":true}\n"
	sealed, err := parseVaultSealed(raw)
	if err != nil {
		t.Fatalf("parse with preamble: %v", err)
	}
	if !sealed {
		t.Fatalf("expected sealed=true from payload with preamble")
	}
}

func TestFallbackWorkersFromInventoryUsesManagedNodes(t *testing.T) {
	orch := &Orchestrator{
		cfg: config.Config{
			ControlPlanes: []string{"titan-0a", "titan-0b", "titan-0c"},
			SSHManagedNodes: []string{
				"titan-db",
				"titan-0a",
				"titan-15",
				"titan-17",
			},
		},
		log: log.New(os.Stdout, "", 0),
	}
	got := orch.fallbackWorkersFromInventory()
	want := []string{"titan-15", "titan-17", "titan-db"}
	if !reflect.DeepEqual(got, want) {
		t.Fatalf("fallback workers mismatch: got=%v want=%v", got, want)
	}
}

func TestFallbackWorkersFromInventoryFallsBackToHosts(t *testing.T) {
	orch := &Orchestrator{
		cfg: config.Config{
			ControlPlanes: []string{"titan-0a", "titan-0b", "titan-0c"},
			SSHNodeHosts: map[string]string{
				"titan-0a": "192.168.22.11",
				"titan-22": "192.168.22.22",
				"titan-24": "192.168.22.26",
			},
		},
		log: log.New(os.Stdout, "", 0),
	}
	got := orch.fallbackWorkersFromInventory()
	want := []string{"titan-22", "titan-24"}
	if !reflect.DeepEqual(got, want) {
		t.Fatalf("fallback workers mismatch: got=%v want=%v", got, want)
	}
}

func TestIntentFreshTreatsZeroTimestampAsFresh(t *testing.T) {
	if !intentFresh(state.Intent{}, 30*time.Second) {
		t.Fatalf("zero updated_at intent should be treated as fresh")
	}
}

func TestIntentFreshRespectsAge(t *testing.T) {
	stale := state.Intent{UpdatedAt: time.Now().Add(-2 * time.Minute)}
	fresh := state.Intent{UpdatedAt: time.Now().Add(-20 * time.Second)}
	if intentFresh(stale, 30*time.Second) {
		t.Fatalf("expected stale intent to be considered not fresh")
	}
	if !intentFresh(fresh, 30*time.Second) {
		t.Fatalf("expected recent intent to be considered fresh")
	}
}

func TestCoordinationPeersDedupesAndIncludesForwardHost(t *testing.T) {
	orch := &Orchestrator{
		cfg: config.Config{
			Coordination: config.Coordination{
				PeerHosts:           []string{"titan-24", "titan-db", "titan-24", " "},
				ForwardShutdownHost: "titan-db",
			},
		},
	}
	got := orch.coordinationPeers()
	want := []string{"titan-24", "titan-db"}
	if !reflect.DeepEqual(got, want) {
		t.Fatalf("coordination peers mismatch: got=%v want=%v", got, want)
	}
}