98 lines
3.3 KiB
Go
98 lines
3.3 KiB
Go
package store
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// Aufräumen erst nach einer Frist (Job: 10 Tage), CLI-purge sofort (0).
|
|
func TestPurgeEnrollmentTokens_Retention(t *testing.T) {
|
|
s, clock := openTest(t)
|
|
mustTenant(t, s, "t1")
|
|
start := clock.now()
|
|
s.AddEnrollmentToken(ctx, "alt-benutzt", "t1", start.Add(time.Hour))
|
|
s.AddEnrollmentToken(ctx, "jung-benutzt", "t1", start.Add(30*24*time.Hour))
|
|
s.AddEnrollmentToken(ctx, "alt-abgelaufen", "t1", start.Add(time.Hour))
|
|
s.AddEnrollmentToken(ctx, "gueltig", "t1", start.Add(30*24*time.Hour))
|
|
s.Enroll(ctx, "alt-benutzt", sampleNode("n1", "k1"))
|
|
clock.advance(9 * 24 * time.Hour)
|
|
s.Enroll(ctx, "jung-benutzt", sampleNode("n2", "k2"))
|
|
clock.advance(2 * 24 * time.Hour) // jetzt: alt-* vor > 10 Tagen benutzt/abgelaufen
|
|
|
|
n, err := s.PurgeEnrollmentTokens(ctx, 10*24*time.Hour)
|
|
if err != nil || n != 2 {
|
|
t.Errorf("gelöscht %d (%v), erwartet 2 (alt-benutzt, alt-abgelaufen)", n, err)
|
|
}
|
|
if n, _ := s.PurgeEnrollmentTokens(ctx, 0); n != 1 {
|
|
t.Errorf("sofort: %d gelöscht, erwartet 1 (jung-benutzt)", n)
|
|
}
|
|
}
|
|
|
|
func TestPurgeLeasesAndCommands(t *testing.T) {
|
|
s, clock := openTest(t)
|
|
mustTenant(t, s, "t1")
|
|
mustNode(t, s, "n1", "t1", "k1")
|
|
mustNode(t, s, "n2", "t1", "k2")
|
|
now := clock.now()
|
|
s.PutLease(ctx, Lease{NodeID: "n1", IssuedAt: now, RenewAfter: now, ValidUntil: now.Add(time.Hour), Document: "{}"})
|
|
s.PutLease(ctx, Lease{NodeID: "n2", IssuedAt: now, RenewAfter: now, ValidUntil: now.Add(20 * 24 * time.Hour), Document: "{}"})
|
|
s.EnqueueNodeCommand(ctx, "n1", "resync", "")
|
|
s.EnqueueNodeCommand(ctx, "n1", "drain", "")
|
|
cmds, _ := s.PendingNodeCommands(ctx, "n1")
|
|
s.MarkNodeCommandSent(ctx, cmds[0].ID)
|
|
clock.advance(11 * 24 * time.Hour)
|
|
|
|
if n, err := s.PurgeLeases(ctx, 10*24*time.Hour); err != nil || n != 1 {
|
|
t.Errorf("Leases: %d %v", n, err)
|
|
}
|
|
if _, err := s.GetLease(ctx, "n2"); err != nil {
|
|
t.Error("gültige Lease gelöscht")
|
|
}
|
|
if n, err := s.PurgeNodeCommands(ctx, 10*24*time.Hour); err != nil || n != 1 {
|
|
t.Errorf("Befehle: %d %v", n, err)
|
|
}
|
|
if cmds, _ := s.PendingNodeCommands(ctx, "n1"); len(cmds) != 1 {
|
|
t.Error("nicht zugestellter Befehl gelöscht")
|
|
}
|
|
}
|
|
|
|
func TestNotificationsOnce(t *testing.T) {
|
|
s, _ := openTest(t)
|
|
if s.Notified(ctx, "contract:t1:2026-12-15:14") {
|
|
t.Error("neu: schon benachrichtigt")
|
|
}
|
|
if err := s.RecordNotification(ctx, "contract:t1:2026-12-15:14"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !s.Notified(ctx, "contract:t1:2026-12-15:14") {
|
|
t.Error("nicht vermerkt")
|
|
}
|
|
if err := s.RecordNotification(ctx, "contract:t1:2026-12-15:14"); err != nil {
|
|
t.Errorf("doppelt vermerken: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestMarkNodeSeen(t *testing.T) {
|
|
s, clock := openTest(t)
|
|
mustTenant(t, s, "t1")
|
|
mustNode(t, s, "n1", "t1", "k1")
|
|
clock.advance(time.Hour)
|
|
if err := s.MarkNodeSeen(ctx, "n1"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if n, _ := s.GetNode(ctx, "n1"); !n.LastSeen.Equal(clock.now()) {
|
|
t.Errorf("last_seen = %v", n.LastSeen)
|
|
}
|
|
}
|
|
|
|
func TestNotificationKeysWithPrefix(t *testing.T) {
|
|
s, _ := openTest(t)
|
|
for _, k := range []string{"offline:n1:a", "offline:n1:b", "offline:n10:c", "contract:x"} {
|
|
s.RecordNotification(ctx, k)
|
|
}
|
|
got, err := s.NotificationKeys(ctx, "offline:n1:")
|
|
if err != nil || len(got) != 2 || got[0] != "offline:n1:a" || got[1] != "offline:n1:b" {
|
|
t.Errorf("Schlüssel: %v %v", got, err)
|
|
}
|
|
}
|