mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-05 05:18:52 +08:00
39 lines
1.1 KiB
Go
39 lines
1.1 KiB
Go
package reader
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestCollection_NewPartition(t *testing.T) {
|
|
ctx := context.Background()
|
|
node := NewQueryNode(ctx, 0, 0)
|
|
|
|
var collection = node.NewCollection(0, "collection0", "")
|
|
var partition = collection.NewPartition("partition0")
|
|
|
|
assert.Equal(t, collection.CollectionName, "collection0")
|
|
assert.Equal(t, collection.CollectionID, uint64(0))
|
|
assert.Equal(t, partition.PartitionName, "partition0")
|
|
assert.Equal(t, len(collection.Partitions), 1)
|
|
}
|
|
|
|
func TestCollection_DeletePartition(t *testing.T) {
|
|
ctx := context.Background()
|
|
node := NewQueryNode(ctx, 0, 0)
|
|
|
|
var collection = node.NewCollection(0, "collection0", "")
|
|
var partition = collection.NewPartition("partition0")
|
|
|
|
assert.Equal(t, collection.CollectionName, "collection0")
|
|
assert.Equal(t, collection.CollectionID, uint64(0))
|
|
assert.Equal(t, partition.PartitionName, "partition0")
|
|
assert.Equal(t, len(collection.Partitions), 1)
|
|
|
|
collection.DeletePartition(node, partition)
|
|
|
|
assert.Equal(t, len(collection.Partitions), 0)
|
|
}
|