From 97385f4f12903517123f9e03fb73ce090fd1e77e Mon Sep 17 00:00:00 2001
From: Will Roden <willabides@github.com>
Date: Sat, 21 Aug 2021 12:53:14 -0500
Subject: [PATCH] include fp.RunFuzz in fuzzing

---
 fuzz.go             |  9 +++++++--
 internal/fp/fuzz.go | 23 ++++++++++++++++-------
 2 files changed, 23 insertions(+), 9 deletions(-)

diff --git a/fuzz.go b/fuzz.go
index 5d6b9b42..8759806b 100644
--- a/fuzz.go
+++ b/fuzz.go
@@ -4,13 +4,18 @@ package rjson
 
 import (
 	"fmt"
+
+	"github.com/willabides/rjson/internal/fp"
 )
 
 // Fuzz is for running go-fuzz tests
 func Fuzz(data []byte) int {
 	score := 0
-
-	for _, fd := range fuzzers {
+	allFuzzers := append(fuzzers, fuzzer{
+		name: "fp.RunFuzz",
+		fn:   fp.RunFuzz,
+	})
+	for _, fd := range allFuzzers {
 		d := make([]byte, len(data))
 		copy(d, data)
 		s, err := fd.fn(d)
diff --git a/internal/fp/fuzz.go b/internal/fp/fuzz.go
index 316e01fa..8e75c3ee 100644
--- a/internal/fp/fuzz.go
+++ b/internal/fp/fuzz.go
@@ -29,25 +29,34 @@ func fuzzOracle(data []byte) (val float64, p int, err error) {
 	return val, int(decoder.InputOffset()), nil
 }
 
-// Fuzz is for running go-fuzz tests
-func Fuzz(data []byte) int {
+// RunFuzz runs a fuzz test but returns an error instead of panicking
+func RunFuzz(data []byte) (int, error) {
 	want, wantOffset, err := fuzzOracle(data)
 	wantErr := err != nil
 	got, gotOffset, gotErr := ParseJSONFloatPrefix(data)
 	if wantErr {
 		if gotErr == nil {
-			panic("missed an error")
+			return 0, fmt.Errorf("missed an error")
 		}
 	} else {
 		if gotErr != nil {
-			panic("unexpected error")
+			return 0, fmt.Errorf("unexpected error")
 		}
 	}
 	if want != got {
-		panic(fmt.Sprintf("wanted %v but got %v\n", want, got))
+		return 0, fmt.Errorf("wanted %v but got %v", want, got)
 	}
 	if wantOffset != gotOffset {
-		panic(fmt.Sprintf("wanted offset %v but got %v\n", wantOffset, gotOffset))
+		return 0, fmt.Errorf("wanted offset %v but got %v", wantOffset, gotOffset)
+	}
+	return 0, nil
+}
+
+// Fuzz is for running go-fuzz tests
+func Fuzz(data []byte) int {
+	n, err := RunFuzz(data)
+	if err != nil {
+		panic(err.Error())
 	}
-	return 0
+	return n
 }