wehub-resource-sync
498b235461
Build and test / Build and test AMD64 Ubuntu 22.04 (push) Failing after 0s
Publish Builder / amazonlinux2023 (push) Failing after 1s
Build and test / UT for Go (push) Has been skipped
Publish KRTE Images / KRTE (push) Failing after 1s
Build and test / Integration Test (push) Has been skipped
Build and test / Upload Code Coverage (push) Has been skipped
Publish Builder / rockylinux9 (push) Failing after 1s
Publish Builder / ubuntu22.04 (push) Failing after 0s
Publish Builder / ubuntu24.04 (push) Failing after 0s
Publish Gpu Builder / publish-gpu-builder (push) Failing after 1s
Publish Test Images / PyTest (push) Failing after 0s
Build and test / UT for Cpp (push) Has been cancelled
87 lines
3.4 KiB
Go
87 lines
3.4 KiB
Go
package utility
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/milvus-io/milvus-proto/go-api/v3/commonpb"
|
|
"github.com/milvus-io/milvus/pkg/v3/proto/streamingpb"
|
|
"github.com/milvus-io/milvus/pkg/v3/streaming/walimpls/impls/rmq"
|
|
)
|
|
|
|
func TestNewWALCheckpointFromProto(t *testing.T) {
|
|
assert.Nil(t, NewWALCheckpointFromProto(nil))
|
|
assert.Nil(t, NewWALCheckpointFromProto(nil).IntoProto())
|
|
|
|
messageID := rmq.NewRmqID(1)
|
|
timeTick := uint64(12345)
|
|
recoveryMagic := int64(1)
|
|
protoCheckpoint := &streamingpb.WALCheckpoint{
|
|
MessageId: messageID.IntoProto(),
|
|
TimeTick: timeTick,
|
|
RecoveryMagic: recoveryMagic,
|
|
}
|
|
checkpoint := NewWALCheckpointFromProto(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(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)
|
|
|
|
protoCheckpoint.ReplicateConfig = &commonpb.ReplicateConfiguration{}
|
|
protoCheckpoint.ReplicateCheckpoint = &commonpb.ReplicateCheckpoint{
|
|
ClusterId: "by-dev",
|
|
Pchannel: "p1",
|
|
MessageId: nil,
|
|
TimeTick: 0,
|
|
}
|
|
newCheckpoint := NewWALCheckpointFromProto(protoCheckpoint)
|
|
assert.Equal(t, "by-dev", newCheckpoint.ReplicateCheckpoint.ClusterID)
|
|
assert.Equal(t, "p1", newCheckpoint.ReplicateCheckpoint.PChannel)
|
|
assert.Equal(t, uint64(0), newCheckpoint.ReplicateCheckpoint.TimeTick)
|
|
assert.Nil(t, newCheckpoint.ReplicateCheckpoint.MessageID)
|
|
assert.NotNil(t, newCheckpoint.ReplicateConfig)
|
|
|
|
protoCheckpoint.ReplicateCheckpoint.MessageId = rmq.NewRmqID(2).IntoProto()
|
|
protoCheckpoint.ReplicateCheckpoint.TimeTick = 123456
|
|
|
|
newCheckpoint = NewWALCheckpointFromProto(protoCheckpoint)
|
|
assert.Equal(t, "by-dev", newCheckpoint.ReplicateCheckpoint.ClusterID)
|
|
assert.Equal(t, "p1", newCheckpoint.ReplicateCheckpoint.PChannel)
|
|
assert.Equal(t, uint64(123456), newCheckpoint.ReplicateCheckpoint.TimeTick)
|
|
assert.True(t, rmq.NewRmqID(2).EQ(newCheckpoint.ReplicateCheckpoint.MessageID))
|
|
assert.NotNil(t, newCheckpoint.ReplicateConfig)
|
|
|
|
proto = newCheckpoint.IntoProto()
|
|
checkpoint2 = NewWALCheckpointFromProto(proto)
|
|
assert.True(t, messageID.EQ(checkpoint2.MessageID))
|
|
assert.Equal(t, timeTick, checkpoint2.TimeTick)
|
|
assert.Equal(t, recoveryMagic, checkpoint2.Magic)
|
|
assert.Equal(t, "by-dev", checkpoint2.ReplicateCheckpoint.ClusterID)
|
|
assert.Equal(t, "p1", checkpoint2.ReplicateCheckpoint.PChannel)
|
|
assert.Equal(t, uint64(123456), checkpoint2.ReplicateCheckpoint.TimeTick)
|
|
assert.True(t, rmq.NewRmqID(2).EQ(checkpoint2.ReplicateCheckpoint.MessageID))
|
|
assert.NotNil(t, checkpoint2.ReplicateConfig)
|
|
|
|
checkpoint2 = newCheckpoint.Clone()
|
|
assert.True(t, messageID.EQ(checkpoint2.MessageID))
|
|
assert.Equal(t, timeTick, checkpoint2.TimeTick)
|
|
assert.Equal(t, recoveryMagic, checkpoint2.Magic)
|
|
assert.Equal(t, "by-dev", checkpoint2.ReplicateCheckpoint.ClusterID)
|
|
assert.Equal(t, "p1", checkpoint2.ReplicateCheckpoint.PChannel)
|
|
assert.Equal(t, uint64(123456), checkpoint2.ReplicateCheckpoint.TimeTick)
|
|
assert.True(t, rmq.NewRmqID(2).EQ(checkpoint2.ReplicateCheckpoint.MessageID))
|
|
assert.NotNil(t, checkpoint2.ReplicateConfig)
|
|
}
|