From 10a781d22ceb921e1aaad91e27c80ed10e218640 Mon Sep 17 00:00:00 2001 From: Zhen Ye Date: Tue, 2 Dec 2025 10:25:10 +0800 Subject: [PATCH] fix: write ahead buffer unittest failure (#45978) issue: #45977 Signed-off-by: chyezh --- .../server/wal/interceptors/wab/write_ahead_buffer.go | 1 + .../server/wal/interceptors/wab/write_ahead_buffer_test.go | 5 ++++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/internal/streamingnode/server/wal/interceptors/wab/write_ahead_buffer.go b/internal/streamingnode/server/wal/interceptors/wab/write_ahead_buffer.go index 71dc80afd6..e01ca052c7 100644 --- a/internal/streamingnode/server/wal/interceptors/wab/write_ahead_buffer.go +++ b/internal/streamingnode/server/wal/interceptors/wab/write_ahead_buffer.go @@ -98,6 +98,7 @@ func (w *WriteAheadBuffer) ReadFromExclusiveTimeTick(ctx context.Context, timeti } return &WriteAheadBufferReader{ nextOffset: nextOffset, + lastTimeTick: timetick, snapshot: snapshot, underlyingBuf: w, }, nil diff --git a/internal/streamingnode/server/wal/interceptors/wab/write_ahead_buffer_test.go b/internal/streamingnode/server/wal/interceptors/wab/write_ahead_buffer_test.go index 5f4a2b0c13..02927dd6b6 100644 --- a/internal/streamingnode/server/wal/interceptors/wab/write_ahead_buffer_test.go +++ b/internal/streamingnode/server/wal/interceptors/wab/write_ahead_buffer_test.go @@ -104,6 +104,9 @@ func TestWriteAheadBuffer(t *testing.T) { } } }() + if rand.Int31n(2) == 0 { + time.Sleep(20 * time.Millisecond) + } r1, err := wb.ReadFromExclusiveTimeTick(context.Background(), 1) assert.NoError(t, err) assert.NotNil(t, r1) @@ -130,7 +133,7 @@ func TestWriteAheadBuffer(t *testing.T) { // Read from half of the timetick <-ch - assert.Equal(t, totalCnt, len(timeticks)) + assert.Equal(t, totalCnt, len(timeticks)+1) targetTimeTickIdx := len(timeticks) / 2 for targetTimeTickIdx < len(timeticks) && timeticks[targetTimeTickIdx+1] == timeticks[targetTimeTickIdx] {