mirror of https://github.com/milvus-io/milvus.git
49 lines
1004 B
Go
49 lines
1004 B
Go
package model
|
|
|
|
import (
|
|
"testing"
|
|
|
|
pb "github.com/milvus-io/milvus/internal/proto/etcdpb"
|
|
"github.com/milvus-io/milvus/internal/util/typeutil"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
var (
|
|
segmentID = typeutil.UniqueID(1)
|
|
buildID = typeutil.UniqueID(1)
|
|
|
|
segmentIdxPb = &pb.SegmentIndexInfo{
|
|
CollectionID: colID,
|
|
PartitionID: partID,
|
|
SegmentID: segmentID,
|
|
FieldID: fieldID,
|
|
IndexID: indexID,
|
|
BuildID: buildID,
|
|
EnableIndex: true,
|
|
CreateTime: 1,
|
|
}
|
|
|
|
indexModel2 = &Index{
|
|
CollectionID: colID,
|
|
IndexID: indexID,
|
|
FieldID: fieldID,
|
|
SegmentIndexes: map[int64]SegmentIndex{
|
|
segmentID: {
|
|
Segment: Segment{
|
|
SegmentID: segmentID,
|
|
PartitionID: partID,
|
|
},
|
|
BuildID: buildID,
|
|
EnableIndex: true,
|
|
CreateTime: 1,
|
|
},
|
|
},
|
|
}
|
|
)
|
|
|
|
func TestUnmarshalSegmentIndexModel(t *testing.T) {
|
|
ret := UnmarshalSegmentIndexModel(segmentIdxPb)
|
|
assert.Equal(t, indexModel2, ret)
|
|
assert.Nil(t, UnmarshalSegmentIndexModel(nil))
|
|
}
|