add test

Author: kkHAIKE

encode_test.go

@@ -7,6 +7,7 @@ import (
 	"math"
 	"net"
 	"os"
+	"strconv"
 	"strings"
 	"testing"
 	"time"
@@ -399,11 +400,13 @@ type (
 	food  struct{ F []string }
 	fun   func()
 	cplx  complex128
+	ints  []int
 
 	sound2 struct{ S string }
 	food2  struct{ F []string }
 	fun2   func()
 	cplx2  complex128
+	ints2  []int
 )
 
 // This is intentionally wrong (pointer receiver)
@@ -415,6 +418,26 @@ func (c cplx) MarshalText() ([]byte, error) {
 	return []byte(fmt.Sprintf("(%f+%fi)", real(cplx), imag(cplx))), nil
 }
 
+func intsValue(is []int) []byte {
+	var buf bytes.Buffer
+	buf.WriteByte('<')
+	for i, v := range is {
+		if i > 0 {
+			buf.WriteByte(',')
+		}
+		buf.WriteString(strconv.Itoa(v))
+	}
+	buf.WriteByte('>')
+	return buf.Bytes()
+}
+
+func (is *ints) MarshalText() ([]byte, error) {
+	if is == nil {
+		return []byte("[]"), nil
+	}
+	return intsValue(*is), nil
+}
+
 func (s *sound2) MarshalTOML() ([]byte, error) { return []byte("\"" + s.S + "\""), nil }
 func (f food2) MarshalTOML() ([]byte, error) {
 	return []byte("[\"" + strings.Join(f.F, "\", \"") + "\"]"), nil
@@ -424,6 +447,13 @@ func (c cplx2) MarshalTOML() ([]byte, error) {
 	cplx := complex128(c)
 	return []byte(fmt.Sprintf("\"(%f+%fi)\"", real(cplx), imag(cplx))), nil
 }
+func (is *ints2) MarshalTOML() ([]byte, error) {
+	// MarshalTOML must quote by self
+	if is == nil {
+		return []byte(`"[]"`), nil
+	}
+	return []byte(fmt.Sprintf(`"%s"`, intsValue(*is))), nil
+}
 
 func TestEncodeTextMarshaler(t *testing.T) {
 	x := struct {
@@ -435,6 +465,8 @@ func TestEncodeTextMarshaler(t *testing.T) {
 		Food2   *food
 		Complex cplx
 		Fun     fun
+		Ints    ints
+		Ints2   *ints2
 	}{
 		Name:   "Goblok",
 		Sound:  sound{"miauw"},
@@ -447,26 +479,28 @@ func TestEncodeTextMarshaler(t *testing.T) {
 		Food2:   &food{[]string{"chicken", "fish"}},
 		Complex: complex(42, 666),
 		Fun:     func() { panic("x") },
+		Ints:    ints{1, 2, 3, 4},
+		Ints2:   &ints2{1, 2, 3, 4},
 	}
 
 	var buf bytes.Buffer
-	if err := NewEncoder(&buf).Encode(x); err != nil {
+	if err := NewEncoder(&buf).Encode(&x); err != nil {
 		t.Fatal(err)
 	}
 
 	want := `Name = "Goblok"
+Sound = "miauw"
 Sound2 = "miauw"
 Food = "chicken, fish"
 Food2 = "chicken, fish"
 Complex = "(42.000000+666.000000i)"
 Fun = "why would you do this?"
+Ints = "<1,2,3,4>"
+Ints2 = "<1,2,3,4>"
 
 [Labels]
   color = "black"
   type = "cat"
-
-[Sound]
-  S = "miauw"
 `
 
 	if buf.String() != want {