mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-06 02:42:53 +08:00
e675da76e4
issue: #41544 - simplify the proto message for flush and create segment. - simplify the msg handler for flowgraph. --------- Signed-off-by: chyezh <chyezh@outlook.com>
40 lines
1.2 KiB
Go
40 lines
1.2 KiB
Go
package recovery
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/milvus-io/milvus/pkg/v2/proto/messagespb"
|
|
"github.com/milvus-io/milvus/pkg/v2/proto/streamingpb"
|
|
"github.com/milvus-io/milvus/pkg/v2/streaming/walimpls/impls/rmq"
|
|
)
|
|
|
|
func TestNewWALCheckpointFromProto(t *testing.T) {
|
|
walName := "rocksmq"
|
|
messageID := rmq.NewRmqID(1)
|
|
timeTick := uint64(12345)
|
|
recoveryMagic := int64(1)
|
|
protoCheckpoint := &streamingpb.WALCheckpoint{
|
|
MessageId: &messagespb.MessageID{Id: messageID.Marshal()},
|
|
TimeTick: timeTick,
|
|
RecoveryMagic: recoveryMagic,
|
|
}
|
|
checkpoint := newWALCheckpointFromProto(walName, protoCheckpoint)
|
|
|
|
assert.True(t, messageID.EQ(checkpoint.MessageID))
|
|
assert.Equal(t, timeTick, checkpoint.TimeTick)
|
|
assert.Equal(t, recoveryMagic, checkpoint.Magic)
|
|
|
|
proto := checkpoint.IntoProto()
|
|
checkpoint2 := newWALCheckpointFromProto(walName, proto)
|
|
assert.True(t, messageID.EQ(checkpoint2.MessageID))
|
|
assert.Equal(t, timeTick, checkpoint2.TimeTick)
|
|
assert.Equal(t, recoveryMagic, checkpoint2.Magic)
|
|
|
|
checkpoint3 := checkpoint.Clone()
|
|
assert.True(t, messageID.EQ(checkpoint3.MessageID))
|
|
assert.Equal(t, timeTick, checkpoint3.TimeTick)
|
|
assert.Equal(t, recoveryMagic, checkpoint3.Magic)
|
|
}
|