Netpulse_SasS/server/internal/store/retention_policy_test.go

package store

import (
	"errors"
	"testing"
)

// Перевіряється рівно одне — правила, за якими строки визнаються
// несумісними. Саме тут ціна помилки не «форма показала не те», а
// «місячні графіки зникли й не відновлюються»: сирі дані, з яких
// рахується згортка, після видалення взяти нізвідки.
//
// Бази ці перевірки не потребують навмисно: якби вони жили в SQL, то
// перевірити їх можна було б лише на живому Postgres — тобто на практиці
// ніколи.

func days(n int) *int { return &n }

func TestValidateRetentionRollupNotShorterThanSource(t *testing.T) {
	cases := []struct {
		name string
		keep map[string]*int
		ok   bool
	}{
		{
			name: "згортка довша за джерело — норма",
			keep: map[string]*int{"icmp_raw": days(7), "icmp_5m": days(400)},
			ok:   true,
		},
		{
			name: "згортка коротша за джерело — діра в графіках",
			keep: map[string]*int{"icmp_raw": days(90), "icmp_5m": days(30)},
			ok:   false,
		},
		{
			name: "джерело не видаляють, згортку видаляють — теж діра",
			keep: map[string]*int{"icmp_raw": nil, "icmp_5m": days(400)},
			ok:   false,
		},
		{
			name: "згортку не видаляють — джерело може бути будь-яким",
			keep: map[string]*int{"icmp_raw": days(7), "icmp_5m": nil},
			ok:   true,
		},
		{
			name: "джерела немає в наборі — умову перевіряти нема з чим",
			keep: map[string]*int{"icmp_5m": days(30)},
			ok:   true,
		},
		{
			name: "ланцюжок сирі → 5 хв → година",
			keep: map[string]*int{
				"ifc_raw": days(7), "ifc_5m": days(400), "ifc_1h": days(30),
			},
			ok: false, // година коротша за п'ять хвилин
		},
	}

	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			err := ValidateRetention(c.keep)
			if c.ok && err != nil {
				t.Fatalf("очікували прийняття, отримали: %v", err)
			}
			if !c.ok {
				if err == nil {
					t.Fatal("очікували відмову, набір прийнято")
				}
				if !errors.Is(err, ErrRetentionInvalid) {
					t.Fatalf("очікували ErrRetentionInvalid, отримали %v", err)
				}
			}
		})
	}
}

func TestValidateRetentionFloors(t *testing.T) {
	// Нижня межа сирих даних — не смак, а вікно, за яке встигає
	// порахуватись згортка. Доба тут означала б видалення сирих вимірів
	// раніше, ніж їх згорнули.
	if err := ValidateRetention(map[string]*int{"icmp_raw": days(1)}); err == nil {
		t.Fatal("одна доба для сирих вимірів має відхилятись")
	}

	// Журнал аудиту коротшим за рік не буває: це доказова база, і
	// скорочення строку тут означає знищення доказів раніше, ніж по них
	// приходять.
	if err := ValidateRetention(map[string]*int{"audit_log": days(90)}); err == nil {
		t.Fatal("90 діб для журналу аудиту мають відхилятись")
	}
	if err := ValidateRetention(map[string]*int{"audit_log": days(3650)}); err != nil {
		t.Fatalf("десять років для журналу мають прийматись: %v", err)
	}

	// «Не видаляти» проходить завжди — це і є типовий стан після
	// оновлення на інсталяції, де строку не було.
	for _, k := range retentionCatalog {
		if err := ValidateRetention(map[string]*int{k.Kind: nil}); err != nil {
			t.Fatalf("%s: «не видаляти» має прийматись, отримали %v", k.Kind, err)
		}
	}

	if err := ValidateRetention(map[string]*int{"такого_виду_немає": days(30)}); err == nil {
		t.Fatal("невідомий вид даних має відхилятись")
	}
}

func TestRetentionAuditMetaMarksShortening(t *testing.T) {
	before := []RetentionRow{
		{RetentionKindInfo: RetentionKindInfo{Kind: "syslog"}, KeepDays: days(90)},
		{RetentionKindInfo: RetentionKindInfo{Kind: "traps"}, KeepDays: days(90)},
		{RetentionKindInfo: RetentionKindInfo{Kind: "alerts_history"}, KeepDays: nil},
	}
	meta := RetentionAuditMeta(before, map[string]*int{
		"syslog":         days(30), // скорочення
		"traps":          days(90), // без змін — у журнал не потрапляє
		"alerts_history": days(365),
	})

	changes, _ := meta["changes"].([]map[string]any)
	if len(changes) != 2 {
		t.Fatalf("очікували два записані рядки, отримали %d", len(changes))
	}

	got := map[string]bool{}
	for _, c := range changes {
		got[c["kind"].(string)] = c["shortened"].(bool)
	}
	if !got["syslog"] {
		t.Error("скорочення строку syslog має бути позначене")
	}
	// Було «не видаляти», стало 365 діб — це теж скорочення, і саме
	// воно найнебезпечніше: до нього не зникало нічого.
	if !got["alerts_history"] {
		t.Error("перехід з «не видаляти» на строк має вважатись скороченням")
	}
	if _, ok := got["traps"]; ok {
		t.Error("незмінений строк не має потрапляти в журнал")
	}
}