80 lines
2.4 KiB
Go
80 lines
2.4 KiB
Go
package store
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestNodeMaintenance(t *testing.T) {
|
|
s, clock := openTest(t)
|
|
mustTenant(t, s, "t1")
|
|
mustNode(t, s, "n1", "t1", "k1")
|
|
|
|
if n, _ := s.GetNode(ctx, "n1"); n.Maintenance != nil {
|
|
t.Fatalf("neuer Knoten mit Wartung: %+v", n.Maintenance)
|
|
}
|
|
until := clock.now().Add(4 * time.Hour)
|
|
err := s.SetNodeMaintenance(ctx, "n1", Maintenance{
|
|
Until: until, Reason: "Kernel-Update", PublicReason: "Wir spielen ein Sicherheitsupdate ein.",
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("SetNodeMaintenance: %v", err)
|
|
}
|
|
n, _ := s.GetNode(ctx, "n1")
|
|
m := n.Maintenance
|
|
if m == nil || !m.Since.Equal(clock.now()) || !m.Until.Equal(until) ||
|
|
m.Reason != "Kernel-Update" || m.PublicReason != "Wir spielen ein Sicherheitsupdate ein." {
|
|
t.Errorf("Wartung: %+v", m)
|
|
}
|
|
if n.Status != NodeOnline {
|
|
t.Errorf("Wartung darf den Verbindungsstatus nicht ändern: %s", n.Status)
|
|
}
|
|
if nodes, _ := s.ListNodes(ctx, ""); nodes[0].Maintenance == nil {
|
|
t.Error("ListNodes ohne Wartungsdaten")
|
|
}
|
|
|
|
// Unbefristet: Until bleibt Nullwert.
|
|
if err := s.SetNodeMaintenance(ctx, "n1", Maintenance{Reason: "Umbau"}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if n, _ := s.GetNode(ctx, "n1"); n.Maintenance == nil || !n.Maintenance.Until.IsZero() {
|
|
t.Errorf("unbefristet: %+v", n.Maintenance)
|
|
}
|
|
|
|
if err := s.ClearNodeMaintenance(ctx, "n1"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if n, _ := s.GetNode(ctx, "n1"); n.Maintenance != nil {
|
|
t.Errorf("nach Clear: %+v", n.Maintenance)
|
|
}
|
|
if err := s.SetNodeMaintenance(ctx, "fehlt", Maintenance{}); !errors.Is(err, ErrNotFound) {
|
|
t.Errorf("unbekannter Knoten: %v", err)
|
|
}
|
|
if err := s.ClearNodeMaintenance(ctx, "fehlt"); !errors.Is(err, ErrNotFound) {
|
|
t.Errorf("unbekannter Knoten: %v", err)
|
|
}
|
|
if err := s.SetNodeMaintenance(ctx, "n1", Maintenance{Until: clock.now().Add(-time.Minute)}); !errors.Is(err, ErrInvalid) {
|
|
t.Errorf("Ende in der Vergangenheit: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestMaintenanceActive(t *testing.T) {
|
|
now := time.Date(2026, 10, 2, 12, 0, 0, 0, time.UTC)
|
|
tests := []struct {
|
|
name string
|
|
m *Maintenance
|
|
want bool
|
|
}{
|
|
{"keine", nil, false},
|
|
{"unbefristet", &Maintenance{Since: now.Add(-time.Hour)}, true},
|
|
{"läuft", &Maintenance{Since: now.Add(-time.Hour), Until: now.Add(time.Minute)}, true},
|
|
{"abgelaufen", &Maintenance{Since: now.Add(-time.Hour), Until: now}, false},
|
|
}
|
|
for _, tt := range tests {
|
|
if got := tt.m.Active(now); got != tt.want {
|
|
t.Errorf("%s: Active = %v", tt.name, got)
|
|
}
|
|
}
|
|
}
|