milvus/internal/querynode/collection_test.go
godchen 639800241d
Make flow graph for delta channel (#11379)
Signed-off-by: godchen <qingxiang.chen@zilliz.com>
2021-11-09 09:27:04 +08:00

139 lines
5.0 KiB
Go

// Copyright (C) 2019-2020 Zilliz. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software distributed under the License
// is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
// or implied. See the License for the specific language governing permissions and limitations under the License.
package querynode
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestCollection_newCollection(t *testing.T) {
collectionID := UniqueID(0)
collectionMeta := genTestCollectionMeta(collectionID, false)
collection := newCollection(collectionMeta.ID, collectionMeta.Schema)
assert.Equal(t, collection.ID(), collectionID)
}
func TestCollection_deleteCollection(t *testing.T) {
collectionID := UniqueID(0)
collectionMeta := genTestCollectionMeta(collectionID, false)
collection := newCollection(collectionMeta.ID, collectionMeta.Schema)
assert.Equal(t, collection.ID(), collectionID)
deleteCollection(collection)
}
func TestCollection_schema(t *testing.T) {
collectionID := UniqueID(0)
collectionMeta := genTestCollectionMeta(collectionID, false)
collection := newCollection(collectionMeta.ID, collectionMeta.Schema)
schema := collection.Schema()
assert.Equal(t, collectionMeta.Schema.Name, schema.Name)
assert.Equal(t, len(collectionMeta.Schema.Fields), len(schema.Fields))
deleteCollection(collection)
}
func TestCollection_vChannel(t *testing.T) {
collectionID := UniqueID(0)
collectionMeta := genTestCollectionMeta(collectionID, false)
collection := newCollection(collectionMeta.ID, collectionMeta.Schema)
collection.addVChannels([]string{defaultVChannel})
collection.addVChannels([]string{defaultVChannel})
collection.addVChannels([]string{"TestCollection_addVChannel_channel"})
channels := collection.getVChannels()
assert.Equal(t, 2, len(channels))
}
func TestCollection_vDeltaChannel(t *testing.T) {
collectionID := UniqueID(0)
collectionMeta := genTestCollectionMeta(collectionID, false)
collection := newCollection(collectionMeta.ID, collectionMeta.Schema)
collection.addVDeltaChannels([]string{defaultHistoricalVChannel})
collection.addVDeltaChannels([]string{defaultHistoricalVChannel})
collection.addVDeltaChannels([]string{"TestCollection_addVDeltaChannel_channel"})
channels := collection.getVDeltaChannels()
assert.Equal(t, 2, len(channels))
}
func TestCollection_pChannel(t *testing.T) {
collectionID := UniqueID(0)
collectionMeta := genTestCollectionMeta(collectionID, false)
collection := newCollection(collectionMeta.ID, collectionMeta.Schema)
collection.addPChannels([]string{"TestCollection_addPChannel_channel-0"})
collection.addPChannels([]string{"TestCollection_addPChannel_channel-0"})
collection.addPChannels([]string{"TestCollection_addPChannel_channel-1"})
channels := collection.getPChannels()
assert.Equal(t, 2, len(channels))
}
func TestCollection_pDeltaChannel(t *testing.T) {
collectionID := UniqueID(0)
collectionMeta := genTestCollectionMeta(collectionID, false)
collection := newCollection(collectionMeta.ID, collectionMeta.Schema)
collection.addPDeltaChannels([]string{"TestCollection_addPDeltaChannel_channel-0"})
collection.addPDeltaChannels([]string{"TestCollection_addPDeltaChannel_channel-0"})
collection.addPDeltaChannels([]string{"TestCollection_addPDeltaChannel_channel-1"})
channels := collection.getPDeltaChannels()
assert.Equal(t, 2, len(channels))
}
func TestCollection_releaseTime(t *testing.T) {
collectionID := UniqueID(0)
collectionMeta := genTestCollectionMeta(collectionID, false)
collection := newCollection(collectionMeta.ID, collectionMeta.Schema)
t0 := Timestamp(1000)
collection.setReleaseTime(t0)
t1 := collection.getReleaseTime()
assert.Equal(t, t0, t1)
}
func TestCollection_releasePartition(t *testing.T) {
collectionID := UniqueID(0)
collectionMeta := genTestCollectionMeta(collectionID, false)
collection := newCollection(collectionMeta.ID, collectionMeta.Schema)
collection.addReleasedPartition(defaultPartitionID)
assert.Equal(t, 1, len(collection.releasedPartitions))
err := collection.checkReleasedPartitions([]UniqueID{defaultPartitionID})
assert.Error(t, err)
err = collection.checkReleasedPartitions([]UniqueID{UniqueID(1000)})
assert.NoError(t, err)
collection.deleteReleasedPartition(defaultPartitionID)
assert.Equal(t, 0, len(collection.releasedPartitions))
}
func TestCollection_loadType(t *testing.T) {
collectionID := UniqueID(0)
collectionMeta := genTestCollectionMeta(collectionID, false)
collection := newCollection(collectionMeta.ID, collectionMeta.Schema)
collection.setLoadType(loadTypeCollection)
lt := collection.getLoadType()
assert.Equal(t, loadTypeCollection, lt)
collection.setLoadType(loadTypePartition)
lt = collection.getLoadType()
assert.Equal(t, loadTypePartition, lt)
}