Function TestLazySprintf

log/gokit_test.go:35–73

(t *testing.T)

}

func TestLazySprintf(t *testing.T) {
	tests := map[string]struct {
		id     int
		lvl    string
		format string
	}{
		"debug level should log debug messages and call Sprintf": {
			id:     1,
			lvl:    "debug",
			format: "debug %d has been logged %v",
		},
		"info level should not log debug messages and should not call Sprintf": {
			id:     2,
			lvl:    "info",
			format: "info %d has not been logged %v",
		},
	}

	buf := bytes.NewBuffer(nil)

	for _, test := range tests {
		buf.Reset()
		logger := NewGoKitWithWriter(LogfmtFormat, buf)
		var lvl Level
		require.NoError(t, lvl.Set(test.lvl))
		logger = level.NewFilter(logger, lvl.Option)
		now := time.Now()
		expectedMessage := fmt.Sprintf(test.format, test.id, now)
		lazySprintf := newLazySprintfWithCount("debug %d has been logged %v", test.id, now)
		level.Debug(logger).Log("msg", lazySprintf)
		if test.lvl == "debug" {
			require.True(t, bytes.Contains(buf.Bytes(), []byte(expectedMessage)))
			require.Equal(t, 1, lazySprintf.count)
		} else {
			require.False(t, bytes.Contains(buf.Bytes(), []byte(expectedMessage)))
			require.Equal(t, 0, lazySprintf.count)
		}
	}
}

func TestNewGoKitWithRateLimitedAndFields(t *testing.T) {
	buf := bytes.NewBuffer(nil)