mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-05 02:12:48 +08:00
1427825133
issue: #44369 woodpecker related[ issue: #59](https://github.com/zilliztech/woodpecker/issues/59) Refactor the WAL retention logic in Milvus StreamingNode: - Remove the simple sampling-based truncation mechanism. - After flush, WAL data is directly truncated. - The retention control is now delegated to the underlying message queue (MQ) implementation. Signed-off-by: tinswzy <zhenyuan.wei@zilliz.com>
52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package recovery
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/milvus-io/milvus/pkg/v2/util/paramtable"
|
|
)
|
|
|
|
func TestNewConfig(t *testing.T) {
|
|
// Mock paramtable values
|
|
paramtable.Init()
|
|
cfg := newConfig()
|
|
|
|
assert.Equal(t, 10*time.Second, cfg.persistInterval)
|
|
assert.Equal(t, 100, cfg.maxDirtyMessages)
|
|
assert.Equal(t, 3*time.Second, cfg.gracefulTimeout)
|
|
}
|
|
|
|
func TestConfigValidate(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
persistInterval time.Duration
|
|
maxDirtyMessages int
|
|
gracefulTimeout time.Duration
|
|
expectError bool
|
|
}{
|
|
{"ValidConfig", 10 * time.Second, 100, 5 * time.Second, false},
|
|
{"InvalidPersistInterval", 0, 100, 5 * time.Second, true},
|
|
{"InvalidMaxDirtyMessages", 10 * time.Second, 0, 5 * time.Second, true},
|
|
{"InvalidGracefulTimeout", 10 * time.Second, 100, 0, true},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
cfg := &config{
|
|
persistInterval: tt.persistInterval,
|
|
maxDirtyMessages: tt.maxDirtyMessages,
|
|
gracefulTimeout: tt.gracefulTimeout,
|
|
}
|
|
err := cfg.validate()
|
|
if tt.expectError {
|
|
assert.Error(t, err)
|
|
} else {
|
|
assert.NoError(t, err)
|
|
}
|
|
})
|
|
}
|
|
}
|