diff options
Diffstat (limited to 'weed/util/log_buffer/log_buffer_test.go')
| -rw-r--r-- | weed/util/log_buffer/log_buffer_test.go | 24 |
1 files changed, 16 insertions, 8 deletions
diff --git a/weed/util/log_buffer/log_buffer_test.go b/weed/util/log_buffer/log_buffer_test.go index 7b851de06..d99a8f20c 100644 --- a/weed/util/log_buffer/log_buffer_test.go +++ b/weed/util/log_buffer/log_buffer_test.go @@ -52,11 +52,13 @@ func TestNewLogBufferFirstBuffer(t *testing.T) { var buf = make([]byte, messageSize) for i := 0; i < messageCount; i++ { rand.Read(buf) - lb.AddToBuffer(&mq_pb.DataMessage{ + if err := lb.AddToBuffer(&mq_pb.DataMessage{ Key: nil, Value: buf, TsNs: 0, - }) + }); err != nil { + t.Fatalf("Failed to add buffer: %v", err) + } } wg.Wait() @@ -141,12 +143,14 @@ func TestReadFromBuffer_OldOffsetReturnsResumeFromDiskError(t *testing.T) { if tt.hasData { testData := []byte("test message") // Use AddLogEntryToBuffer to preserve offset information - lb.AddLogEntryToBuffer(&filer_pb.LogEntry{ + if err := lb.AddLogEntryToBuffer(&filer_pb.LogEntry{ TsNs: time.Now().UnixNano(), Key: []byte("key"), Data: testData, Offset: tt.currentOffset, // Add data at current offset - }) + }); err != nil { + t.Fatalf("Failed to add log entry: %v", err) + } } // Create an offset-based position for the requested offset @@ -365,11 +369,13 @@ func TestReadFromBuffer_InitializedFromDisk(t *testing.T) { lb.offset, lb.bufferStartOffset) // Now write a new message at offset 4 - lb.AddToBuffer(&mq_pb.DataMessage{ + if err := lb.AddToBuffer(&mq_pb.DataMessage{ Key: []byte("new-key"), Value: []byte("new-message-at-offset-4"), TsNs: time.Now().UnixNano(), - }) + }); err != nil { + t.Fatalf("Failed to add buffer: %v", err) + } // After AddToBuffer: offset=5, pos>0 // Schema Registry tries to read offset 0 (should be on disk) @@ -503,11 +509,13 @@ func TestLoopProcessLogDataWithOffset_DiskReadRetry(t *testing.T) { // Now add data and flush it t.Logf("➕ Adding message to buffer...") - logBuffer.AddToBuffer(&mq_pb.DataMessage{ + if err := logBuffer.AddToBuffer(&mq_pb.DataMessage{ Key: []byte("key-0"), Value: []byte("message-0"), TsNs: time.Now().UnixNano(), - }) + }); err != nil { + t.Fatalf("Failed to add buffer: %v", err) + } // Force flush t.Logf("Force flushing...") |
