milvus/internal/metastore/model/segment_index_test.go
cai.zhang e5f408dceb
Merge IndexCoord and DataCoord (#21267)
Signed-off-by: cai.zhang <cai.zhang@zilliz.com>
2023-01-04 19:37:36 +08:00

56 lines
1.2 KiB
Go

package model
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/milvus-io/milvus-proto/go-api/commonpb"
"github.com/milvus-io/milvus/internal/proto/datapb"
)
var (
segmentID = int64(1)
buildID = int64(1)
segmentIdxPb = &datapb.SegmentIndex{
CollectionID: colID,
PartitionID: partID,
SegmentID: segmentID,
NumRows: 1025,
IndexID: indexID,
BuildID: buildID,
NodeID: 0,
IndexVersion: 0,
State: commonpb.IndexState_Finished,
FailReason: "",
IndexFileKeys: nil,
Deleted: false,
CreateTime: 1,
SerializeSize: 0,
}
indexModel2 = &SegmentIndex{
CollectionID: colID,
PartitionID: partID,
SegmentID: segmentID,
NumRows: 1025,
IndexID: indexID,
BuildID: buildID,
NodeID: 0,
IndexState: commonpb.IndexState_Finished,
FailReason: "",
IndexVersion: 0,
IsDeleted: false,
CreateTime: 1,
IndexFileKeys: nil,
IndexSize: 0,
}
)
func TestUnmarshalSegmentIndexModel(t *testing.T) {
ret := UnmarshalSegmentIndexModel(segmentIdxPb)
assert.Equal(t, indexModel2.SegmentID, ret.SegmentID)
assert.Nil(t, UnmarshalSegmentIndexModel(nil))
}