Function FuzzBytesToDataUpload

provisionersdk/proto/dataupload_test.go:112–148

Fuzz must be run manually with the `-fuzz` flag to generate random test cases. By default, it only runs the added seed corpus cases. go test -fuzz=FuzzBytesToDataUpload

(f *testing.F)

// By default, it only runs the added seed corpus cases.
// go test -fuzz=FuzzBytesToDataUpload
func FuzzBytesToDataUpload(f *testing.F) {
	// Cases to always run in standard `go test` runs.
	always := [][]byte{
		{},
		[]byte("1"),
		[]byte("small"),
	}
	for _, data := range always {
		f.Add(data)
	}

	f.Fuzz(func(t *testing.T, data []byte) {
		first, chunks, err := proto.BytesToDataUpload(proto.DataUploadType_UPLOAD_TYPE_MODULE_FILES, data)
		if err != nil {
			// Data exceeds MaxFileSize, which is expected for large fuzz inputs.
			return
		}

		builder, err := proto.NewDataBuilder(first)
		require.NoError(t, err)

		var done bool
		for _, chunk := range chunks {
			require.False(t, done)
			done, err = builder.Add(chunk)
			require.NoError(t, err)
		}

		if len(chunks) > 0 {
			require.True(t, done)
		}

		finalData, err := builder.Complete()
		require.NoError(t, err)
		require.Equal(t, data, finalData)
	})
}

// TestBytesToDataUpload tests the BytesToDataUpload function and the DataBuilder
// with large random data uploads.