mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-05 13:28:49 +08:00
0edb49209d
Signed-off-by: longjiquan <jiquan.long@zilliz.com> Signed-off-by: longjiquan <jiquan.long@zilliz.com>
120 lines
2.9 KiB
Go
120 lines
2.9 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-proto/go-api/schemapb"
|
|
pb "github.com/milvus-io/milvus/internal/proto/etcdpb"
|
|
)
|
|
|
|
var (
|
|
colID int64 = 1
|
|
colName = "c"
|
|
fieldID int64 = 101
|
|
fieldName = "field110"
|
|
partID int64 = 20
|
|
partName = "testPart"
|
|
tenantID = "tenant-1"
|
|
typeParams = []*commonpb.KeyValuePair{
|
|
{
|
|
Key: "field110-k1",
|
|
Value: "field110-v1",
|
|
},
|
|
}
|
|
startPositions = []*commonpb.KeyDataPair{
|
|
{
|
|
Key: "k1",
|
|
Data: []byte{byte(1)},
|
|
},
|
|
}
|
|
|
|
colModel = &Collection{
|
|
TenantID: tenantID,
|
|
CollectionID: colID,
|
|
Name: colName,
|
|
AutoID: false,
|
|
Description: "none",
|
|
Fields: []*Field{fieldModel},
|
|
VirtualChannelNames: []string{"vch"},
|
|
PhysicalChannelNames: []string{"pch"},
|
|
ShardsNum: 1,
|
|
CreateTime: 1,
|
|
StartPositions: startPositions,
|
|
ConsistencyLevel: commonpb.ConsistencyLevel_Strong,
|
|
Partitions: []*Partition{
|
|
{
|
|
PartitionID: partID,
|
|
PartitionName: partName,
|
|
PartitionCreatedTimestamp: 1,
|
|
},
|
|
},
|
|
Properties: []*commonpb.KeyValuePair{
|
|
{
|
|
Key: "k",
|
|
Value: "v",
|
|
},
|
|
},
|
|
}
|
|
|
|
deprecatedColPb = &pb.CollectionInfo{
|
|
ID: colID,
|
|
Schema: &schemapb.CollectionSchema{
|
|
Name: colName,
|
|
Description: "none",
|
|
AutoID: false,
|
|
Fields: []*schemapb.FieldSchema{filedSchemaPb},
|
|
},
|
|
CreateTime: 1,
|
|
PartitionIDs: []int64{partID},
|
|
PartitionNames: []string{partName},
|
|
PartitionCreatedTimestamps: []uint64{1},
|
|
FieldIndexes: []*pb.FieldIndexInfo{
|
|
{
|
|
FiledID: fieldID,
|
|
IndexID: indexID,
|
|
},
|
|
},
|
|
VirtualChannelNames: []string{"vch"},
|
|
PhysicalChannelNames: []string{"pch"},
|
|
ShardsNum: 1,
|
|
StartPositions: startPositions,
|
|
ConsistencyLevel: commonpb.ConsistencyLevel_Strong,
|
|
Properties: []*commonpb.KeyValuePair{
|
|
{
|
|
Key: "k",
|
|
Value: "v",
|
|
},
|
|
},
|
|
}
|
|
)
|
|
|
|
func TestUnmarshalCollectionModel(t *testing.T) {
|
|
ret := UnmarshalCollectionModel(deprecatedColPb)
|
|
ret.TenantID = tenantID
|
|
assert.Equal(t, ret, colModel)
|
|
|
|
assert.Nil(t, UnmarshalCollectionModel(nil))
|
|
}
|
|
|
|
func TestMarshalCollectionModel(t *testing.T) {
|
|
assert.Nil(t, MarshalCollectionModel(nil))
|
|
}
|
|
|
|
func TestCollection_GetPartitionNum(t *testing.T) {
|
|
coll := &Collection{
|
|
Partitions: []*Partition{
|
|
{State: pb.PartitionState_PartitionCreated},
|
|
{State: pb.PartitionState_PartitionCreating},
|
|
{State: pb.PartitionState_PartitionCreated},
|
|
{State: pb.PartitionState_PartitionDropping},
|
|
{State: pb.PartitionState_PartitionCreated},
|
|
{State: pb.PartitionState_PartitionDropped},
|
|
},
|
|
}
|
|
assert.Equal(t, 3, coll.GetPartitionNum(true))
|
|
assert.Equal(t, 6, coll.GetPartitionNum(false))
|
|
}
|