Implement phase 1 foundations

internal/rules/engine_test.go

@@ -0,0 +1,42 @@
+package rules
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"time"
+
+	"gitea.wayfinderak.com/wayfinderak/log-guardian/internal/actions"
+	"gitea.wayfinderak.com/wayfinderak/log-guardian/internal/alerts"
+	"gitea.wayfinderak.com/wayfinderak/log-guardian/internal/analysis"
+	"gitea.wayfinderak.com/wayfinderak/log-guardian/internal/loki"
+	"gitea.wayfinderak.com/wayfinderak/log-guardian/internal/store"
+)
+
+func TestCooldownSuppressesDuplicateIncident(t *testing.T) {
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		_, _ = w.Write([]byte(`{"status":"success","data":{"result":[{"stream":{},"values":[["1700000000000000000","error"]]}]}}`))
+	}))
+	defer ts.Close()
+
+	s := store.NewFileStore(t.TempDir() + "/config.json")
+	if err := s.UpsertRule(store.Rule{Name: "Errors", Enabled: true, LogQL: `{service="api"}`, Threshold: 1, Window: "5m", Cooldown: "1h", Actions: []store.Action{{Type: "record_recommendation", Enabled: true}}}); err != nil {
+		t.Fatal(err)
+	}
+	engine := NewEngine(s, loki.New(ts.URL, "", "", ""), analysis.NoopAnalyzer{}, actions.NewRunner(true), alerts.NewDispatcher())
+	engine.CheckAll(t.Context())
+	engine.CheckAll(t.Context())
+	data, err := s.Load()
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(data.Incidents) != 1 {
+		t.Fatalf("expected one incident, got %d", len(data.Incidents))
+	}
+	if data.Rules[0].SuppressedCount != 1 {
+		t.Fatalf("expected suppressed count 1, got %d", data.Rules[0].SuppressedCount)
+	}
+	if time.Since(data.Rules[0].LastAlertedAt) > time.Minute {
+		t.Fatalf("last alerted not set: %s", data.Rules[0].LastAlertedAt)
+	}
+}