Example Test (#31)

* cleanup * test
nikolaydubina · Aug 27, 2024 · 25e02f6 · 25e02f6
1 parent a520af3
commit 25e02f6
Show file tree

Hide file tree

Showing 16 changed files with 200 additions and 41 deletions.
diff --git a/enum_test.go.template b/enum_test.go.template
@@ -3,10 +3,28 @@
 package {{.Package}}
 
 import (
+	"fmt"
 	"testing"
 	"errors"
 )
 
+func Example{{.Type}}_{{.EncodeMethod}}() {
+	for _, v := range []{{.Type}}{{{.Values}}} {
+		b, _ := v.{{.EncodeMethod}}()
+		fmt.Printf("%s ", string(b))
+	}
+	// Output: {{.TagsNaked}}
+}
+
+func Example{{.Type}}_{{.DecodeMethod}}() {
+	for _, s := range []string{{{.Tags}}} {
+		var v {{.Type}}
+		if err := (&v).{{.DecodeMethod}}([]byte(s)); err != nil {
+			fmt.Println(err)
+		}
+	}
+}
+
 func Test{{.Type}}_{{.EncodeMethod}}_{{.DecodeMethod}}(t *testing.T) {
 	for _, v := range []{{.Type}}{{{.Values}}} {
 		b, err := v.{{.EncodeMethod}}()

diff --git a/internal/testdata/color.go b/internal/testdata/color.go
@@ -2,7 +2,6 @@ package color
 
 type Color struct{ c uint8 }
 
-//go:generate go-enum-encoding -type=Color
 var (
 	UndefinedColor = Color{}            // json:""
 	Red            = Color{1}           // json:"red"
@@ -17,7 +16,6 @@ type V struct {
 
 type ColorString int
 
-//go:generate go-enum-encoding -type=ColorString -string
 const (
 	RedS   ColorString = iota // json:"red"
 	GreenS                    // json:"green"

diff --git a/internal/testdata/currency.go b/internal/testdata/currency.go
@@ -2,7 +2,6 @@ package color
 
 type Currency struct{ c uint8 }
 
-//go:generate go-enum-encoding -type=Currency
 var (
 	UndefinedCurrency = Currency{}   // json:""
 	SGD               = Currency{1}  // json:"SGD"

diff --git a/internal/testdata/currency_string.go b/internal/testdata/currency_string.go
@@ -2,7 +2,6 @@ package color
 
 type CurrencyString struct{ c uint8 }
 
-//go:generate go-enum-encoding -type=CurrencyString -string
 var (
 	UndefinedCurrencyS = CurrencyString{}   // json:""
 	SGDS               = CurrencyString{1}  // json:"SGD"

diff --git a/internal/testdata/currency_string_custom.go b/internal/testdata/currency_string_custom.go
@@ -2,7 +2,6 @@ package color
 
 type CurrencyStringCustom struct{ c uint8 }
 
-//go:generate go-enum-encoding -type=CurrencyStringCustom -string -encode-method=MarshalTextName -decode-method=UnmarshalTextName -string-method=StringName
 var (
 	UndefinedCurrencySC = CurrencyStringCustom{}   // json:""
 	SGDSC               = CurrencyStringCustom{1}  // json:"SGD"

diff --git a/internal/testdata/exp/color2_enum_encoding_test.go b/internal/testdata/exp/color2_enum_encoding_test.go
diff --git a/internal/testdata/exp/color_enum_encoding_test.go b/internal/testdata/exp/color_enum_encoding_test.go
diff --git a/internal/testdata/exp/colorstring_enum_encoding_test.go b/internal/testdata/exp/colorstring_enum_encoding_test.go
diff --git a/internal/testdata/exp/currency2_enum_encoding_test.go b/internal/testdata/exp/currency2_enum_encoding_test.go
diff --git a/internal/testdata/exp/currency_enum_encoding_test.go b/internal/testdata/exp/currency_enum_encoding_test.go
diff --git a/internal/testdata/exp/currencystring_enum_encoding_test.go b/internal/testdata/exp/currencystring_enum_encoding_test.go
diff --git a/internal/testdata/exp/currencystringcustom_enum_encoding_test.go b/internal/testdata/exp/currencystringcustom_enum_encoding_test.go
diff --git a/internal/testdata/multiple.go b/internal/testdata/multiple.go
@@ -2,7 +2,6 @@ package color
 
 type Color2 struct{ c uint8 }
 
-//go:generate go-enum-encoding -type=Color2
 var (
 	UndefinedColor2  = Color2{}             // json:""
 	Red2             = Color2{1}            // json:"red"
@@ -11,7 +10,6 @@ var (
 
 type Currency2 uint8
 
-//go:generate go-enum-encoding -type=Currency2
 const (
 	UndefCurrency2 Currency2 = iota // json:""
 	SGD2                            // json:"SGD"

diff --git a/internal/testdata/size.go b/internal/testdata/size.go
@@ -2,7 +2,6 @@ package color
 
 type Size uint8
 
-//go:generate go-enum-encoding -type=Size
 const (
 	UndefinedSize Size = iota // json:""
 	Small                     // json:"small"

diff --git a/main.go b/main.go
@@ -160,6 +160,7 @@ func process(typeName, fileName, lineNum, packageName, mode, encodeMethod, decod
 		With("{{.StringMethod}}", stringMethod).
 		WithMap("{{.Values}}", func(_ int, v [2]string) string { return v[0] }, ", ").
 		WithMap("{{.Tags}}", func(_ int, v [2]string) string { return `"` + v[1] + `"` }, ",").
+		WithMap("{{.TagsNaked}}", func(_ int, v [2]string) string { return v[1] }, " ").
 		WithMap("{{.seq_bytes}}", func(_ int, v [2]string) string { return `[]byte("` + v[1] + `")` }, ", ").
 		WithMap("{{.seq_string}}", func(_ int, v [2]string) string { return `"` + v[1] + `"` }, ", ")