batch_tob_test.go 1.38 KB
Newer Older
Maurelian's avatar
Maurelian committed
1 2 3 4 5
package derive

import (
	"testing"

Sabnock01's avatar
Sabnock01 committed
6
	"github.com/ethereum-optimism/optimism/op-service/testutils/fuzzerutils"
Maurelian's avatar
Maurelian committed
7
	fuzz "github.com/google/gofuzz"
8
	"github.com/stretchr/testify/require"
Maurelian's avatar
Maurelian committed
9 10 11 12
)

// FuzzBatchRoundTrip executes a fuzz test similar to TestBatchRoundTrip, which tests that arbitrary BatchData will be
// encoded and decoded without loss of its original values.
13
// Does not test the span batch type because the fuzzer is not aware of the structure of a span batch.
Maurelian's avatar
Maurelian committed
14 15 16 17 18 19
func FuzzBatchRoundTrip(f *testing.F) {
	f.Fuzz(func(t *testing.T, fuzzedData []byte) {
		// Create our fuzzer wrapper to generate complex values
		typeProvider := fuzz.NewFromGoFuzz(fuzzedData).NilChance(0).MaxDepth(10000).NumElements(0, 0x100).AllowUnexportedFields(true)
		fuzzerutils.AddFuzzerFunctions(typeProvider)

20 21 22
		var singularBatch SingularBatch
		typeProvider.Fuzz(&singularBatch)

Maurelian's avatar
Maurelian committed
23 24
		// Create our batch data from fuzzed data
		var batchData BatchData
25
		// force batchdata to only contain singular batch
26
		batchData.inner = &singularBatch
27

Maurelian's avatar
Maurelian committed
28 29
		// Encode our batch data
		enc, err := batchData.MarshalBinary()
30
		require.NoError(t, err)
Maurelian's avatar
Maurelian committed
31 32 33 34

		// Decode our encoded batch data
		var dec BatchData
		err = dec.UnmarshalBinary(enc)
35
		require.NoError(t, err)
Maurelian's avatar
Maurelian committed
36 37

		// Ensure the round trip encoding of batch data did not result in data loss
38
		require.Equal(t, &batchData, &dec, "round trip batch encoding/decoding did not match original values")
Maurelian's avatar
Maurelian committed
39 40
	})
}