From e78b45ff61024f1fe68f79e99c190391a84b318d Mon Sep 17 00:00:00 2001
From: Lukasz Mierzwa <lukasz@cloudflare.com>
Date: Mon, 24 Jun 2024 14:13:01 +0100
Subject: [PATCH] More complete rules benchmark

---
 cmd/pint/bench_test.go | 64 ++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 62 insertions(+), 2 deletions(-)

diff --git a/cmd/pint/bench_test.go b/cmd/pint/bench_test.go
index 0459a56f..eff74211 100644
--- a/cmd/pint/bench_test.go
+++ b/cmd/pint/bench_test.go
@@ -2,10 +2,15 @@ package main
 
 import (
 	"context"
+	"fmt"
 	"log/slog"
+	"net/http"
+	"net/http/httptest"
+	"os"
 	"testing"
 
 	"github.com/prometheus/client_golang/prometheus"
+	"github.com/stretchr/testify/require"
 
 	"github.com/cloudflare/pint/internal/config"
 	"github.com/cloudflare/pint/internal/discovery"
@@ -38,10 +43,65 @@ func BenchmarkCheckRules(b *testing.B) {
 		b.FailNow()
 	}
 
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		switch r.URL.Path {
+		case "/api/v1/status/config":
+			w.WriteHeader(200)
+			w.Header().Set("Content-Type", "application/json")
+			_, _ = w.Write([]byte(`{"status":"success","data":{"yaml":"global:\n  scrape_interval: 30s\n"}}`))
+		case "/api/v1/status/flags":
+			w.WriteHeader(200)
+			w.Header().Set("Content-Type", "application/json")
+			_, _ = w.Write([]byte(`{"status":"success","data":{"storage.tsdb.retention.time": "1d"}}`))
+		case "/api/v1/metadata":
+			w.WriteHeader(200)
+			w.Header().Set("Content-Type", "application/json")
+			_, _ = w.Write([]byte(`{"status":"success","data":{}}`))
+		case "/api/v1/query":
+			w.WriteHeader(200)
+			w.Header().Set("Content-Type", "application/json")
+			_, _ = w.Write([]byte(`{"status":"success","data":{"resultType":"vector","result":[]}}`))
+		case "/api/v1/query_range":
+			w.WriteHeader(200)
+			w.Header().Set("Content-Type", "application/json")
+			_, _ = w.Write([]byte(`{"status":"success","data":{"resultType":"matrix","result":[]}}`))
+		default:
+			b.Logf("%s %s", r.Method, r.URL.Path)
+			w.WriteHeader(404)
+			_, _ = w.Write([]byte(`Not found`))
+		}
+	}))
+	defer srv.Close()
+
+	tmp := b.TempDir()
+	content := []byte(fmt.Sprintf(`prometheus "prom" {
+  uri         = "%s"
+  timeout     = "30s"
+  uptime      = "prometheus_ready"
+  concurrency = 10
+  rateLimit   = 5000
+}
+
+rule {
+  alerts {
+    range    = "1h"
+    step     = "1m"
+    resolve  = "5m"
+    minCount = 50
+  }
+}
+`, srv.URL))
+	require.NoError(b, os.WriteFile(tmp+"/.pint.hcl", content, 0o644))
+
 	ctx := context.Background()
-	cfg, _ := config.Load("", false)
+	cfg, err := config.Load(tmp+"/.pint.hcl", false)
+	require.NoError(b, err)
+
 	gen := config.NewPrometheusGenerator(cfg, prometheus.NewRegistry())
+	require.NoError(b, gen.GenerateStatic())
+
+	b.ResetTimer()
 	for n := 0; n < b.N; n++ {
-		_, _ = checkRules(ctx, 10, true, gen, cfg, entries)
+		_, _ = checkRules(ctx, 10, false, gen, cfg, entries)
 	}
 }