Add benchmark + Get

corazawaf · Nov 7, 2024 · fd6c77c · fd6c77c
1 parent 8ee2395
commit fd6c77c
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/internal/collections/map_test.go b/internal/collections/map_test.go
@@ -16,6 +16,7 @@ package collections
 import (
 	"fmt"
 	"regexp"
+	"strconv"
 	"testing"
 
 	"github.com/corazawaf/coraza/v3/types/variables"
@@ -106,3 +107,14 @@ func TestNewCaseSensitiveKeyMap(t *testing.T) {
 	}
 
 }
+
+func BenchmarkTxGet(b *testing.B) {
+	c := NewCaseSensitiveKeyMap(variables.RequestHeaders)
+	for i := 0; i < b.N; i++ {
+		c.Set("key"+strconv.Itoa(i), []string{"value1", "value2"})
+	}
+	// Benchmark the Get method
+	for i := 0; i < b.N; i++ {
+		c.Get("key500")
+	}
+}