223 lines
6.8 KiB
Go
223 lines
6.8 KiB
Go
package ws
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"gitea.dcglab.co.uk/steve/restic-manager/internal/api"
|
|
"gitea.dcglab.co.uk/steve/restic-manager/internal/store"
|
|
)
|
|
|
|
// openWSTestStore opens an isolated file-backed db in t.TempDir.
|
|
func openWSTestStore(t *testing.T) *store.Store {
|
|
t.Helper()
|
|
dir := t.TempDir()
|
|
s, err := store.Open(context.Background(), filepath.Join(dir, "rm.db"))
|
|
if err != nil {
|
|
t.Fatalf("open: %v", err)
|
|
}
|
|
t.Cleanup(func() { _ = s.Close() })
|
|
return s
|
|
}
|
|
|
|
// seedHostWS inserts a minimal host row directly via the store's DB.
|
|
func seedHostWS(t *testing.T, s *store.Store, hostID string) {
|
|
t.Helper()
|
|
_, err := s.DB().Exec(
|
|
`INSERT INTO hosts (id, name, os, arch, enrolled_at) VALUES (?,?,?,?,?)`,
|
|
hostID, hostID, "linux", "amd64", "2026-01-01T00:00:00Z")
|
|
if err != nil {
|
|
t.Fatalf("seed host %q: %v", hostID, err)
|
|
}
|
|
}
|
|
|
|
func int64ptrWS(v int64) *int64 { return &v }
|
|
func boolptrWS(v bool) *bool { return &v }
|
|
|
|
func TestUnknownCancelResultTerminalisesOrphanedJob(t *testing.T) {
|
|
t.Parallel()
|
|
s := openWSTestStore(t)
|
|
ctx := context.Background()
|
|
const hostID = "h-orphan-ws"
|
|
seedHostWS(t, s, hostID)
|
|
jobID := "j-orphan-ws"
|
|
started := time.Now().UTC().Add(-time.Hour)
|
|
if err := s.CreateJob(ctx, store.Job{ID: jobID, HostID: hostID, Kind: "backup", ActorKind: "user", CreatedAt: started}); err != nil {
|
|
t.Fatalf("create job: %v", err)
|
|
}
|
|
if err := s.MarkJobStarted(ctx, jobID, started); err != nil {
|
|
t.Fatalf("start job: %v", err)
|
|
}
|
|
env, _ := api.Marshal(api.MsgCommandResult, jobID, api.CommandResultPayload{
|
|
JobID: jobID, Accepted: false, Error: "job_not_found",
|
|
})
|
|
dispatchAgentMessage(ctx, nil, hostID, env, HandlerDeps{Store: s})
|
|
job, _ := s.GetJob(ctx, jobID)
|
|
if job.Status != "cancelled" || job.FinishedAt == nil {
|
|
t.Fatalf("orphan not terminalised: %+v", job)
|
|
}
|
|
}
|
|
|
|
func TestUnknownCancelResultCannotTerminaliseAnotherHostsJob(t *testing.T) {
|
|
t.Parallel()
|
|
s := openWSTestStore(t)
|
|
ctx := context.Background()
|
|
seedHostWS(t, s, "reporting-host")
|
|
seedHostWS(t, s, "job-host")
|
|
const jobID = "j-foreign-ws"
|
|
started := time.Now().UTC().Add(-time.Hour)
|
|
if err := s.CreateJob(ctx, store.Job{ID: jobID, HostID: "job-host", Kind: "backup", ActorKind: "user", CreatedAt: started}); err != nil {
|
|
t.Fatalf("create job: %v", err)
|
|
}
|
|
if err := s.MarkJobStarted(ctx, jobID, started); err != nil {
|
|
t.Fatalf("start job: %v", err)
|
|
}
|
|
env, _ := api.Marshal(api.MsgCommandResult, jobID, api.CommandResultPayload{
|
|
JobID: jobID, Accepted: false, Error: "job_not_found",
|
|
})
|
|
dispatchAgentMessage(ctx, nil, "reporting-host", env, HandlerDeps{Store: s})
|
|
job, _ := s.GetJob(ctx, jobID)
|
|
if job.Status != "running" || job.FinishedAt != nil {
|
|
t.Fatalf("foreign job was modified: %+v", job)
|
|
}
|
|
}
|
|
|
|
func TestRepoStatsReportPersisted(t *testing.T) {
|
|
t.Parallel()
|
|
s := openWSTestStore(t)
|
|
ctx := context.Background()
|
|
|
|
const hostID = "h-stats-ws"
|
|
seedHostWS(t, s, hostID)
|
|
|
|
now := time.Now().UTC().Truncate(time.Second)
|
|
pruneAt := now.Add(-2 * time.Hour)
|
|
payload := api.RepoStatsPayload{
|
|
TotalSizeBytes: int64ptrWS(1024),
|
|
RawSizeBytes: int64ptrWS(2048),
|
|
UniqueFiles: int64ptrWS(42),
|
|
SnapshotCount: int64ptrWS(7),
|
|
LastCheckAt: &now,
|
|
LastCheckStatus: "ok",
|
|
LockPresent: boolptrWS(false),
|
|
LastPruneAt: &pruneAt,
|
|
LastPruneFreedBytes: int64ptrWS(512),
|
|
}
|
|
env, err := api.Marshal(api.MsgRepoStats, "", payload)
|
|
if err != nil {
|
|
t.Fatalf("marshal: %v", err)
|
|
}
|
|
|
|
deps := HandlerDeps{Store: s}
|
|
dispatchAgentMessage(ctx, nil, hostID, env, deps)
|
|
|
|
got, err := s.GetHostRepoStats(ctx, hostID)
|
|
if err != nil {
|
|
t.Fatalf("get host repo stats: %v", err)
|
|
}
|
|
if got.TotalSizeBytes == nil || *got.TotalSizeBytes != 1024 {
|
|
t.Errorf("TotalSizeBytes: got %v want 1024", got.TotalSizeBytes)
|
|
}
|
|
if got.RawSizeBytes == nil || *got.RawSizeBytes != 2048 {
|
|
t.Errorf("RawSizeBytes: got %v want 2048", got.RawSizeBytes)
|
|
}
|
|
if got.UniqueFiles == nil || *got.UniqueFiles != 42 {
|
|
t.Errorf("UniqueFiles: got %v want 42", got.UniqueFiles)
|
|
}
|
|
if got.SnapshotCount == nil || *got.SnapshotCount != 7 {
|
|
t.Errorf("SnapshotCount: got %v want 7", got.SnapshotCount)
|
|
}
|
|
if got.LastCheckAt == nil || !got.LastCheckAt.Equal(now) {
|
|
t.Errorf("LastCheckAt: got %v want %v", got.LastCheckAt, now)
|
|
}
|
|
if got.LastCheckStatus != "ok" {
|
|
t.Errorf("LastCheckStatus: got %q want %q", got.LastCheckStatus, "ok")
|
|
}
|
|
if got.LockPresent == nil || *got.LockPresent != false {
|
|
t.Errorf("LockPresent: got %v want false", got.LockPresent)
|
|
}
|
|
if got.LastPruneAt == nil || !got.LastPruneAt.Equal(pruneAt) {
|
|
t.Errorf("LastPruneAt: got %v want %v", got.LastPruneAt, pruneAt)
|
|
}
|
|
if got.LastPruneFreedBytes == nil || *got.LastPruneFreedBytes != 512 {
|
|
t.Errorf("LastPruneFreedBytes: got %v want 512", got.LastPruneFreedBytes)
|
|
}
|
|
}
|
|
|
|
func TestRepoStatsReportPartialUpdate(t *testing.T) {
|
|
t.Parallel()
|
|
s := openWSTestStore(t)
|
|
ctx := context.Background()
|
|
|
|
const hostID = "h-stats-partial"
|
|
seedHostWS(t, s, hostID)
|
|
|
|
// Pre-seed: TotalSizeBytes = 100.
|
|
if err := s.UpsertHostRepoStats(ctx, hostID, store.HostRepoStats{
|
|
TotalSizeBytes: int64ptrWS(100),
|
|
}); err != nil {
|
|
t.Fatalf("pre-seed upsert: %v", err)
|
|
}
|
|
|
|
// Send a repo.stats payload that only sets LastCheckStatus.
|
|
env, err := api.Marshal(api.MsgRepoStats, "", api.RepoStatsPayload{
|
|
LastCheckStatus: "ok",
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("marshal: %v", err)
|
|
}
|
|
dispatchAgentMessage(ctx, nil, hostID, env, HandlerDeps{Store: s})
|
|
|
|
got, err := s.GetHostRepoStats(ctx, hostID)
|
|
if err != nil {
|
|
t.Fatalf("get: %v", err)
|
|
}
|
|
if got.TotalSizeBytes == nil || *got.TotalSizeBytes != 100 {
|
|
t.Errorf("TotalSizeBytes lost: got %v want 100", got.TotalSizeBytes)
|
|
}
|
|
if got.LastCheckStatus != "ok" {
|
|
t.Errorf("LastCheckStatus: got %q want ok", got.LastCheckStatus)
|
|
}
|
|
}
|
|
|
|
func TestRepoStatsReportWritesHistoryRow(t *testing.T) {
|
|
t.Parallel()
|
|
s := openWSTestStore(t)
|
|
ctx := context.Background()
|
|
|
|
const hostID = "h-stats-history"
|
|
seedHostWS(t, s, hostID)
|
|
|
|
payload := api.RepoStatsPayload{
|
|
TotalSizeBytes: int64ptrWS(12345),
|
|
SnapshotCount: int64ptrWS(7),
|
|
}
|
|
env, err := api.Marshal(api.MsgRepoStats, "", payload)
|
|
if err != nil {
|
|
t.Fatalf("marshal: %v", err)
|
|
}
|
|
|
|
deps := HandlerDeps{Store: s}
|
|
dispatchAgentMessage(ctx, nil, hostID, env, deps)
|
|
|
|
pts, err := s.ListHostRepoStatsHistory(ctx, hostID, time.Time{})
|
|
if err != nil {
|
|
t.Fatalf("list history: %v", err)
|
|
}
|
|
if len(pts) != 1 {
|
|
t.Fatalf("want 1 history row, got %d", len(pts))
|
|
}
|
|
wantDay := time.Now().UTC().Format("2006-01-02")
|
|
if got := pts[0].Day.Format("2006-01-02"); got != wantDay {
|
|
t.Errorf("day: want %s, got %s", wantDay, got)
|
|
}
|
|
if pts[0].TotalSizeBytes == nil || *pts[0].TotalSizeBytes != 12345 {
|
|
t.Errorf("TotalSizeBytes: want 12345, got %v", pts[0].TotalSizeBytes)
|
|
}
|
|
if pts[0].SnapshotCount == nil || *pts[0].SnapshotCount != 7 {
|
|
t.Errorf("SnapshotCount: want 7, got %v", pts[0].SnapshotCount)
|
|
}
|
|
}
|