Function TestNewGoKitWithRateLimitedAndFields

log/gokit_test.go:75–102

(t *testing.T)

}

func TestNewGoKitWithRateLimitedAndFields(t *testing.T) {
	buf := bytes.NewBuffer(nil)
	var lvl Level
	require.NoError(t, lvl.Set("info"))
	now := log.DefaultTimestampUTC()
	logger := NewGoKitWithWriter(LogfmtFormat, buf)

	rateLimitedLogger := NewRateLimitedLogger(logger, 1, 2, prometheus.DefaultRegisterer)

	rateLimitedLogger = log.With(rateLimitedLogger, "ts", now, "caller", log.Caller(5))
	rateLimitedLogger = level.NewFilter(rateLimitedLogger, lvl.Option)

	for i := 0; i < 1000; i++ {
		level.Info(rateLimitedLogger).Log("msg", LazySprintf("info %d", i))
		level.Debug(rateLimitedLogger).Log("msg", LazySprintf("debug %d", i))
	}

	format := "ts=%s caller=gokit_test.go:%d level=%s msg=\"info %d\""
	for i := 0; i < 1000; i++ {
		if i < 2 {
			require.True(t, bytes.Contains(buf.Bytes(), []byte(fmt.Sprintf(format, now, 88, "info", i))))
		} else {
			require.False(t, bytes.Contains(buf.Bytes(), []byte(fmt.Sprintf(format, now, 88, "info", i))))
		}

		require.False(t, bytes.Contains(buf.Bytes(), []byte(fmt.Sprintf(format, now, 89, "debug", i))))
	}
}

type lazySprintfWithCount struct {
	next  Sprintf