milvus/internal/metastore/model/segment_index_test.go
SimFG a55f739608
Separate public proto files (#19782)
Signed-off-by: SimFG <bang.fu@zilliz.com>

Signed-off-by: SimFG <bang.fu@zilliz.com>
2022-10-16 20:49:27 +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/indexpb"
)
var (
segmentID = int64(1)
buildID = int64(1)
segmentIdxPb = &indexpb.SegmentIndex{
CollectionID: colID,
PartitionID: partID,
SegmentID: segmentID,
NumRows: 1025,
IndexID: indexID,
BuildID: buildID,
NodeID: 0,
IndexVersion: 0,
State: commonpb.IndexState_Finished,
FailReason: "",
IndexFilesPaths: 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,
IndexFilePaths: nil,
IndexSize: 0,
}
)
func TestUnmarshalSegmentIndexModel(t *testing.T) {
ret := UnmarshalSegmentIndexModel(segmentIdxPb)
assert.Equal(t, indexModel2.SegmentID, ret.SegmentID)
assert.Nil(t, UnmarshalSegmentIndexModel(nil))
}