test: Add test for the --tla-* and --ext-* flags parsing.

cmd/tk/flags_test.go

@@ -0,0 +1,34 @@
+package main
+
+import (
+	"testing"
+
+	"github.com/spf13/pflag"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCliCodeParser(t *testing.T) {
+	fs := pflag.NewFlagSet("test-cli-code-parser", pflag.ContinueOnError)
+	parseExt, parseTLA := cliCodeParser(fs)
+	err := fs.Parse([]string{
+		"--ext-str", "es=1a \" \U0001f605 ' b\nc\u010f",
+		"--tla-str", "ts=2a \" \U0001f605 ' b\nc\u010f",
+		"--ext-code", "ec=1+2",
+		"--tla-code", "tc=2+3",
+		"-A", "ts2=ts2", // tla-str
+		"-V", "es2=es2", // ext-str
+	})
+	assert.NoError(t, err)
+	ext := parseExt()
+	assert.Equal(t, map[string]string{
+		"es":  `"1a \" ` + "\U0001f605" + ` ' b\nc` + "\u010f" + `"`,
+		"ec":  "1+2",
+		"es2": `"es2"`,
+	}, ext)
+	tla := parseTLA()
+	assert.Equal(t, map[string]string{
+		"ts":  `"2a \" ` + "\U0001f605" + ` ' b\nc` + "\u010f" + `"`,
+		"tc":  "2+3",
+		"ts2": `"ts2"`,
+	}, tla)
+}