Function TestCallerFormatter

options_test.go:30–75

(t *testing.T)

}

func TestCallerFormatter(t *testing.T) {
	var buf bytes.Buffer
	l := NewWithOptions(colorprofile.NewWriter(&buf, os.Environ()), Options{ReportCaller: true})
	frames := l.frames(0)
	frame, _ := frames.Next()
	file, line, fn := frame.File, frame.Line, frame.Function
	hi := func() { l.Info("hi") }
	cases := []struct {
		name     string
		expected string
		format   CallerFormatter
	}{
		{
			name:     "short caller formatter",
			expected: fmt.Sprintf("INFO <log/options_test.go:%d> hi\n", line+3),
			format:   ShortCallerFormatter,
		},
		{
			name:     "long caller formatter",
			expected: fmt.Sprintf("INFO <%s:%d> hi\n", file, line+3),
			format:   LongCallerFormatter,
		},
		{
			name:     "foo caller formatter",
			expected: "INFO <foo> hi\n",
			format: func(file string, line int, fn string) string {
				return "foo"
			},
		},
		{
			name:     "custom caller formatter",
			expected: fmt.Sprintf("INFO <%s:%d:%s.func1> hi\n", file, line+3, fn),
			format: func(file string, line int, fn string) string {
				return fmt.Sprintf("%s:%d:%s", file, line, fn)
			},
		},
	}
	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			buf.Reset()
			l.callerFormatter = c.format
			hi()
			require.Equal(t, c.expected, buf.String())
		})
	}
}