Netpulse_SasS/server/internal/alerting/events_test.go

package alerting

import (
	"strings"
	"testing"
	"time"

	"github.com/netpulse/netpulse/server/internal/store"
)

// Головне, заради чого писався обмежувач: потік журналу не має
// перетворюватись на потік звернень до бази.
func TestThrottleCollapsesFlood(t *testing.T) {
	s := NewEventSink(nil, nil)

	ok, n := s.throttle("r:dev:d1", 60, 1)
	if !ok || n != 1 {
		t.Fatalf("перша подія має пройти з лічильником 1, отримали %v/%d", ok, n)
	}
	for i := 0; i < 500; i++ {
		if ok, _ := s.throttle("r:dev:d1", 60, 1); ok {
			t.Fatal("подія в межах проміжку не має йти в базу")
		}
	}

	// Проміжок минув — накопичене доливається одним зверненням, а не
	// губиться: людині важливо не кожне повідомлення окремо, а те, що
	// їх було п'ятсот.
	s.rate["r:dev:d1"].last = time.Now().Add(-2 * time.Minute)
	ok, n = s.throttle("r:dev:d1", 60, 1)
	if !ok {
		t.Fatal("після проміжку подія має пройти")
	}
	if n != 501 {
		t.Fatalf("накопичене мало долитись: очікували 501, отримали %d", n)
	}
}

// Різні хости — різні алерти, і обмежувач одного не має глушити інший.
func TestThrottleIsPerAlert(t *testing.T) {
	s := NewEventSink(nil, nil)
	if ok, _ := s.throttle("r:dev:d1", 60, 1); !ok {
		t.Fatal("перша подія хоста d1")
	}
	if ok, _ := s.throttle("r:dev:d2", 60, 1); !ok {
		t.Fatal("хост d2 не має залежати від d1")
	}
}

// Нульовий проміжок означає «без обмежень»: правило про рідкісну подію
// не має чекати хвилину, щоб про неї сказати.
func TestThrottleZeroIntervalAlwaysPasses(t *testing.T) {
	s := NewEventSink(nil, nil)
	for i := 0; i < 5; i++ {
		if ok, n := s.throttle("k", 0, 3); !ok || n != 3 {
			t.Fatalf("без проміжку має проходити все, отримали %v/%d", ok, n)
		}
	}
}

// Порожній селектор — «усі хости». Це найчастіший випадок, і саме він
// має працювати без перетворення на перелік.
func TestScopeCoversEverythingWhenEmpty(t *testing.T) {
	if !(compiledRule{}).covers("будь-який") {
		t.Fatal("правило без селектора має накривати всіх")
	}
	c := compiledRule{scope: map[string]bool{"d1": true}}
	if !c.covers("d1") || c.covers("d2") {
		t.Fatal("правило із селектором має накривати лише своїх")
	}
}

// Тригер без переліку правил відповідності ловить будь-яке порушення:
// стандарт, доданий завтра, має підпасти під нього сам.
func TestComplianceMatchDefaultsToEverything(t *testing.T) {
	ev := ComplianceEvent{RuleID: "c1", Severity: "warning"}
	if !matchesComplianceRule(store.Condition{}, ev) {
		t.Fatal("порожня умова має ловити будь-яку знахідку")
	}
}

func TestComplianceMatchNarrowsByRuleAndSeverity(t *testing.T) {
	ev := ComplianceEvent{RuleID: "c1", Severity: "warning"}

	if matchesComplianceRule(store.Condition{RuleIDs: []string{"c2"}}, ev) {
		t.Error("чуже правило відповідності не має підпадати")
	}
	if !matchesComplianceRule(store.Condition{RuleIDs: []string{"c1", "c2"}}, ev) {
		t.Error("своє правило має підпадати")
	}
	if matchesComplianceRule(store.Condition{MinSeverity: "high"}, ev) {
		t.Error("знахідка нижче порога не має підпадати")
	}
	if !matchesComplianceRule(store.Condition{MinSeverity: "info"}, ev) {
		t.Error("знахідка вище порога має підпадати")
	}
}

// Рядок журналу буває довжиною в кілограм. У заголовку алерту від
// цього немає користі, а є втрата решти тексту.
func TestTrimLineFlattensAndCuts(t *testing.T) {
	if got := trimLine("  два\nрядки  "); got != "два рядки" {
		t.Errorf("отримали %q", got)
	}
	long := strings.Repeat("я", 500)
	got := trimLine(long)
	if !strings.HasSuffix(got, "…") {
		t.Error("надто довгий рядок має бути обрізаний із багатокрапкою")
	}
	if len([]rune(got)) != 301 {
		t.Errorf("очікували 300 символів плюс багатокрапка, отримали %d", len([]rune(got)))
	}
}