116 lines
3.1 KiB
Go
116 lines
3.1 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestLatestJobByKind(t *testing.T) {
|
|
t.Parallel()
|
|
s := openTestStore(t)
|
|
ctx := context.Background()
|
|
hostID := makeSchedHost(t, s)
|
|
|
|
// No jobs yet → ErrNotFound.
|
|
if _, err := s.LatestJobByKind(ctx, hostID, "forget"); !errors.Is(err, ErrNotFound) {
|
|
t.Fatalf("expected ErrNotFound on empty, got %v", err)
|
|
}
|
|
|
|
// Insert two finished jobs of kind=forget; the newer one should win.
|
|
older := time.Now().UTC().Add(-2 * time.Hour)
|
|
newer := time.Now().UTC().Add(-1 * time.Hour)
|
|
|
|
if err := s.CreateJob(ctx, Job{
|
|
ID: "j-old", HostID: hostID, Kind: "forget",
|
|
ActorKind: "system", CreatedAt: older,
|
|
}); err != nil {
|
|
t.Fatalf("create older: %v", err)
|
|
}
|
|
if err := s.MarkJobFinished(ctx, "j-old", "succeeded", 0, nil, "", older.Add(time.Minute)); err != nil {
|
|
t.Fatalf("finish older: %v", err)
|
|
}
|
|
if err := s.CreateJob(ctx, Job{
|
|
ID: "j-new", HostID: hostID, Kind: "forget",
|
|
ActorKind: "system", CreatedAt: newer,
|
|
}); err != nil {
|
|
t.Fatalf("create newer: %v", err)
|
|
}
|
|
if err := s.MarkJobFinished(ctx, "j-new", "failed", 1, nil, "boom", newer.Add(time.Minute)); err != nil {
|
|
t.Fatalf("finish newer: %v", err)
|
|
}
|
|
|
|
got, err := s.LatestJobByKind(ctx, hostID, "forget")
|
|
if err != nil {
|
|
t.Fatalf("LatestJobByKind: %v", err)
|
|
}
|
|
if got.ID != "j-new" {
|
|
t.Errorf("want j-new, got %q", got.ID)
|
|
}
|
|
|
|
// A queued job should be ignored — terminal-status filter.
|
|
queuedAt := time.Now().UTC()
|
|
if err := s.CreateJob(ctx, Job{
|
|
ID: "j-queued", HostID: hostID, Kind: "forget",
|
|
ActorKind: "system", CreatedAt: queuedAt,
|
|
}); err != nil {
|
|
t.Fatalf("create queued: %v", err)
|
|
}
|
|
got2, err := s.LatestJobByKind(ctx, hostID, "forget")
|
|
if err != nil {
|
|
t.Fatalf("LatestJobByKind 2: %v", err)
|
|
}
|
|
if got2.ID != "j-new" {
|
|
t.Errorf("queued job should be ignored; want j-new, got %q", got2.ID)
|
|
}
|
|
|
|
// Different kind → ErrNotFound.
|
|
if _, err := s.LatestJobByKind(ctx, hostID, "prune"); !errors.Is(err, ErrNotFound) {
|
|
t.Fatalf("expected ErrNotFound for prune, got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestListAllMaintenance(t *testing.T) {
|
|
t.Parallel()
|
|
s := openTestStore(t)
|
|
ctx := context.Background()
|
|
|
|
// Empty case.
|
|
rows, err := s.ListAllMaintenance(ctx)
|
|
if err != nil {
|
|
t.Fatalf("empty list: %v", err)
|
|
}
|
|
if len(rows) != 0 {
|
|
t.Errorf("want empty, got %+v", rows)
|
|
}
|
|
|
|
// Seed two hosts with maintenance rows.
|
|
h1 := "01HMAINTHOST00000000000A1"
|
|
h2 := "01HMAINTHOST00000000000A2"
|
|
for i, id := range []string{h1, h2} {
|
|
if err := s.CreateHost(ctx, Host{
|
|
ID: id, Name: "maint-host-" + string(rune('a'+i)),
|
|
OS: "linux", Arch: "amd64",
|
|
AgentVersion: "dev", ResticVersion: "0.16.0", ProtocolVersion: 1,
|
|
EnrolledAt: time.Now().UTC(),
|
|
}, "th-"+id, ""); err != nil {
|
|
t.Fatalf("create host %s: %v", id, err)
|
|
}
|
|
}
|
|
if err := s.CreateDefaultRepoMaintenance(ctx, h1); err != nil {
|
|
t.Fatalf("seed h1: %v", err)
|
|
}
|
|
if err := s.CreateDefaultRepoMaintenance(ctx, h2); err != nil {
|
|
t.Fatalf("seed h2: %v", err)
|
|
}
|
|
|
|
rows, err = s.ListAllMaintenance(ctx)
|
|
if err != nil {
|
|
t.Fatalf("list: %v", err)
|
|
}
|
|
if len(rows) != 2 {
|
|
t.Errorf("want 2 rows, got %d", len(rows))
|
|
}
|
|
}
|