This is an automated email from the ASF dual-hosted git repository.
danoliveira pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git
The following commit(s) were added to refs/heads/master by this push:
new 341a8364f36 [BEAM-14440] Add basic fuzz tests to the coders package
(#17587)
341a8364f36 is described below
commit 341a8364f361b7d4d5dbd3f8b9c7bf4bc02f52a4
Author: Jack McCluskey <[email protected]>
AuthorDate: Mon May 16 13:46:20 2022 -0400
[BEAM-14440] Add basic fuzz tests to the coders package (#17587)
* UTF8 fuzz test
* Ints fuzz tests
* Bytes fuzz tests
* Double fuzz test
* Move fuzz tests to standalone test file
---
.../pkg/beam/core/graph/coder/coder_fuzz_test.go | 129 +++++++++++++++++++++
1 file changed, 129 insertions(+)
diff --git a/sdks/go/pkg/beam/core/graph/coder/coder_fuzz_test.go
b/sdks/go/pkg/beam/core/graph/coder/coder_fuzz_test.go
new file mode 100644
index 00000000000..11c516f7378
--- /dev/null
+++ b/sdks/go/pkg/beam/core/graph/coder/coder_fuzz_test.go
@@ -0,0 +1,129 @@
+// Licensed to the Apache Software Foundation (ASF) under one or more
+// contributor license agreements. See the NOTICE file distributed with
+// this work for additional information regarding copyright ownership.
+// The ASF licenses this file to You under the Apache License, Version 2.0
+// (the "License"); you may not use this file except in compliance with
+// the License. You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package coder
+
+import (
+ "bytes"
+ "math"
+ "strings"
+ "testing"
+
+ "github.com/google/go-cmp/cmp"
+)
+
+func FuzzEncodeDecodeBytes(f *testing.F) {
+ f.Add([]byte{10})
+ f.Fuzz(func(t *testing.T, b []byte) {
+ var buf bytes.Buffer
+ err := EncodeBytes(b, &buf)
+ if err != nil {
+ return
+ }
+
+ got, err := DecodeBytes(&buf)
+ if err != nil {
+ t.Fatalf("failed to decode bytes, got %v", err)
+ }
+
+ if d := cmp.Diff(got, b); d != "" {
+ t.Errorf("decoded output does not match input: got %v,
want %v", got, b)
+ }
+ })
+}
+
+const floatPrecision = float64(0.001)
+
+func FuzzEncodeDecodeDouble(f *testing.F) {
+ f.Add(float64(3.141))
+ f.Fuzz(func(t *testing.T, a float64) {
+ var buf bytes.Buffer
+ err := EncodeDouble(a, &buf)
+ if err != nil {
+ return
+ }
+
+ actual, err := DecodeDouble(&buf)
+ if err != nil {
+ t.Fatalf("DecodeDouble(%v) failed: %v", buf, err)
+ }
+ if math.Abs(actual-a) > floatPrecision {
+ t.Fatalf("got %f, want %f +/- %f", actual, a,
floatPrecision)
+ }
+ })
+}
+
+func FuzzEncodeDecodeUInt64(f *testing.F) {
+ f.Add(uint64(42))
+ f.Fuzz(func(t *testing.T, b uint64) {
+ var buf bytes.Buffer
+ err := EncodeUint64(b, &buf)
+ if err != nil {
+ return
+ }
+
+ got, err := DecodeUint64(&buf)
+ if err != nil {
+ t.Fatalf("failed to decode bytes, got %v", err)
+ }
+
+ if got != b {
+ t.Errorf("decoded output does not match input: got %v,
want %v", got, b)
+ }
+ })
+}
+
+func FuzzEncodeDecodeInt32(f *testing.F) {
+ f.Add(int32(42))
+ f.Fuzz(func(t *testing.T, b int32) {
+ var buf bytes.Buffer
+ err := EncodeInt32(b, &buf)
+ if err != nil {
+ return
+ }
+
+ got, err := DecodeInt32(&buf)
+ if err != nil {
+ t.Fatalf("failed to decode bytes, got %v", err)
+ }
+
+ if got != b {
+ t.Errorf("decoded output does not match input: got %v,
want %v", got, b)
+ }
+ })
+}
+
+func FuzzEncodeDecodeStringUTF8LP(f *testing.F) {
+ for _, s := range testValues {
+ f.Add(s)
+ }
+ f.Fuzz(func(t *testing.T, b string) {
+ var build strings.Builder
+ err := EncodeStringUTF8(b, &build)
+ if err != nil {
+ return
+ }
+
+ buf := bytes.NewBufferString(build.String())
+ got, err := DecodeStringUTF8(buf)
+ if err != nil {
+ t.Fatalf("failed to decode bytes, got %v", err)
+ }
+
+ if got != b {
+ t.Errorf("decoded output does not match input: got %v,
want %v", got, b)
+ }
+ })
+}
