Netpulse_SasS/server/internal/alerting/traps_test.go

package alerting

import (
	"testing"

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

func trapRule(cond store.Condition, scope map[string]bool) compiledRule {
	return compiledRule{
		rule:  store.Rule{ID: "r1", Source: "trap", Condition: cond},
		scope: scope,
	}
}

var linkDown = TrapEvent{
	DeviceID: "d1",
	SourceIP: "10.20.0.7",
	TrapOID:  "1.3.6.1.6.3.1.1.5.3",
	Varbinds: []TrapVarbind{
		{OID: "1.3.6.1.2.1.1.3.0", Value: "12345"},
		{OID: "1.3.6.1.2.1.2.2.1.1.7", Value: "7"},
		{OID: "1.3.6.1.2.1.2.2.1.8.7", Value: "2"},
	},
}

// Найпростіша умова, заради якої все й робилось: «цей OID».
func TestTrapMatchesByOID(t *testing.T) {
	if !trapMatches(trapRule(store.Condition{TrapOID: "1.3.6.1.6.3.1.1.5.3"}, nil), linkDown) {
		t.Fatal("linkDown не збігся з власним OID")
	}
	if trapMatches(trapRule(store.Condition{TrapOID: "1.3.6.1.6.3.1.1.5.4"}, nil), linkDown) {
		t.Fatal("linkUp зловив linkDown")
	}
	// Провідна крапка не має вирішувати долю правила.
	if !trapMatches(trapRule(store.Condition{TrapOID: ".1.3.6.1.6.3.1.1.5.3"}, nil), linkDown) {
		t.Fatal("крапка на початку зламала збіг")
	}
}

// Умова на varbind порівнюється зі СТОВПЦЕМ, а не з конкретним рядком:
// у трапі приходить ifOperStatus.7, а людина пише ifOperStatus, бо
// номера порту наперед не знає.
func TestTrapMatchesVarbindByColumn(t *testing.T) {
	c := trapRule(store.Condition{
		TrapOID:      "1.3.6.1.6.3.1.1.5.3",
		VarbindOID:   "1.3.6.1.2.1.2.2.1.8",
		VarbindValue: "2",
	}, nil)
	if !trapMatches(c, linkDown) {
		t.Fatal("ifOperStatus=down не збігся")
	}

	c.rule.Condition.VarbindValue = "1"
	if trapMatches(c, linkDown) {
		t.Fatal("ifOperStatus=up збігся з трапом, у якому down")
	}

	// Порожнє значення означає «щоб такий varbind узагалі був».
	c.rule.Condition.VarbindValue = ""
	if !trapMatches(c, linkDown) {
		t.Fatal("умова «є такий varbind» не спрацювала")
	}
	c.rule.Condition.VarbindOID = "1.3.6.1.4.1.9.9.999"
	if trapMatches(c, linkDown) {
		t.Fatal("умова спрацювала на varbind, якого в трапі немає")
	}
}

// Адреса в умові приймає і одну адресу, і підмережу.
func TestTrapMatchesBySourceIP(t *testing.T) {
	if !trapMatches(trapRule(store.Condition{SourceIP: "10.20.0.7"}, nil), linkDown) {
		t.Fatal("точна адреса не збіглася")
	}
	if !trapMatches(trapRule(store.Condition{SourceIP: "10.20.0.0/24"}, nil), linkDown) {
		t.Fatal("підмережа не збіглася")
	}
	if trapMatches(trapRule(store.Condition{SourceIP: "10.30.0.0/24"}, nil), linkDown) {
		t.Fatal("чужа підмережа збіглася")
	}
}

// Головне рішення цього джерела: трап від адреси, яку не впізнали,
// піднімає алерт ЛИШЕ якщо правило назвало адресу явно. Інакше він
// лишається в переліку невідомих джерел, а не на дошці алертів.
func TestTrapWithoutDeviceNeedsExplicitSource(t *testing.T) {
	stranger := linkDown
	stranger.DeviceID = ""
	stranger.SourceIP = "10.20.0.250"

	if trapMatches(trapRule(store.Condition{TrapOID: "1.3.6.1.6.3.1.1.5.3"}, nil), stranger) {
		t.Fatal("трап від невідомої адреси підняв алерт без явної адреси в умові")
	}
	c := trapRule(store.Condition{
		TrapOID:  "1.3.6.1.6.3.1.1.5.3",
		SourceIP: "10.20.0.0/24",
	}, nil)
	if !trapMatches(c, stranger) {
		t.Fatal("правило з явною адресою не спрацювало на незнайомця")
	}
}

// Селектор звужує правило до хостів. До трапа від адреси поза
// інвентарем його застосувати неможливо — інакше правило з адресою
// джерела не спрацювало б ніколи.
func TestTrapSelectorAppliesOnlyToKnownHosts(t *testing.T) {
	scope := map[string]bool{"d2": true}
	if trapMatches(trapRule(store.Condition{TrapOID: "1.3.6.1.6.3.1.1.5.3"}, scope), linkDown) {
		t.Fatal("хост поза селектором підпав під правило")
	}

	stranger := linkDown
	stranger.DeviceID = ""
	c := trapRule(store.Condition{SourceIP: "10.20.0.7"}, scope)
	if !trapMatches(c, stranger) {
		t.Fatal("селектор заблокував правило про адресу поза інвентарем")
	}
}

func TestVarbindPrefixMatchIsNotSubstring(t *testing.T) {
	// 1.3.6.1.2.1.2.2.1.10 (ifInOctets) не має вважатись збігом для
	// 1.3.6.1.2.1.2.2.1.1 (ifIndex) — інакше умова про індекс порту
	// зривалась би на лічильнику байтів.
	if varbindIs("1.3.6.1.2.1.2.2.1.10", "1.3.6.1.2.1.2.2.1.1") {
		t.Fatal("сусідній стовпець зарахований як збіг")
	}
	if !varbindIs("1.3.6.1.2.1.2.2.1.1.7", "1.3.6.1.2.1.2.2.1.1") {
		t.Fatal("рядок стовпця не зарахований")
	}
	if !varbindIs("1.3.6.1.2.1.1.3.0", "1.3.6.1.2.1.1.3.0") {
		t.Fatal("точний збіг не зарахований")
	}
}

// Текст алерту має пояснювати подію, а не перелічувати OID-и.
func TestTrapDetailShowsNamedFields(t *testing.T) {
	view := &tenantView{}
	got := trapDetail(view, linkDown)
	if got != "ifIndex=7, ifOperStatus=down" {
		t.Fatalf("подробиці трапа: %q", got)
	}
}