113 lines
3.2 KiB
Go
113 lines
3.2 KiB
Go
package logstash
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
jww "github.com/spf13/jwalterweatherman"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
var (
|
|
s1 = "[test1 ] TRACE test1"
|
|
s2 = "[test2 ] ERROR test2"
|
|
s3 = "[test1 ] TRACE test3"
|
|
)
|
|
|
|
func TestLog(t *testing.T) {
|
|
log := New(3)
|
|
|
|
// old to new
|
|
log.Write([]byte(s1))
|
|
log.Write([]byte(s2))
|
|
log.Write([]byte(s3))
|
|
|
|
trace, other := log.trace.data, log.other.data
|
|
|
|
assert.Equal(t, []string{s1, s2, s3}, log.All(nil, jww.LevelTrace, 0))
|
|
assert.Equal(t, []string{s1, s2, s3}, log.All([]string{}, jww.LevelTrace, 0))
|
|
|
|
assert.Equal(t, []string{s1, s3}, log.All([]string{"test1"}, jww.LevelTrace, 0))
|
|
assert.Equal(t, []string{s1, s2, s3}, log.All(nil, jww.LevelTrace, 0))
|
|
assert.Equal(t, []string{s3}, log.All(nil, jww.LevelTrace, 1))
|
|
|
|
assert.Equal(t, []string{}, log.All(nil, jww.LevelFatal, 0))
|
|
|
|
assert.Same(t, trace, log.trace.data, "data should not be changed after All() call")
|
|
assert.Same(t, other, log.other.data, "data should not be changed after All() call")
|
|
assert.Equal(t, []string{"test1", "test2"}, log.Areas())
|
|
}
|
|
|
|
// TestTraceBudget verifies that a flood of trace lines cannot evict non-trace
|
|
// entries, which is what made the log view show mqtt/httpd only (#32673).
|
|
func TestTraceBudget(t *testing.T) {
|
|
log := New(3)
|
|
|
|
log.Write([]byte(s2))
|
|
for i := range 100 {
|
|
log.Write([]byte(fmt.Sprintf("[mqtt ] TRACE send %d", i)))
|
|
}
|
|
|
|
assert.Equal(t, []string{s2}, log.All(nil, jww.LevelDebug, 0))
|
|
assert.Equal(t, []string{"mqtt", "test2"}, log.Areas())
|
|
}
|
|
|
|
// TestRingGrowsThenCaps writes more lines than the configured size and verifies
|
|
// the ring grows up to size and then keeps only the most recent size entries.
|
|
func TestRingGrowsThenCaps(t *testing.T) {
|
|
const size = 3
|
|
log := New(size)
|
|
|
|
lines := []string{
|
|
"[test ] TRACE line1",
|
|
"[test ] TRACE line2",
|
|
"[test ] TRACE line3",
|
|
"[test ] TRACE line4",
|
|
"[test ] TRACE line5",
|
|
}
|
|
for _, l := range lines {
|
|
log.Write([]byte(l))
|
|
}
|
|
|
|
// only the last `size` lines survive, in chronological order
|
|
got := log.All(nil, jww.LevelTrace, 0)
|
|
assert.Len(t, got, size, "ring must not exceed configured size")
|
|
assert.Equal(t, lines[len(lines)-size:], got)
|
|
}
|
|
|
|
// TestRingSkipCacheLinesDoesNotGrow verifies that "[cache ]" lines take the
|
|
// early-return path and neither grow the ring nor the length counter.
|
|
func TestRingSkipCacheLinesDoesNotGrow(t *testing.T) {
|
|
log := New(3)
|
|
|
|
// grow the ring past a single node first, so a stray cursor advance on the
|
|
// cache path would actually be observable
|
|
log.Write([]byte(s1))
|
|
log.Write([]byte(s2))
|
|
|
|
stored := log.All(nil, jww.LevelTrace, 0)
|
|
lenBefore := log.trace.length
|
|
cursorBefore := log.trace.data
|
|
|
|
for range 10 {
|
|
log.Write([]byte("[cache ] cache line"))
|
|
}
|
|
|
|
assert.Equal(t, stored, log.All(nil, jww.LevelTrace, 0), "cache lines must not change stored content")
|
|
assert.Equal(t, log.trace.data.Len(), log.trace.length, "length counter must stay in sync with the ring")
|
|
assert.Equal(t, lenBefore, log.trace.length, "cache lines must not grow the ring")
|
|
assert.Same(t, cursorBefore, log.trace.data, "cache lines must not advance the write cursor")
|
|
}
|
|
|
|
func BenchmarkLog(b *testing.B) {
|
|
log := New(10000)
|
|
|
|
// old to new
|
|
log.Write([]byte(s1))
|
|
log.Write([]byte(s2))
|
|
log.Write([]byte(s3))
|
|
|
|
for b.Loop() {
|
|
log.All(nil, jww.LevelTrace, 1)
|
|
}
|
|
}
|