mirror of https://github.com/milvus-io/milvus.git
37 lines
956 B
Go
37 lines
956 B
Go
package querynode
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestCollection_Partitions(t *testing.T) {
|
|
node := newQueryNodeMock()
|
|
collectionID := UniqueID(0)
|
|
initTestMeta(t, node, collectionID, 0)
|
|
|
|
collection, err := node.replica.getCollectionByID(collectionID)
|
|
assert.NoError(t, err)
|
|
|
|
partitions := collection.Partitions()
|
|
assert.Equal(t, 1, len(*partitions))
|
|
}
|
|
|
|
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)
|
|
}
|