Added benchmark for Args.Peek()

1 files changed, 21 insertions, 4 deletions

diff --git a/args_timing_test.go b/args_timing_test.go
index 9fb22a7..d6e9e98 100644
--- a/args_timing_test.go
+++ b/args_timing_test.go
@@ -1,13 +1,30 @@
 package fasthttp
 
 import (
+	"bytes"
 	"testing"
 )
 
 func BenchmarkArgsParse(b *testing.B) {
-	var a Args
 	s := []byte("foo=bar&baz=qqq&aaaaa=bbbb")
-	for i := 0; i < b.N; i++ {
-		a.ParseBytes(s)
-	}
+	b.RunParallel(func(pb *testing.PB) {
+		var a Args
+		for pb.Next() {
+			a.ParseBytes(s)
+		}
+	})
+}
+
+func BenchmarkArgsPeek(b *testing.B) {
+	value := []byte("foobarbaz1234")
+	key := "foobarbaz"
+	b.RunParallel(func(pb *testing.PB) {
+		var a Args
+		a.SetBytesV(key, value)
+		for pb.Next() {
+			if !bytes.Equal(a.Peek(key), value) {
+				b.Fatalf("unexpected arg value %q. Expecting %q", a.Peek(key), value)
+			}
+		}
+	})
 }