consolidate marshaler tests

type_tests/marshaler_string_test.go

@@ -0,0 +1,52 @@
+package test
+
+import (
+	"bytes"
+	"encoding/base64"
+	"strings"
+	"encoding/json"
+)
+
+type stringMarshaler string
+
+func (m stringMarshaler) encode(str string) string {
+	buf := bytes.Buffer{}
+	b64 := base64.NewEncoder(base64.StdEncoding, &buf)
+	if _, err := b64.Write([]byte(str)); err != nil {
+		panic(err)
+	}
+	if err := b64.Close(); err != nil {
+		panic(err)
+	}
+	return buf.String()
+}
+
+func (m stringMarshaler) decode(str string) string {
+	if len(str) == 0 {
+		return ""
+	}
+	b64 := base64.NewDecoder(base64.StdEncoding, strings.NewReader(str))
+	bs := make([]byte, len(str))
+	if n, err := b64.Read(bs); err != nil {
+		panic(err)
+	} else {
+		bs = bs[:n]
+	}
+	return string(bs)
+}
+
+func (m stringMarshaler) MarshalJSON() ([]byte, error) {
+	return []byte(`"MANUAL__` + m.encode(string(m)) + `"`), nil
+}
+
+func (m *stringMarshaler) UnmarshalJSON(text []byte) error {
+	*m = stringMarshaler(m.decode(strings.TrimPrefix(strings.Trim(string(text), `"`), "MANUAL__")))
+	return nil
+}
+
+var _ json.Marshaler = *new(stringMarshaler)
+var _ json.Unmarshaler = new(stringMarshaler)
+
+func init() {
+	testCases = append(testCases, (*stringMarshaler)(nil))
+}

type_tests/marshaler_struct_test.go

@@ -0,0 +1,69 @@
+package test
+
+import (
+	"strings"
+	"encoding/base64"
+	"bytes"
+	"encoding/json"
+)
+
+type structMarshaler struct {
+	X string
+}
+
+func (m structMarshaler) encode(str string) string {
+	buf := bytes.Buffer{}
+	b64 := base64.NewEncoder(base64.StdEncoding, &buf)
+	if _, err := b64.Write([]byte(str)); err != nil {
+		panic(err)
+	}
+	if err := b64.Close(); err != nil {
+		panic(err)
+	}
+	return buf.String()
+}
+
+func (m structMarshaler) decode(str string) string {
+	if len(str) == 0 {
+		return ""
+	}
+	b64 := base64.NewDecoder(base64.StdEncoding, strings.NewReader(str))
+	bs := make([]byte, len(str))
+	if n, err := b64.Read(bs); err != nil {
+		panic(err)
+	} else {
+		bs = bs[:n]
+	}
+	return string(bs)
+}
+
+func (m structMarshaler) MarshalJSON() ([]byte, error) {
+	return []byte(`"MANUAL__` + m.encode(m.X) + `"`), nil
+}
+
+func (m *structMarshaler) UnmarshalJSON(text []byte) error {
+	m.X = m.decode(strings.TrimPrefix(strings.Trim(string(text), `"`), "MANUAL__"))
+	return nil
+}
+
+var _ json.Marshaler = structMarshaler{}
+var _ json.Unmarshaler = &structMarshaler{}
+
+type structMarshalerAlias structMarshaler
+
+func init() {
+	testCases = append(testCases,
+		(*structMarshaler)(nil),
+		(*structMarshalerAlias)(nil),
+		(*struct {
+			S string
+			M structMarshaler
+			I int8
+		})(nil),
+		(*struct {
+			S string
+			M structMarshalerAlias
+			I int8
+		})(nil),
+	)
+}