fix test2

This commit is contained in:
chrislu 2024-03-16 23:48:31 -07:00
parent ea2601799c
commit 3c559dc294

View File

@ -13,12 +13,12 @@ import (
func TestNewLogBufferFirstBuffer(t *testing.T) {
flushInterval := time.Second
lb := NewLogBuffer("test", flushInterval, func(startTime, stopTime time.Time, buf []byte) {
lb := NewLogBuffer("test", flushInterval, func(logBuffer *LogBuffer, startTime time.Time, stopTime time.Time, buf []byte) {
fmt.Printf("flush from %v to %v %d bytes\n", startTime, stopTime, len(buf))
}, nil, func() {
})
startTime := time.Now()
startTime := MessagePosition{Time:time.Now()}
messageSize := 1024
messageCount := 5000
@ -31,13 +31,13 @@ func TestNewLogBufferFirstBuffer(t *testing.T) {
lastProcessedTime, isDone, err := lb.LoopProcessLogData("test", startTime, 0, func() bool {
// stop if no more messages
return receivedMessageCount < messageCount
}, func(logEntry *filer_pb.LogEntry) error {
}, func(logEntry *filer_pb.LogEntry) (isDone bool, err error) {
receivedMessageCount++
if receivedMessageCount >= messageCount {
println("processed all messages")
return io.EOF
return true, io.EOF
}
return nil
return false,nil
})
fmt.Printf("before flush: sent %d received %d\n", messageCount, receivedMessageCount)