test: use integers.json file to benchmark integer decoder

go-faster · Mar 20, 2022 · 7787838 · 7787838
1 parent 8bfec7d
commit 7787838
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 21 deletions.
diff --git a/dec_int_test.go b/dec_int_test.go
@@ -7,28 +7,50 @@ import (
 )
 
 func BenchmarkDecoder_Int(b *testing.B) {
-	data := []byte(`69315063`)
-	d := GetDecoder()
-	for i := 0; i < b.N; i++ {
-		d.ResetBytes(data)
-		if _, err := d.Int(); err != nil {
-			b.Fatal(err)
-		}
-	}
+	runTestdataFile("integers.json", b.Fatal, func(name string, data []byte) {
+		b.Run(name, func(b *testing.B) {
+			d := GetDecoder()
+			cb := func(d *Decoder) error {
+				_, err := d.Int()
+				return err
+			}
+			b.ReportAllocs()
+			b.ResetTimer()
+
+			for i := 0; i < b.N; i++ {
+				d.ResetBytes(data)
+
+				if err := d.Arr(cb); err != nil {
+					b.Fatal(err)
+				}
+			}
+		})
+	})
 }
 
 func BenchmarkDecoder_Uint(b *testing.B) {
-	data := []byte(`69315063`)
-	d := GetDecoder()
-	for i := 0; i < b.N; i++ {
-		d.ResetBytes(data)
-		if _, err := d.UInt(); err != nil {
-			b.Fatal(err)
-		}
-	}
+	runTestdataFile("integers.json", b.Fatal, func(name string, data []byte) {
+		b.Run(name, func(b *testing.B) {
+			d := GetDecoder()
+			cb := func(d *Decoder) error {
+				_, err := d.UInt()
+				return err
+			}
+			b.ReportAllocs()
+			b.ResetTimer()
+
+			for i := 0; i < b.N; i++ {
+				d.ResetBytes(data)
+
+				if err := d.Arr(cb); err != nil {
+					b.Fatal(err)
+				}
+			}
+		})
+	})
 }
 
-func TestDecoder_int_sizes(t *testing.T) {
+func TestDecoderIntSizes(t *testing.T) {
 	data := []byte(`69315063`)
 	d := GetDecoder()
 	for _, size := range []int{32, 64} {
@@ -39,7 +61,7 @@ func TestDecoder_int_sizes(t *testing.T) {
 	}
 }
 
-func TestDecoder_uint_sizes(t *testing.T) {
+func TestDecoderUintSizes(t *testing.T) {
 	data := []byte(`69315063`)
 	d := GetDecoder()
 	for _, size := range []int{32, 64} {

diff --git a/enc_test.go b/enc_test.go
@@ -7,7 +7,7 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
-func TestEncoder_byte_should_grow_buffer(t *testing.T) {
+func TestEncoderByteShouldGrowBuffer(t *testing.T) {
 	should := require.New(t)
 	e := GetEncoder()
 	e.byte('1')
@@ -41,14 +41,14 @@ func TestEncoder(t *testing.T) {
 	})
 }
 
-func TestEncoder_Raw_should_grow_buffer(t *testing.T) {
+func TestEncoderRawShouldGrowBuffer(t *testing.T) {
 	should := require.New(t)
 	e := GetEncoder()
 	e.RawStr("123")
 	should.Equal("123", string(e.Bytes()))
 }
 
-func TestEncoder_Str_should_grow_buffer(t *testing.T) {
+func TestEncoderStrShouldGrowBuffer(t *testing.T) {
 	should := require.New(t)
 	e := GetEncoder()
 	e.Str("123")