admin
ec9b997151
gates / gates (push) Successful in 1m0s
Co-Authored-By: Claude Opus 5.5 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_0159rPz1ZhFKsS53msqPYxtS
76 lines
3.0 KiB
Go
76 lines
3.0 KiB
Go
package agentapi
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
// R-921: the pre-check reads the agent's real wire shapes — GET /backup/tiers, then GET /backup/status per
|
|
// target — and reports exactly the tiers whose job is running or snapshotted (the agent's in-flight set).
|
|
func TestBackupJobsInFlight_ReportsRunningAndSnapshottedTiers(t *testing.T) {
|
|
phases := map[string]string{"local": "done", "felhom-pbs": "snapshotted", "extra": "running"}
|
|
mux := http.NewServeMux()
|
|
mux.HandleFunc("GET /backup/tiers", func(w http.ResponseWriter, r *http.Request) {
|
|
_, _ = w.Write([]byte(`{"ok":true,"data":{"vmid":9201,"tiers":[` +
|
|
`{"target":"local","cadence_seconds":86400,"primary":true,"storage":"present"},` +
|
|
`{"target":"felhom-pbs","cadence_seconds":604800,"primary":false,"storage":"present"},` +
|
|
`{"target":"extra","cadence_seconds":604800,"primary":false,"storage":"present"}]}}`))
|
|
})
|
|
mux.HandleFunc("GET /backup/status", func(w http.ResponseWriter, r *http.Request) {
|
|
tg := r.URL.Query().Get("target")
|
|
ph, ok := phases[tg]
|
|
if !ok {
|
|
http.Error(w, "unexpected target "+tg, http.StatusBadRequest)
|
|
return
|
|
}
|
|
_, _ = w.Write([]byte(`{"ok":true,"data":{"vmid":9201,"phase":"` + ph + `","target":"` + tg + `"}}`))
|
|
})
|
|
s := httptest.NewTLSServer(mux)
|
|
defer s.Close()
|
|
c := clientFor(t, s, strings.TrimPrefix(s.URL, "https://"))
|
|
|
|
got, err := c.BackupJobsInFlight(context.Background())
|
|
if err != nil {
|
|
t.Fatalf("BackupJobsInFlight: %v", err)
|
|
}
|
|
if want := []string{"felhom-pbs", "extra"}; !reflect.DeepEqual(got, want) {
|
|
t.Fatalf("in-flight tiers = %v, want %v (done must not count; running and snapshotted must)", got, want)
|
|
}
|
|
}
|
|
|
|
// A pre-R-82 agent (no /backup/tiers → 404) has no per-tier jobs: nil, nil — never an error that would be
|
|
// read as "cannot tell", and never a busy verdict.
|
|
func TestBackupJobsInFlight_PreR82AgentIsNotBusy(t *testing.T) {
|
|
mux := http.NewServeMux()
|
|
s := httptest.NewTLSServer(mux) // every route 404s
|
|
defer s.Close()
|
|
c := clientFor(t, s, strings.TrimPrefix(s.URL, "https://"))
|
|
|
|
got, err := c.BackupJobsInFlight(context.Background())
|
|
if err != nil || got != nil {
|
|
t.Fatalf("pre-R-82 agent: got %v, %v — want nil, nil", got, err)
|
|
}
|
|
}
|
|
|
|
// A status read that fails is returned, so the loop can fail toward backing up and say so.
|
|
func TestBackupJobsInFlight_StatusErrorIsReturned(t *testing.T) {
|
|
mux := http.NewServeMux()
|
|
mux.HandleFunc("GET /backup/tiers", func(w http.ResponseWriter, r *http.Request) {
|
|
_, _ = w.Write([]byte(`{"ok":true,"data":{"vmid":9201,"tiers":[{"target":"local","primary":true}]}}`))
|
|
})
|
|
mux.HandleFunc("GET /backup/status", func(w http.ResponseWriter, r *http.Request) {
|
|
http.Error(w, "boom", http.StatusInternalServerError)
|
|
})
|
|
s := httptest.NewTLSServer(mux)
|
|
defer s.Close()
|
|
c := clientFor(t, s, strings.TrimPrefix(s.URL, "https://"))
|
|
|
|
if _, err := c.BackupJobsInFlight(context.Background()); err == nil {
|
|
t.Fatal("a failed status read was swallowed — the caller could not tell it from 'nothing in flight'")
|
|
}
|
|
}
|