munin/internal/store/sysinfo_test.go
2026-10-19 10:13:56 +02:00

189 lines
5.9 KiB
Go

package store
import (
"context"
"testing"
"time"
)
// Tests der Kennzahlen für die Systemseite (sysinfo.go). Laufen mit
// MUNIN_TEST_PG auch gegen PostgreSQL (siehe testStore).
// TestInventory zählt Projekte, Einträge, Papierkorb, Fassungen und Keys —
// Keys nach denselben Regeln wie die Seite "Zugänge".
func TestInventory(t *testing.T) {
s := testStore(t)
ctx := context.Background()
now := time.Now()
empty, err := s.Inventory(ctx, now)
if err != nil {
t.Fatal(err)
}
if empty != (Inventory{}) {
t.Errorf("leerer Workspace: %+v, erwartet nur Nullen", empty)
}
p := testProject(t, s)
for _, slug := range []string{"a", "b", "c"} {
if _, err := s.PutEntry(ctx, EntryWrite{ProjectID: p.ID, Slug: slug, Title: slug, Content: "x", Source: SourceHuman}); err != nil {
t.Fatal(err)
}
}
// Eine Änderung legt eine ältere Fassung ab.
if _, err := s.PutEntry(ctx, EntryWrite{ProjectID: p.ID, Slug: "a", Title: "a", Content: "neu", Source: SourceHuman}); err != nil {
t.Fatal(err)
}
if err := s.MoveEntryToTrash(ctx, p.ID, "c", "test"); err != nil {
t.Fatal(err)
}
// Keys: einer frisch, einer läuft bald ab, einer zurückgezogen, einer
// abgelaufen. Gültig sind die ersten beiden.
if _, err := s.CreateAPIKey(ctx, p.ID, "frisch", "h1", "write"); err != nil {
t.Fatal(err)
}
if _, err := s.CreateAPIKeyWithExpiry(ctx, p.ID, "bald", "h2", "read", now.Add(3*24*time.Hour)); err != nil {
t.Fatal(err)
}
gone, err := s.CreateAPIKey(ctx, p.ID, "weg", "h3", "read")
if err != nil {
t.Fatal(err)
}
if err := s.RevokeAPIKey(ctx, gone.ID); err != nil {
t.Fatal(err)
}
if _, err := s.CreateAPIKeyWithExpiry(ctx, p.ID, "abgelaufen", "h4", "read", now.Add(time.Hour)); err != nil {
t.Fatal(err)
}
// Zwei Stunden später: "abgelaufen" ist es dann.
b, err := s.Inventory(ctx, now.Add(2*time.Hour))
if err != nil {
t.Fatal(err)
}
want := Inventory{Projects: 1, Entries: 2, Versions: 1, Trash: 1, Keys: 2, KeysExpiring: 1}
if b != want {
t.Errorf("Bestand = %+v\nerwartet %+v", b, want)
}
// 100 Tage später ist "frisch" nie benutzt worden und damit ungenutzt;
// "bald" ist bis dahin abgelaufen und zählt gar nicht mehr.
b, err = s.Inventory(ctx, now.Add(100*24*time.Hour))
if err != nil {
t.Fatal(err)
}
if b.Keys != 1 || b.KeysUnused != 1 || b.KeysExpiring != 0 {
t.Errorf("nach 100 Tagen: Keys %d, ungenutzt %d, laufen ab %d — erwartet 1, 1, 0", b.Keys, b.KeysUnused, b.KeysExpiring)
}
}
// TestDiskUsage: Ein Workspace mit Inhalt belegt Platz. Wie viel genau, hängt
// von der Datenbank ab — geprüft wird nur, dass die Zahl plausibel ist und
// mit dem Inhalt wächst.
func TestDiskUsage(t *testing.T) {
s := testStore(t)
ctx := context.Background()
before, err := s.DiskUsage(ctx)
if err != nil {
t.Fatal(err)
}
if before.Data <= 0 {
t.Fatalf("leerer Workspace belegt %d Bytes — die Tabellen allein brauchen Platz", before.Data)
}
p := testProject(t, s)
content := make([]byte, 64<<10)
for i := range content {
content[i] = 'a' + byte(i%26)
}
for _, slug := range []string{"gross-1", "gross-2", "gross-3", "gross-4"} {
if _, err := s.PutEntry(ctx, EntryWrite{ProjectID: p.ID, Slug: slug, Title: slug, Content: string(content), Source: SourceHuman}); err != nil {
t.Fatal(err)
}
}
after, err := s.DiskUsage(ctx)
if err != nil {
t.Fatal(err)
}
// Bei SQLite landet das Geschriebene zunächst in der WAL-Datei, bei
// PostgreSQL in den Tabellen. Zusammen muss es mehr geworden sein.
if after.Total() <= before.Total() {
t.Errorf("Belegung vorher %d, nachher %d — sollte gewachsen sein", before.Total(), after.Total())
}
}
// TestActivity zählt Schreibvorgänge je Projekt, Tag und Weg sowie Suchen,
// und lässt alles vor der Grenze weg.
func TestActivity(t *testing.T) {
s := testStore(t)
ctx := context.Background()
p := testProject(t, s)
// Mit festen Zeitstempeln direkt eintragen — LogAudit und LogSearch
// nehmen immer "jetzt".
audit := func(ts, action, source string) {
t.Helper()
_, err := s.write.ExecContext(ctx,
`INSERT INTO audit_events (id, ts, project_id, actor, action, source) VALUES (?, ?, ?, 'test', ?, ?)`,
NewID(), ts, p.ID, action, source)
if err != nil {
t.Fatal(err)
}
}
search := func(ts string) {
t.Helper()
_, err := s.write.ExecContext(ctx,
`INSERT INTO search_log (ts, project_id, query, results) VALUES (?, ?, 'frage', 1)`, ts, p.ID)
if err != nil {
t.Fatal(err)
}
}
audit("2026-10-01T08:00:00Z", ActionEntryWrite, SourceMCP)
audit("2026-10-01T09:00:00Z", ActionEntryWrite, SourceMCP)
audit("2026-10-01T10:00:00Z", ActionEntryDelete, SourceWeb)
audit("2026-10-01T11:00:00Z", ActionLoginSuccess, SourceWeb) // kein Schreibvorgang
audit("2026-10-02T08:00:00Z", ActionEntryWrite, SourceWeb)
audit("2026-09-20T08:00:00Z", ActionEntryWrite, SourceMCP) // vor der Grenze
search("2026-10-02T12:00:00Z")
search("2026-10-02T13:00:00Z")
search("2026-09-20T13:00:00Z") // vor der Grenze
list, err := s.Activity(ctx, time.Date(2026, 9, 25, 0, 0, 0, 0, time.UTC))
if err != nil {
t.Fatal(err)
}
per := map[string]DailyActivity{}
for _, a := range list {
if a.ProjectID != p.ID {
t.Errorf("fremdes Projekt %q", a.ProjectID)
}
per[a.Tag] = a
}
if len(per) != 2 {
t.Fatalf("%d Tage, erwartet 2: %+v", len(per), list)
}
if a := per["2026-10-01"]; a.AgentWrites != 2 || a.HumanWrites != 1 || a.Searches != 0 {
t.Errorf("1.10.: %+v, erwartet 2 Agent, 1 Mensch, 0 Suchen", a)
}
if a := per["2026-10-02"]; a.AgentWrites != 0 || a.HumanWrites != 1 || a.Searches != 2 {
t.Errorf("2.10.: %+v, erwartet 0 Agent, 1 Mensch, 2 Suchen", a)
}
}
// TestSchemaVersion: Nach dem Öffnen steht der Workspace auf der neuesten
// Migration, die das Programm mitbringt.
func TestSchemaVersion(t *testing.T) {
s := testStore(t)
actual, expected, err := s.SchemaVersion(context.Background())
if err != nil {
t.Fatal(err)
}
if expected == "" || actual != expected {
t.Errorf("Schemastand ist %q, soll %q", actual, expected)
}
}