Allow to pass a custom fuzzing function to FuzzConsistency (#53)

Allow to pass a custom fuzzing function to FuzzConsistentcy
spacemeshos · Mar 27, 2023 · 1dc7783 · 1dc7783
1 parent 816695f
commit 1dc7783
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tester/fuzz.go b/tester/fuzz.go
@@ -13,9 +13,9 @@ import (
 	"github.com/spacemeshos/go-scale"
 )
 
-func FuzzConsistency[T any, H scale.TypePtr[T]](f *testing.F) {
+func FuzzConsistency[T any, H scale.TypePtr[T]](f *testing.F, fuzzFuncs ...any) {
 	f.Fuzz(func(t *testing.T, data []byte) {
-		fuzzer := fuzz.NewFromGoFuzz(data)
+		fuzzer := fuzz.NewFromGoFuzz(data).Funcs(fuzzFuncs...)
 		var object T
 		fuzzer.Fuzz(&object)