steve
7b1990cf11
Agent: new runner.BackupHooks struct + runHook helper invoked via /bin/sh -c (cmd.exe /C on Windows). pre_hook non-zero exit aborts the backup; post_hook always runs with RM_JOB_STATUS=succeeded|failed in env. Output streamed as 'hook(<phase>): …' log.stream lines. Hooks only run for kind=backup (other kinds skip both phases). Server: resolveBackupHooks resolves group → host default → empty, decrypts via crypto.AEAD with per-slot ad bytes, plumbs plaintext into CommandRunPayload for both schedule.fire and per-group Run-now dispatch sites. Decrypt failures degrade silently to no hook so a malformed blob can't poison every backup.
107 lines
2.9 KiB
Go
107 lines
2.9 KiB
Go
// hooks_test.go — covers the pre/post hook columns added in
|
|
// migration 0010 (P2R-10): set + reload roundtrip on both
|
|
// source_groups and hosts; nil clears the column.
|
|
package store
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/oklog/ulid/v2"
|
|
)
|
|
|
|
func newTestStore(t *testing.T) *Store {
|
|
t.Helper()
|
|
dir := t.TempDir()
|
|
st, err := Open(context.Background(), filepath.Join(dir, "rm.db"))
|
|
if err != nil {
|
|
t.Fatalf("open store: %v", err)
|
|
}
|
|
t.Cleanup(func() { _ = st.Close() })
|
|
return st
|
|
}
|
|
|
|
func makeHostInStore(t *testing.T, st *Store, name string) string {
|
|
t.Helper()
|
|
id := ulid.Make().String()
|
|
if err := st.CreateHost(context.Background(), Host{
|
|
ID: id, Name: name, OS: "linux", Arch: "amd64",
|
|
EnrolledAt: time.Now().UTC(),
|
|
}, "tokenhash-"+id, ""); err != nil {
|
|
t.Fatalf("create host: %v", err)
|
|
}
|
|
return id
|
|
}
|
|
|
|
func TestSourceGroupHooksRoundTrip(t *testing.T) {
|
|
t.Parallel()
|
|
st := newTestStore(t)
|
|
hostID := makeHostInStore(t, st, "hooks-host")
|
|
|
|
g := &SourceGroup{
|
|
ID: ulid.Make().String(), HostID: hostID, Name: "etc",
|
|
PreHook: "ENC-PRE",
|
|
PostHook: "ENC-POST",
|
|
}
|
|
if err := st.CreateSourceGroup(context.Background(), g); err != nil {
|
|
t.Fatalf("create: %v", err)
|
|
}
|
|
got, err := st.GetSourceGroup(context.Background(), hostID, g.ID)
|
|
if err != nil {
|
|
t.Fatalf("get: %v", err)
|
|
}
|
|
if got.PreHook != "ENC-PRE" {
|
|
t.Fatalf("PreHook: got %q, want ENC-PRE", got.PreHook)
|
|
}
|
|
if got.PostHook != "ENC-POST" {
|
|
t.Fatalf("PostHook: got %q, want ENC-POST", got.PostHook)
|
|
}
|
|
|
|
// Update: clear PreHook, change PostHook.
|
|
got.PreHook = ""
|
|
got.PostHook = "ENC-POST-2"
|
|
if err := st.UpdateSourceGroup(context.Background(), got); err != nil {
|
|
t.Fatalf("update: %v", err)
|
|
}
|
|
got, err = st.GetSourceGroup(context.Background(), hostID, g.ID)
|
|
if err != nil {
|
|
t.Fatalf("get: %v", err)
|
|
}
|
|
if got.PreHook != "" {
|
|
t.Fatalf("PreHook: want empty after clear, got %q", got.PreHook)
|
|
}
|
|
if got.PostHook != "ENC-POST-2" {
|
|
t.Fatalf("PostHook: got %q, want ENC-POST-2", got.PostHook)
|
|
}
|
|
}
|
|
|
|
func TestHostHookDefaultsRoundTrip(t *testing.T) {
|
|
t.Parallel()
|
|
st := newTestStore(t)
|
|
hostID := makeHostInStore(t, st, "host-hooks-host")
|
|
|
|
if err := st.SetHostHooks(context.Background(), hostID, "PRE", "POST"); err != nil {
|
|
t.Fatalf("set: %v", err)
|
|
}
|
|
h, err := st.GetHost(context.Background(), hostID)
|
|
if err != nil {
|
|
t.Fatalf("get: %v", err)
|
|
}
|
|
if h.PreHookDefault != "PRE" || h.PostHookDefault != "POST" {
|
|
t.Fatalf("after set: pre=%q post=%q", h.PreHookDefault, h.PostHookDefault)
|
|
}
|
|
// Clear by passing empty strings.
|
|
if err := st.SetHostHooks(context.Background(), hostID, "", ""); err != nil {
|
|
t.Fatalf("clear: %v", err)
|
|
}
|
|
h, err = st.GetHost(context.Background(), hostID)
|
|
if err != nil {
|
|
t.Fatalf("get: %v", err)
|
|
}
|
|
if h.PreHookDefault != "" || h.PostHookDefault != "" {
|
|
t.Fatalf("after clear: pre=%q post=%q (want empty)", h.PreHookDefault, h.PostHookDefault)
|
|
}
|
|
}
|