Add a HashName benchmark

miekg · Nov 28, 2018 · 2f8cf50 · 2f8cf50
1 parent 1afd100
commit 2f8cf50
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/nsecx_test.go b/nsecx_test.go
@@ -1,6 +1,9 @@
 package dns
 
-import "testing"
+import (
+	"strconv"
+	"testing"
+)
 
 func TestPackNsec3(t *testing.T) {
 	nsec3 := HashName("dnsex.nl.", SHA1, 0, "DEAD")
@@ -151,3 +154,17 @@ func TestNsec3EmptySalt(t *testing.T) {
 		t.Fatalf("expected record to match com. label")
 	}
 }
+
+func BenchmarkHashName(b *testing.B) {
+	for _, iter := range []uint16{
+		150, 2500, 5000, 10000, ^uint16(0),
+	} {
+		b.Run(strconv.Itoa(int(iter)), func(b *testing.B) {
+			for n := 0; n < b.N; n++ {
+				if HashName("some.example.org.", SHA1, iter, "deadbeef") == "" {
+					b.Fatalf("HashName failed")
+				}
+			}
+		})
+	}
+}