test: add zero allocation tests

alloc_test.go

@@ -0,0 +1,72 @@
+//go:build !race && !gofuzz
+// +build !race,!gofuzz
+
+package jx
+
+import (
+	"testing"
+)
+
+const defaultAllocRuns = 10
+
+func zeroAlloc(t *testing.T, f func()) {
+	t.Helper()
+	avg := testing.AllocsPerRun(defaultAllocRuns, f)
+	if avg > 0 {
+		t.Errorf("Allocated %f bytes per run", avg)
+	}
+}
+
+func zeroAllocDec(t *testing.T, buf []byte, f func(d *Decoder) error) {
+	t.Helper()
+	d := DecodeBytes(buf)
+	zeroAlloc(t, func() {
+		d.ResetBytes(buf)
+		if err := f(d); err != nil {
+			t.Fatal(err)
+		}
+	})
+}
+
+func zeroAllocDecStr(t *testing.T, s string, f func(d *Decoder) error) {
+	t.Helper()
+	zeroAllocDec(t, []byte(s), f)
+}
+
+func TestZeroAlloc(t *testing.T) {
+	// Tests that checks for zero allocations.
+	t.Run("Decoder", func(t *testing.T) {
+		t.Run("Validate", func(t *testing.T) {
+			zeroAllocDec(t, data, func(d *Decoder) error {
+				return d.Validate()
+			})
+		})
+		t.Run("Int", func(t *testing.T) {
+			zeroAllocDecStr(t, "12345", func(d *Decoder) error {
+				v, err := d.Int()
+				if v != 12345 {
+					t.Fatal(v)
+				}
+				return err
+			})
+		})
+		t.Run("StrBytes", func(t *testing.T) {
+			zeroAllocDecStr(t, `"hello"`, func(d *Decoder) error {
+				v, err := d.StrBytes()
+				if string(v) != "hello" {
+					t.Fatal(string(v))
+				}
+				return err
+			})
+		})
+		t.Run("Str", func(t *testing.T) {
+			zeroAllocDecStr(t, `"hello"`, func(d *Decoder) error {
+				v, err := d.Str()
+				if v != "hello" {
+					t.Fatal(err)
+				}
+				return err
+			})
+		})
+	})
+}

go.test.sh

@@ -2,5 +2,8 @@
 
 set -e
 
-# test with -race
+echo "test"
+go test --timeout 5m ./...
+
+echo "test -race"
 go test --timeout 5m -race ./...