github.com/ecadlabs/pretty@v0.0.0-20230412123216-0f3d25fb750b/formatter_test.go

package pretty

import (
	"fmt"
	"io"
	"strings"
	"testing"
	"time"
	"unsafe"
)

type test struct {
	v   interface{}
	s   string
	opt []Option
}

type passtest struct {
	v    interface{}
	f, s string
}

type LongStructTypeName struct {
	longFieldName      interface{}
	otherLongFieldName interface{}
}

type SA struct {
	t *T
	v T
}

type T struct {
	x, y int
}

type F int

func (f F) Format(s fmt.State, c rune) {
	fmt.Fprintf(s, "F(%d)", int(f))
}

type Stringer struct {
	//lint:ignore U1000 test only
	i int
}

func (s *Stringer) String() string { return "foo" }

var long = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"

var passthrough = []passtest{
	{1, "%d", "1"},
	{"a", "%s", "a"},
	{&Stringer{}, "%s", "foo"},
}

func TestPassthrough(t *testing.T) {
	for _, tt := range passthrough {
		s := fmt.Sprintf(tt.f, Formatter(tt.v))
		if tt.s != s {
			t.Errorf("expected %q", tt.s)
			t.Errorf("got      %q", s)
			t.Errorf("expraw\n%s", tt.s)
			t.Errorf("gotraw\n%s", s)
		}
	}
}

type StructWithPrivateFields struct {
	A string
	b string
}

func NewStructWithPrivateFields(a string) StructWithPrivateFields {
	return StructWithPrivateFields{a, "fixedb"}
}

func (s StructWithPrivateFields) GoString() string {
	return fmt.Sprintf("NewStructWithPrivateFields(%q)", s.A)
}

var gosyntax = []test{
	{nil, `nil`, nil},
	{"", `""`, nil},
	{"a", `"a"`, nil},
	{1, "int(1)", nil},
	{1.0, "float64(1)", nil},
	{[]int(nil), "[]int(nil)", nil},
	{[0]int{}, "[0]int{}", nil},
	{complex(1, 0), "(1+0i)", nil},
	//{make(chan int), "(chan int)(0x1234)"},
	{unsafe.Pointer(uintptr(unsafe.Pointer(&long))), fmt.Sprintf("unsafe.Pointer(0x%02x)", uintptr(unsafe.Pointer(&long))), nil},
	{func(int) {}, "func(int) {...}", nil},
	{map[string]string{"a": "a", "b": "b"}, "map[string]string{\"a\":\"a\", \"b\":\"b\"}", nil},
	{map[int]int{1: 1}, "map[int]int{1:1}", nil},
	{int32(1), "int32(1)", nil},
	{io.EOF, `&errors.errorString{s:"EOF"}`, nil},
	{[]string{"a"}, `[]string{"a"}`, nil},
	{
		[]string{long},
		`[]string{"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"}`,
		nil,
	},
	{F(5), "pretty.F(5)", nil},
	{NewStructWithPrivateFields("foo"), `NewStructWithPrivateFields("foo")`, nil},
	{
		SA{&T{1, 2}, T{3, 4}},
		`pretty.SA{
    t:  &pretty.T{x:1, y:2},
    v:  pretty.T{x:3, y:4},
}`,
		nil,
	},
	{
		map[int][]byte{1: {}},
		`map[int][]uint8{
    1:  {},
}`,
		nil,
	},
	{
		map[int]T{1: {}},
		`map[int]pretty.T{
    1:  {},
}`,
		nil,
	},
	{
		long,
		`"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"`,
		nil,
	},
	{
		LongStructTypeName{
			longFieldName:      LongStructTypeName{},
			otherLongFieldName: long,
		},
		`pretty.LongStructTypeName{
    longFieldName:      pretty.LongStructTypeName{},
    otherLongFieldName: "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789",
}`,
		nil,
	},
	{
		&LongStructTypeName{
			longFieldName:      &LongStructTypeName{},
			otherLongFieldName: (*LongStructTypeName)(nil),
		},
		`&pretty.LongStructTypeName{
    longFieldName:      &pretty.LongStructTypeName{},
    otherLongFieldName: (*pretty.LongStructTypeName)(nil),
}`,
		nil,
	},
	{
		[]LongStructTypeName{
			{nil, nil},
			{3, 3},
			{long, nil},
		},
		`[]pretty.LongStructTypeName{
    {},
    {
        longFieldName:      int(3),
        otherLongFieldName: int(3),
    },
    {
        longFieldName:      "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789",
        otherLongFieldName: nil,
    },
}`,
		nil,
	},
	{
		[]interface{}{
			LongStructTypeName{nil, nil},
			[]byte{1, 2, 3},
			T{3, 4},
			LongStructTypeName{long, nil},
		},
		`[]interface {}{
    pretty.LongStructTypeName{},
    []uint8{0x1, 0x2, 0x3},
    pretty.T{x:3, y:4},
    pretty.LongStructTypeName{
        longFieldName:      "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789",
        otherLongFieldName: nil,
    },
}`,
		nil,
	},
	{(*time.Time)(nil), "(*time.Time)(nil)", nil},
	{&ValueGoString{"vgs"}, `VGS vgs`, nil},
	{&ValueGoString{"vgs"}, `&pretty.ValueGoString{s:"vgs"}`, []Option{OptGoStringer(false)}},
	{&ValueGoString{"vgs"}, `"VS vgs"`, []Option{OptGoStringer(false), OptStringer(true)}},
	{&ValueGoString{"vgs"}, `"VT vgs"`, []Option{OptGoStringer(false), OptTextMarshaler(true)}},
	{(*ValueGoString)(nil), `(*pretty.ValueGoString)(nil)`, nil},
	{(*VGSWrapper)(nil), `(*pretty.VGSWrapper)(nil)`, nil},
	{&PointerGoString{"pgs"}, `PGS pgs`, nil},
	{(*PointerGoString)(nil), "(*pretty.PointerGoString)(nil)", nil},
	{&PanicGoString{"oops!"}, `(*pretty.PanicGoString)(PANIC=calling method "GoString": oops!)`, nil},
}

type ValueGoString struct {
	s string
}

func (g ValueGoString) GoString() string {
	return "VGS " + g.s
}

func (g ValueGoString) String() string {
	return "VS " + g.s
}

func (g ValueGoString) MarshalText() (text []byte, err error) {
	return []byte("VT " + g.s), nil
}

type VGSWrapper struct {
	ValueGoString
}

type PointerGoString struct {
	s string
}

func (g *PointerGoString) GoString() string {
	return "PGS " + g.s
}

type PanicGoString struct {
	s string
}

func (g *PanicGoString) GoString() string {
	panic(g.s)
}

func TestGoSyntax(t *testing.T) {
	for _, tt := range gosyntax {
		s := fmt.Sprintf("%# v", Formatter(tt.v, tt.opt...))
		if tt.s != s {
			t.Errorf("expected %q", tt.s)
			t.Errorf("got      %q", s)
			t.Errorf("expraw\n%s", tt.s)
			t.Errorf("gotraw\n%s", s)
		}
	}
}

type I struct {
	i int
	R interface{}
}

func (i *I) I() *I { return i.R.(*I) }

func TestCycle(t *testing.T) {
	type A struct{ *A }
	v := &A{}
	v.A = v

	// panics from stack overflow without cycle detection
	t.Logf("Example cycle:\n%# v", Formatter(v))

	p := &A{}
	s := fmt.Sprintf("%# v", Formatter([]*A{p, p}))
	if strings.Contains(s, "CYCLIC") {
		t.Errorf("Repeated address detected as cyclic reference:\n%s", s)
	}

	type R struct {
		i int
		*R
	}
	r := &R{
		i: 1,
		R: &R{
			i: 2,
			R: &R{
				i: 3,
			},
		},
	}
	r.R.R.R = r
	t.Logf("Example longer cycle:\n%# v", Formatter(r))

	r = &R{
		i: 1,
		R: &R{
			i: 2,
			R: &R{
				i: 3,
				R: &R{
					i: 4,
					R: &R{
						i: 5,
						R: &R{
							i: 6,
							R: &R{
								i: 7,
								R: &R{
									i: 8,
									R: &R{
										i: 9,
										R: &R{
											i: 10,
											R: &R{
												i: 11,
											},
										},
									},
								},
							},
						},
					},
				},
			},
		},
	}
	// here be pirates
	r.R.R.R.R.R.R.R.R.R.R.R = r
	t.Logf("Example very long cycle:\n%# v", Formatter(r))

	i := &I{
		i: 1,
		R: &I{
			i: 2,
			R: &I{
				i: 3,
				R: &I{
					i: 4,
					R: &I{
						i: 5,
						R: &I{
							i: 6,
							R: &I{
								i: 7,
								R: &I{
									i: 8,
									R: &I{
										i: 9,
										R: &I{
											i: 10,
											R: &I{
												i: 11,
											},
										},
									},
								},
							},
						},
					},
				},
			},
		},
	}
	iv := i.I().I().I().I().I().I().I().I().I().I()
	*iv = *i
	t.Logf("Example long interface cycle:\n%# v", Formatter(i))
}