// 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_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_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) }