test: Add fuzz test

cmd/cmd_test.go

@@ -1,15 +1,18 @@
 package cmd
 
 import (
+	"encoding/base64"
 	"io"
 	"os"
 	"strings"
 	"testing"
+	"unicode/utf8"
 
 	"github.com/clevyr/yampl/internal/config"
 	"github.com/spf13/cobra"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
+	"gopkg.in/yaml.v3"
 )
 
 func Test_run(t *testing.T) {
@@ -183,3 +186,36 @@ func Test_openAndTemplateFile(t *testing.T) {
 		})
 	}
 }
+
+func Fuzz_templateReader(f *testing.F) {
+	const template = `value: "" #yampl {{ .newVal }}`
+	f.Add("hello world")
+	f.Add("123")
+	f.Add("true")
+
+	f.Fuzz(func(t *testing.T, content string) {
+		conf := config.New()
+		conf.Vars = config.Vars{"newVal": content}
+
+		got, err := templateReader(conf, "", strings.NewReader(template), int64(len(content)))
+		require.NoError(t, err)
+
+		var decoded map[string]any
+		require.NoError(t, yaml.Unmarshal([]byte(got), &decoded))
+
+		switch val := decoded["value"].(type) {
+		case string:
+			if utf8.ValidString(content) {
+				assert.Equal(t, content, val)
+			} else {
+				raw, err := base64.StdEncoding.DecodeString(val)
+				require.NoError(t, err)
+				assert.Equal(t, content, string(raw))
+			}
+		default:
+			var want any
+			require.NoError(t, yaml.Unmarshal([]byte(content), &want))
+			assert.Equal(t, want, val)
+		}
+	})
+}

cmd/testdata/fuzz/Fuzz_templateReader/0dc5ed1e283739f6

@@ -0,0 +1,2 @@
+go test fuzz v1
+string("\xdfA7")

cmd/testdata/fuzz/Fuzz_templateReader/10bb5b2960e772f0

@@ -0,0 +1,2 @@
+go test fuzz v1
+string("\xed")

cmd/testdata/fuzz/Fuzz_templateReader/60c24a059993f7ae

@@ -0,0 +1,2 @@
+go test fuzz v1
+string("010")

cmd/testdata/fuzz/Fuzz_templateReader/64f100070ec426a6

@@ -0,0 +1,2 @@
+go test fuzz v1
+string("0_")

cmd/testdata/fuzz/Fuzz_templateReader/771e938e4458e983

@@ -0,0 +1,2 @@
+go test fuzz v1
+string("0")

cmd/testdata/fuzz/Fuzz_templateReader/906bad78e0106290

@@ -0,0 +1,2 @@
+go test fuzz v1
+string("0._0")

cmd/testdata/fuzz/Fuzz_templateReader/93076e9d6b6510b6

@@ -0,0 +1,2 @@
+go test fuzz v1
+string(".0")

cmd/testdata/fuzz/Fuzz_templateReader/a941c4713b3a921d

@@ -0,0 +1,2 @@
+go test fuzz v1
+string("01000e80")

cmd/testdata/fuzz/Fuzz_templateReader/db7ae63e0377da75

@@ -0,0 +1,2 @@
+go test fuzz v1
+string("~")

cmd/testdata/fuzz/Fuzz_templateReader/def578230616f8b9

@@ -0,0 +1,2 @@
+go test fuzz v1
+string("\xff")

cmd/testdata/fuzz/Fuzz_templateReader/e222fa9b3faf0bc5

@@ -0,0 +1,2 @@
+go test fuzz v1
+string("\"\"\"\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17\x17")

cmd/testdata/fuzz/Fuzz_templateReader/ec1cb26f1a43547c

@@ -0,0 +1,2 @@
+go test fuzz v1
+string("    ")

cmd/testdata/fuzz/Fuzz_templateReader/fe0e339227bff527

@@ -0,0 +1,2 @@
+go test fuzz v1
+string("\t\n")